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