user_db test

1 files changed, 54 insertions, 0 deletions

diff --git a/src/server-tests/test_rz_user_db.py b/src/server-tests/test_rz_user_db.py
new file mode 100644
index 00000000..383a7c14
--- /dev/null
+++ b/src/server-tests/test_rz_user_db.py
@@ -0,0 +1,54 @@
+import tempfile
+import unittest
+from rz_user_db import User_DB
+
+
+class TestUser_DB(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(self):
+        pass
+
+    def setUp(self):
+        pass
+
+    def test_db_lifecycle(self):
+        tmp_file = tempfile.NamedTemporaryFile(prefix='rz_userdb_', dir='/tmp', suffix='_db')
+        tmp_file.close()
+
+        u_name = 'bob'
+        u_email = 'bob@example.org'
+
+        user_db = User_DB(db_path=tmp_file.name)
+        user_db.init(mode='c')
+        uid = user_db.user_add(u_name, u_email)
+        user_db.user_add_role(uid, 'admin')
+        user_db.shutdown()
+
+        # reload & validate
+        user_db = User_DB(db_path=tmp_file.name)
+        user_db.init()
+
+        # lookup_user__by_uid
+        ret_uid, ret_u = user_db.lookup_user__by_uid(uid)
+        self.assertEqual(ret_uid, uid)
+        self.assertEqual(u_name, ret_u['user_name'])
+        self.assertEqual(u_email, ret_u['email_address'])
+
+        # lookup_user__by_username
+        ret_uid, ret_u = user_db.lookup_user__by_email_address(u_email)
+        self.assertEqual(ret_uid, uid)
+        self.assertEqual(u_name, ret_u['user_name'])
+        self.assertEqual(u_email, ret_u['email_address'])
+
+        user_db.dump_to_file__str()
+        user_db.shutdown()
+
+    def tearDown(self):
+        pass
+
+def main():
+    unittest.main()
+
+if __name__ == "__main__":
+    main()