riak_test/tests/ts_A_activate_table_pass_2.erl

89 lines
2.8 KiB
Erlang

%% -------------------------------------------------------------------
%%
%% Copyright (c) 2015 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_A_activate_table_pass_2).
-behavior(riak_test).
-include_lib("eunit/include/eunit.hrl").
-export([confirm/0]).
confirm() ->
Cluster = ts_util:build_cluster(single),
% individual assert matches to show line numbers in failures
?assertMatch(
{error_creating_bucket_type, _},
create_and_activate_bucket_type(Cluster, table_def("^"))
),
?assertMatch(
{error_creating_bucket_type, _},
create_and_activate_bucket_type(Cluster, table_def("$"))
),
?assertMatch(
{error_creating_bucket_type, _},
create_and_activate_bucket_type(Cluster, table_def("!"))
),
?assertMatch(
{error_creating_bucket_type, _},
create_and_activate_bucket_type(
Cluster,
table_def("mytable", "!", "series", "time"))
),
?assertMatch(
{error_creating_bucket_type, _},
create_and_activate_bucket_type(
Cluster,
table_def("mytable", "#@#@", "series", "time"))
),
pass.
%%
create_and_activate_bucket_type(Cluster, {TableName, DDL}) ->
{ok, Out} = ts_util:create_bucket_type(Cluster, DDL, TableName, 3),
case iolist_to_binary(Out) of
<<"Error", _/binary>> ->
{error_creating_bucket_type, Out};
<<"Cannot create", _/binary>> ->
{error_creating_bucket_type, Out};
_ ->
Retries = 0,
ts_util:activate_bucket_type(Cluster, TableName, Retries)
end.
%%
table_def(TableName) ->
table_def(TableName, "family", "series", "time").
%%
table_def(TableName, FamilyName, SeriesName, TimeName) ->
{TableName, lists:flatten(io_lib:format(
"CREATE TABLE ~p ("
" ~p VARCHAR NOT NULL,"
" ~p VARCHAR NOT NULL,"
" ~p TIMESTAMP NOT NULL,"
" PRIMARY KEY ((~p, ~p, quantum(~p, 15, 'm')), "
" ~p, ~p, ~p))",
[TableName,
FamilyName, SeriesName, TimeName, % column defs
FamilyName, SeriesName, TimeName, % partition key
FamilyName, SeriesName, TimeName]))}. % local key