jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor login tests and import statements

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-25 16:17:13 -03:00
commit a76f90157e
1 changed files with 7 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

22
tests/test_login.py
View file

@ -1,17 +1,15 @@
from langflow.services.database.utils import session_getter
from langflow.services.getters import get_db_service
import pytest import pytest
from langflow.services.database.models.user import User
from langflow.services.auth.utils import get_password_hash from langflow.services.auth.utils import get_password_hash
from langflow.services.database.models.user import User
from langflow.services.database.utils import session_getter
from langflow.services.deps import get_db_service
@pytest.fixture @pytest.fixture
def test_user(): def test_user():
return User( return User(
username="testuser", username="testuser",
password=get_password_hash( password=get_password_hash("testpassword"), # Assuming password needs to be hashed
"testpassword"
), # Assuming password needs to be hashed
is_active=True, is_active=True,
is_superuser=False, is_superuser=False,
) )
@ -23,17 +21,13 @@ def test_login_successful(client, test_user):
session.add(test_user) session.add(test_user)
session.commit() session.commit()
response = client.post( response = client.post("api/v1/login", data={"username": "testuser", "password": "testpassword"})
"api/v1/login", data={"username": "testuser", "password": "testpassword"}
)
assert response.status_code == 200 assert response.status_code == 200
assert "access_token" in response.json() assert "access_token" in response.json()
def test_login_unsuccessful_wrong_username(client): def test_login_unsuccessful_wrong_username(client):
response = client.post( response = client.post("api/v1/login", data={"username": "wrongusername", "password": "testpassword"})
"api/v1/login", data={"username": "wrongusername", "password": "testpassword"}
)
assert response.status_code == 401 assert response.status_code == 401
assert response.json()["detail"] == "Incorrect username or password" assert response.json()["detail"] == "Incorrect username or password"
@ -43,8 +37,6 @@ def test_login_unsuccessful_wrong_password(client, test_user, session):
session.add(test_user) session.add(test_user)
session.commit() session.commit()
response = client.post( response = client.post("api/v1/login", data={"username": "testuser", "password": "wrongpassword"})
"api/v1/login", data={"username": "testuser", "password": "wrongpassword"}
)
assert response.status_code == 401 assert response.status_code == 401
assert response.json()["detail"] == "Incorrect username or password" assert response.json()["detail"] == "Incorrect username or password"