Update datetime handling in ApiKey and Variable models (#1673)

* Update datetime handling in ApiKey and Variable models

* Refactor test_login_successful function in test_login.py

* Fix nullable attribute for created_at and updated_at fields in Variable model

tests/test_api_key.py

@@ -1,5 +1,4 @@
 import pytest
-
 from langflow.services.database.models.api_key import ApiKeyCreate
 
 
@@ -7,7 +6,7 @@ from langflow.services.database.models.api_key import ApiKeyCreate
 def api_key(client, logged_in_headers, active_user):
     api_key = ApiKeyCreate(name="test-api-key")
 
-    response = client.post("api/v1/api_key", data=api_key.json(), headers=logged_in_headers)
+    response = client.post("api/v1/api_key", data=api_key.model_dump_json(), headers=logged_in_headers)
     assert response.status_code == 200, response.text
     return response.json()
 

tests/test_initial_setup.py

@@ -1,17 +1,12 @@
 from datetime import datetime
 from pathlib import Path
 
-import pytest
-from langflow.graph.graph.base import Graph
-from langflow.graph.schema import RunOutputs
 from langflow.initial_setup.setup import (
     STARTER_FOLDER_NAME,
     create_or_update_starter_projects,
     get_project_data,
     load_starter_projects,
 )
-from langflow.memory import delete_messages
-from langflow.processing.process import process_tweaks
 from langflow.services.database.models.flow.model import Flow
 from langflow.services.deps import session_scope
 from sqlalchemy import func

tests/test_login.py

@@ -1,8 +1,8 @@
 import pytest
 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
+from langflow.services.deps import session_scope
+from sqlalchemy.exc import IntegrityError
 
 
 @pytest.fixture
@@ -17,9 +17,12 @@ def test_user():
 
 def test_login_successful(client, test_user):
     # Adding the test user to the database
-    with session_getter(get_db_service()) as session:
-        session.add(test_user)
-        session.commit()
+    try:
+        with session_scope() as session:
+            session.add(test_user)
+            session.commit()
+    except IntegrityError:
+        pass
 
     response = client.post("api/v1/login", data={"username": "testuser", "password": "testpassword"})
     assert response.status_code == 200