From 7b897906b9dcf660dc0ccbb7677d9d76b4639d73 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 25 Aug 2023 10:17:36 -0300
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20feat(test=5Flogin.py):=20add=20test?=
 =?UTF-8?q?s=20for=20login=20functionality?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

🐛 fix(test_login.py): fix typo in test_login_unsuccessful_wrong_username test

🐛 fix(test_login.py): fix typo in test_login_unsuccessful_wrong_password test
---
 tests/test_login.py | 47 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 tests/test_login.py

diff --git a/tests/test_login.py b/tests/test_login.py
new file mode 100644
index 000000000..07abb35ab
--- /dev/null
+++ b/tests/test_login.py
@@ -0,0 +1,47 @@
+import pytest
+from langflow.services.database.models.user import User
+from langflow.services.auth.utils import get_password_hash
+
+
+@pytest.fixture
+def test_user():
+    return User(
+        username="testuser",
+        password=get_password_hash(
+            "testpassword"
+        ),  # Assuming password needs to be hashed
+        is_active=True,
+        is_superuser=False,
+    )
+
+
+def test_login_successful(client, test_user, session):
+    # Adding the test user to the database
+    session.add(test_user)
+    session.commit()
+
+    response = client.post(
+        "api/v1/login", data={"username": "testuser", "password": "testpassword"}
+    )
+    assert response.status_code == 200
+    assert "access_token" in response.json()
+
+
+def test_login_unsuccessful_wrong_username(client):
+    response = client.post(
+        "api/v1/login", data={"username": "wrongusername", "password": "testpassword"}
+    )
+    assert response.status_code == 401
+    assert response.json()["detail"] == "Incorrect username or password"
+
+
+def test_login_unsuccessful_wrong_password(client, test_user, session):
+    # Adding the test user to the database
+    session.add(test_user)
+    session.commit()
+
+    response = client.post(
+        "api/v1/login", data={"username": "testuser", "password": "wrongpassword"}
+    )
+    assert response.status_code == 401
+    assert response.json()["detail"] == "Incorrect username or password"