klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-05 21:29:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update chat director, client tests to call chat API using new POST method

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2024-09-11 16:45:44 -07:00
parent 3f51af9a96
commit bc2e889d72
3 changed files with 160 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_client.py
View File

@@ -459,11 +459,12 @@ def test_user_no_data_returns_empty(client, sample_org_data, api_user3: KhojApiU
@pytest.mark.django_db(transaction=True)
async def test_chat_with_unauthenticated_user(chat_client_with_auth, api_user2: KhojApiUser):
# Arrange
query = "Hello!"
headers = {"Authorization": f"Bearer {api_user2.token}"}
# Act
auth_response = chat_client_with_auth.post(f'/api/chat?q="Hello!"', headers=headers)
no_auth_response = chat_client_with_auth.post(f'/api/chat?q="Hello!"')
auth_response = chat_client_with_auth.post(f"/api/chat", json={"q": query}, headers=headers)
no_auth_response = chat_client_with_auth.post(f"/api/chat", json={"q": query})
# Assert
assert auth_response.status_code == 200