169 lines
4.7 KiB
Erlang
169 lines
4.7 KiB
Erlang
%%
|
|
%% Test suite for esqlite.
|
|
%%
|
|
|
|
-module(esqlite_test).
|
|
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
|
|
open_single_database_test() ->
|
|
{ok, _C1} = esqlite3:open("test.db"),
|
|
ok.
|
|
|
|
open_multiple_same_databases_test() ->
|
|
{ok, _C1} = esqlite3:open("test.db"),
|
|
{ok, _C2} = esqlite3:open("test.db"),
|
|
ok.
|
|
|
|
open_multiple_different_databases_test() ->
|
|
{ok, _C1} = esqlite3:open("test1.db"),
|
|
{ok, _C2} = esqlite3:open("test2.db"),
|
|
ok.
|
|
|
|
simple_query_test() ->
|
|
{ok, Db} = esqlite3:open(":memory:"),
|
|
ok = esqlite3:exec(Db, "begin;"),
|
|
ok = esqlite3:exec(Db, "create table test_table(one varchar(10), two int);"),
|
|
ok = esqlite3:exec(Db, ["insert into test_table values(", "\"hello1\"", ",", "10" ");"]),
|
|
ok = esqlite3:exec(Db, ["insert into test_table values(", "\"hello2\"", ",", "11" ");"]),
|
|
ok = esqlite3:exec(Db, ["insert into test_table values(", "\"hello3\"", ",", "12" ");"]),
|
|
ok = esqlite3:exec(Db, ["insert into test_table values(", "\"hello4\"", ",", "13" ");"]),
|
|
ok = esqlite3:exec(Db, "commit;"),
|
|
ok = esqlite3:exec(Db, "select * from test_table;"),
|
|
ok.
|
|
|
|
prepare_test() ->
|
|
{ok, Db} = esqlite3:open(":memory:"),
|
|
esqlite3:exec(Db, "begin;"),
|
|
esqlite3:exec(Db, "create table test_table(one varchar(10), two int);"),
|
|
{ok, Statement} = esqlite3:prepare(Db, "insert into test_table values(\"one\", 2)"),
|
|
|
|
'$done' = esqlite3:step(Statement),
|
|
|
|
ok = esqlite3:exec(Db, ["insert into test_table values(", "\"hello4\"", ",", "13" ");"]),
|
|
{ok, St2} = esqlite3:prepare(Db, "select * from test_table order by two"),
|
|
[{"one", 2}, {"hello4", 13}] = exec(St2),
|
|
esqlite3:exec(Db, "commit;"),
|
|
esqlite3:close(Db),
|
|
|
|
ok.
|
|
|
|
bind_test() ->
|
|
{ok, Db} = esqlite3:open(":memory:"),
|
|
ok = esqlite3:exec(Db, "begin;"),
|
|
ok = esqlite3:exec(Db, "create table test_table(one varchar(10), two int);"),
|
|
ok = esqlite3:exec(Db, ["insert into test_table values(", "\"hello1\"", ",", "10" ");"]),
|
|
ok = esqlite3:exec(Db, ["insert into test_table values(", "\"hello2\"", ",", "11" ");"]),
|
|
ok = esqlite3:exec(Db, ["insert into test_table values(", "\"hello3\"", ",", "12" ");"]),
|
|
ok = esqlite3:exec(Db, ["insert into test_table values(", "\"hello4\"", ",", "13" ");"]),
|
|
ok = esqlite3:exec(Db, "commit;"),
|
|
|
|
%% Create a prepared statement
|
|
{ok, Statement} = esqlite3:prepare(Db, "insert into test_table values(?1, ?2)"),
|
|
esqlite3:bind(Statement, [one, 2]),
|
|
esqlite3:step(Statement),
|
|
esqlite3:bind(Statement, ["three", 4]),
|
|
esqlite3:step(Statement),
|
|
esqlite3:bind(Statement, [<<"five">>, 6]),
|
|
esqlite3:step(Statement),
|
|
|
|
[{"one", 2}] = q(Db, "select * from test_table where two = '2'"),
|
|
[{"three", 4}] = q(Db, "select * from test_table where two = 4"),
|
|
[{<<"five">>, 6}] = q(Db, "select * from test_table where two = 6"),
|
|
|
|
ok.
|
|
|
|
gen_db_test() ->
|
|
{ok, Conn} = gen_db:open(sqlite, ":memory:"),
|
|
[] = gen_db:execute("create table some_shit(hole_one varchar(10), hole_two int);", [], Conn),
|
|
[] = gen_db:execute("insert into some_shit values('dung', 100);", Conn),
|
|
[] = gen_db:execute("insert into some_shit values(?, ?);", ["manure", 1000], Conn),
|
|
ok.
|
|
|
|
|
|
%% Handy functions...
|
|
%%
|
|
q(Connection, Sql) ->
|
|
{ok, Statement} = esqlite3:prepare(Connection, Sql),
|
|
exec(Statement).
|
|
|
|
q(Connection, Sql, []) ->
|
|
q(Connection, Sql);
|
|
q(Connection, Sql, Args) ->
|
|
{ok, Statement} = esqlite3:prepare(Connection, Sql),
|
|
esqlite3:bind(Statement, Args),
|
|
exec(Statement).
|
|
|
|
|
|
exec(Statement) ->
|
|
exec(Statement, [], 0).
|
|
|
|
exec(_Statement, _Acc, Tries) when Tries > 5 ->
|
|
throw(too_many_tries);
|
|
exec(Statement, Acc, Tries) ->
|
|
case esqlite3:step(Statement) of
|
|
'$done' ->
|
|
lists:reverse(Acc);
|
|
'$busy' ->
|
|
timer:sleep(100),
|
|
exec(Statement, Acc, Tries + 1);
|
|
V when is_tuple(V) ->
|
|
exec(Statement, [V | Acc], 0)
|
|
end.
|
|
|
|
%% esqlite:bind(Statement, ":one", "hello"),
|
|
%% esqlite:bind(Statement, ":two", 11),
|
|
|
|
%% %% of
|
|
%% esqlite:bind(Statement, ["hello", 11]),
|
|
|
|
%% {ok, []} = esqlite:execute(Statement),
|
|
|
|
%% esqlite:exec(Db, "commit;"),
|
|
|
|
%%
|
|
|
|
% api...
|
|
|
|
%% q(Sql, Connection) ->
|
|
%% ok.
|
|
|
|
%% q(Sql, Args, Connection) ->
|
|
%% ok.
|
|
|
|
|
|
%%
|
|
%% options --
|
|
%% [{sync, bool()}
|
|
%% -- asynchronous or synchronous. Default true
|
|
%% {receiver, (pid()|function/1|{M, F, A}},
|
|
%% -- for asynchronous requests, to send the result to.
|
|
%% {stream, pid()},
|
|
%% -- instead of receiving all rows at once, send them one by one.
|
|
%% {..},
|
|
%% ]
|
|
%%
|
|
%% return
|
|
%% {ok, Result} -- result kan dan headers, rows of request-id (voor async)
|
|
%% {error, ...
|
|
%% q(Sql, Args, Options, Connection) ->
|
|
%% ok.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|