mirror of
https://github.com/valitydev/riak_test.git
synced 2024-11-06 08:35:22 +00:00
75 lines
2.8 KiB
Erlang
75 lines
2.8 KiB
Erlang
|
%% -------------------------------------------------------------------
|
||
|
|
%%
|
||
|
|
%% Copyright (c) 2016 Basho Technologies, Inc.
|
||
|
|
%%
|
||
|
|
%% This file is provided to you under the Apache License,
|
||
|
|
%% Version 2.0 (the "License"); you may not use this file
|
||
|
|
%% except in compliance with the License. You may obtain
|
||
|
|
%% a copy of the License at
|
||
|
|
%%
|
||
|
|
%% http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
%%
|
||
|
|
%% Unless required by applicable law or agreed to in writing,
|
||
|
|
%% software distributed under the License is distributed on an
|
||
|
|
%% "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||
|
|
%% KIND, either express or implied. See the License for the
|
||
|
|
%% specific language governing permissions and limitations
|
||
|
|
%% under the License.
|
||
|
|
%%
|
||
|
|
%% -------------------------------------------------------------------
|
||
|
|
|
||
|
|
-module(ts_simple_insert).
|
||
|
|
|
||
|
|
-behavior(riak_test).
|
||
|
|
|
||
|
|
-include_lib("eunit/include/eunit.hrl").
|
||
|
|
|
||
|
|
-export([confirm/0]).
|
||
|
|
|
||
|
|
confirm() ->
|
||
|
|
DDL = ts_util:get_ddl(),
|
||
|
|
Table = ts_util:get_default_bucket(),
|
||
|
|
Columns = ts_util:get_cols(),
|
||
|
|
Expected =
|
||
|
|
{ok,
|
||
|
|
"GeoCheckin has been activated\n"
|
||
|
|
"\n"
|
||
|
|
"WARNING: Nodes in this cluster can no longer be\n"
|
||
|
|
"downgraded to a version of Riak prior to 2.0\n"},
|
||
|
|
{Cluster, Conn} = ts_util:cluster_and_connect(single),
|
||
|
|
Got = ts_util:create_and_activate_bucket_type(Cluster, DDL),
|
||
|
|
?assertEqual(Expected, Got),
|
||
|
|
|
||
|
|
Data1 = ts_util:get_valid_select_data(),
|
||
|
|
Insert1Fn = fun(Datum, Acc) ->
|
||
|
|
[ts_util:ts_insert(Conn, Table, Columns, Datum) | Acc]
|
||
|
|
end,
|
||
|
|
Got1 = lists:reverse(lists:foldl(Insert1Fn, [], Data1)),
|
||
|
|
Expected1 = lists:duplicate(10, {[],[]}),
|
||
|
|
Result1 = ts_util:assert("Insert With Columns", Expected1, Got1),
|
||
|
|
|
||
|
|
Qry2 = "select * from GeoCheckin Where time >= 1 and time <= 10 and myfamily = 'family1' and myseries ='seriesX'",
|
||
|
|
Got2 = ts_util:single_query(Conn, Qry2),
|
||
|
|
Expected2 = {Columns, ts_util:exclusive_result_from_data(Data1, 1, 10)},
|
||
|
|
Result2 = ts_util:assert("Insert With Columns (results)", Expected2, Got2),
|
||
|
|
|
||
|
|
Data3 = ts_util:get_valid_select_data(fun() -> lists:seq(11, 20) end),
|
||
|
|
Insert3Fn = fun(Datum, Acc) ->
|
||
|
|
[ts_util:ts_insert_no_columns(Conn, Table, Datum) | Acc]
|
||
|
|
end,
|
||
|
|
Got3 = lists:reverse(lists:foldl(Insert3Fn, [], Data3)),
|
||
|
|
Result3 = ts_util:assert("Insert Without Columns", Expected1, Got3),
|
||
|
|
|
||
|
|
Qry4 = "select * from GeoCheckin Where time >= 11 and time <= 20 and myfamily = 'family1' and myseries ='seriesX'",
|
||
|
|
Got4 = ts_util:single_query(Conn, Qry4),
|
||
|
|
Expected4 = {Columns, ts_util:exclusive_result_from_data(Data3, 1, 10)},
|
||
|
|
Result4 = ts_util:assert("Insert Without Columns (results)", Expected4, Got4),
|
||
|
|
|
||
|
|
ts_util:results([
|
||
|
|
Result1,
|
||
|
|
Result2,
|
||
|
|
Result3,
|
||
|
|
Result4
|
||
|
|
]),
|
||
|
|
pass.
|