57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
import streamlit
|
|
|
|
import queries.user as user_query
|
|
import user
|
|
|
|
def test_get_default_user():
|
|
users = user_query.find_default_user()
|
|
assert len(users) == 1
|
|
assert users["email"][0] == "default"
|
|
|
|
def test_update_default_user_and_find_user():
|
|
user_query.update_default_user(email="test@testbase.com", name="Test User", oidc_user_id="1111-2222-3333")
|
|
|
|
users = user_query.find_default_user()
|
|
assert len(users) == 0
|
|
|
|
users = user_query.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_query.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_query.create_user(email="test@testbase.com", name="Test User", oidc_user_id="333-4444-5555")
|
|
|
|
users = user_query.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_query.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
|