Add user specific profiles

tests/database/user_db_test.py

@@ -0,0 +1,55 @@
+import streamlit
+
+import queries.user as user
+
+def test_get_default_user():
+    users = user.find_default_user()
+    assert len(users) == 1
+    assert users["email"][0] == "default"
+
+def test_update_default_user_and_find_user():
+    user.update_default_user(email="test@testbase.com", name="Test User", oidc_user_id="1111-2222-3333")
+
+    users = user.find_default_user()
+    assert len(users) == 0
+
+    users = user.find_user_by_oidc_id("1111-2222-3333")
+    assert len(users) == 1
+    assert users["email"][0] == "test@testbase.com"
+    assert users["name"][0] == "Test User"
+    assert users["oidc_user_id"][0] == "1111-2222-3333"
+
+    users = user.find_user_by_email("test@testbase.com")
+    assert len(users) == 1
+    assert users["email"][0] == "test@testbase.com"
+    assert users["name"][0] == "Test User"
+    assert users["oidc_user_id"][0] == "1111-2222-3333"
+
+def test_add_user():
+    user.create_user(email="test@testbase.com", name="Test User", oidc_user_id="333-4444-5555")
+
+    users = user.find_user_by_oidc_id("333-4444-5555")
+    assert len(users) == 1
+    assert users["email"][0] == "test@testbase.com"
+    assert users["name"][0] == "Test User"
+    assert users["oidc_user_id"][0] == "333-4444-5555"
+
+    users = user.find_user_by_email("test@testbase.com")
+    assert len(users) == 1
+    assert users["email"][0] == "test@testbase.com"
+    assert users["name"][0] == "Test User"
+    assert users["oidc_user_id"][0] == "333-4444-5555"
+
+def test_update_user_in_session():
+    userInfo = lambda: None
+    userInfo.email ="test@testbase.com"
+    userInfo.name = "Test User"
+    userInfo.sub = "1111-2222-3333"
+
+    user.set_user_in_session(userInfo)
+
+    state = streamlit.session_state
+    assert state.user_id == 1
+    assert state.user_name == userInfo.name
+    assert state.user_email == userInfo.email
+    assert state.user_external_id == userInfo.sub