eunit test

Author: wulei

c_src/elmdb_nif.c

@@ -123,12 +123,12 @@ static ERL_NIF_TERM elmdb_init(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv
     CHECK(mdb_env_set_maxdbs(ctx, 256), err2);
     CHECK(mdb_env_set_mapsize(ctx, 10485760), err2);
 
-    unsigned int envFlags = 0; 
+    unsigned int envFlags = 0 | MDB_NOTLS; 
     CHECK(mdb_env_open(ctx, dirname, envFlags, 0664), err2);
 
     // Each transaction belongs to one thread.
     // The MDB_NOTLS flag changes this for read-only transactions
-//    CHECK(mdb_env_set_flags(handle->env, MDB_NOTLS, 1), err2);
+//    CHECK(mdb_env_set_flags(ctx, MDB_NOTLS, 1), err2);
 
     MDB_txn *txn = NULL;
     CHECK(mdb_txn_begin(ctx, NULL, MDB_RDONLY, &txn), err2);

test/elmdb_test.erl

@@ -0,0 +1,26 @@
+-module(elmdb_test).
+
+-include_lib("eunit/include/eunit.hrl").
+
+startup() ->
+    Dir = "./mytestdb1/",
+    filelib:ensure_dir(Dir),
+    D = elmdb:init(Dir),
+    D.
+
+teardown(_) ->
+    Dir = "./mytestdb1/",
+    [file:delete(F) || F <- filelib:wildcard(Dir ++ "*")],
+    file:del_dir(Dir).
+
+
+list_db(D) ->
+    PI = <<"3.14159">>,
+    E = <<"2.71828">>,
+    elmdb:put(D, {"math", <<"pi">>}, PI),
+    elmdb:put(D, {"math", <<"e">>},  E),
+    [?_assertEqual(PI, elmdb:get(D, {"math", <<"pi">>})),
+     ?_assertEqual(E, elmdb:get(D, {"math", <<"e">>}))].
+
+list_test_() ->
+    [{"Try to list all dbs", {setup, fun startup/0, fun teardown/1, fun list_db/1}}].