fix tests

This commit is contained in:
hyusap 2024-01-30 22:30:11 -05:00
parent e6e6a4843d
commit daae76203c
1 changed files with 18 additions and 12 deletions

View File

@ -4,8 +4,9 @@ from uuid import uuid1
def test_session_creation_retrieval():
client = Client("http://localhost:8000")
created_session = client.create_session("ayush")
retrieved_session = client.get_session("ayush", created_session.id)
user_id = str(uuid1())
created_session = client.create_session(user_id)
retrieved_session = client.get_session(user_id, created_session.id)
assert retrieved_session.id == created_session.id
assert retrieved_session.is_active == True
assert retrieved_session.location_id == "default"
@ -14,38 +15,43 @@ def test_session_creation_retrieval():
def test_session_multiple_retrieval():
client = Client("http://localhost:8000")
user = str(uuid1())
created_session_1 = client.create_session(user)
created_session_2 = client.create_session(user)
retrieved_sessions = client.get_sessions(user)
user_id = str(uuid1())
created_session_1 = client.create_session(user_id)
created_session_2 = client.create_session(user_id)
retrieved_sessions = client.get_sessions(user_id)
assert len(retrieved_sessions) == 2
assert retrieved_sessions[0].id == created_session_1.id
assert retrieved_sessions[1].id == created_session_2.id
def test_session_update():
user_id = str(uuid1())
client = Client("http://localhost:8000")
created_session = client.create_session("ayush")
created_session = client.create_session(user_id)
assert created_session.update({"foo": "bar"})
retrieved_session = client.get_session("ayush", created_session.id)
retrieved_session = client.get_session(user_id, created_session.id)
assert retrieved_session.session_data == {"foo": "bar"}
def test_session_deletion():
user_id = str(uuid1())
client = Client("http://localhost:8000")
created_session = client.create_session("ayush")
created_session = client.create_session(user_id)
assert created_session.is_active == True
created_session.delete()
retrieved_session = client.get_session("ayush", created_session.id)
assert created_session.is_active == False
retrieved_session = client.get_session(user_id, created_session.id)
assert retrieved_session.is_active == False
assert retrieved_session.id == created_session.id
def test_messages():
user_id = str(uuid1())
client = Client("http://localhost:8000")
created_session = client.create_session("ayush")
created_session = client.create_session(user_id)
created_session.create_message(is_user=True, content="Hello")
created_session.create_message(is_user=False, content="Hi")
retrieved_session = client.get_session("ayush", created_session.id)
retrieved_session = client.get_session(user_id, created_session.id)
messages = retrieved_session.get_messages()
assert len(messages) == 2
user_message, ai_message = messages