iatoolkit 0.7.11__tar.gz → 0.7.12__tar.gz

{iatoolkit-0.7.11 → iatoolkit-0.7.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.7.11
+Version: 0.7.12
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-0.7.11 → iatoolkit-0.7.12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "iatoolkit"
-version = "0.7.11"
+version = "0.7.12"
 requires-python = ">=3.11"
 description = "IAToolkit"
 readme = "readme.md"
@@ -19,11 +19,9 @@ package-dir = {"" = "src"}
 
 [tool.setuptools.packages.find]
 where = ["src"]
-include = ["iatoolkit*", "services*", "common*", "repositories*", "infra*", "views*"]
 
 [tool.setuptools.dynamic]
 dependencies = { file = ["requirements.txt"] }
 
 [tool.setuptools.package-data]
-# 'src/system_prompts' relativo al paquete 'iatoolkit'
 iatoolkit = ["system_prompts/*.prompt", "templates/**/*", "static/**/*"]

{iatoolkit-0.7.11 → iatoolkit-0.7.12}/src/iatoolkit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.7.11
+Version: 0.7.12
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-0.7.11 → iatoolkit-0.7.12}/src/iatoolkit.egg-info/SOURCES.txt

@@ -129,4 +129,70 @@ src/iatoolkit/views/signup_view.py
 src/iatoolkit/views/tasks_review_view.py
 src/iatoolkit/views/tasks_view.py
 src/iatoolkit/views/user_feedback_view.py
-src/iatoolkit/views/verify_user_view.py
+src/iatoolkit/views/verify_user_view.py
+src/tests/__init__.py
+src/tests/conftest.py
+src/tests/common/__init__.py
+src/tests/common/test_auth.py
+src/tests/common/test_routes.py
+src/tests/common/test_session_manager.py
+src/tests/common/test_util.py
+src/tests/companies/__init__.py
+src/tests/infra/__init__.py
+src/tests/infra/test_call_service.py
+src/tests/infra/test_database_manager.py
+src/tests/infra/test_gemini_adapter.py
+src/tests/infra/test_google_chat_app.py
+src/tests/infra/test_llm_client.py
+src/tests/infra/test_llm_proxy.py
+src/tests/infra/test_mail_app.py
+src/tests/infra/test_openai_adapter.py
+src/tests/infra/test_redis_session_manager_service.py
+src/tests/infra/connectors/__init__.py
+src/tests/infra/connectors/test_google_drive_connector.py
+src/tests/infra/connectors/test_local_file_connector.py
+src/tests/infra/connectors/test_s3_connector.py
+src/tests/repositories/__init__.py
+src/tests/repositories/test_database_manager.py
+src/tests/repositories/test_document_repo.py
+src/tests/repositories/test_llm_query_repo.py
+src/tests/repositories/test_models.py
+src/tests/repositories/test_profile_repo.py
+src/tests/repositories/test_tasks_repo.py
+src/tests/repositories/test_vs_repo.py
+src/tests/services/__init__.py
+src/tests/services/test_dispatcher_service.py
+src/tests/services/test_document_service.py
+src/tests/services/test_excel_service.py
+src/tests/services/test_file_processor_service.py
+src/tests/services/test_history_service.py
+src/tests/services/test_jwt_service.py
+src/tests/services/test_load_documents_service.py
+src/tests/services/test_mail_service.py
+src/tests/services/test_profile_service.py
+src/tests/services/test_prompt_manager_service.py
+src/tests/services/test_query_service.py
+src/tests/services/test_search_service.py
+src/tests/services/test_sql_service.py
+src/tests/services/test_tasks_service.py
+src/tests/services/test_user_feedback_service.py
+src/tests/services/test_user_session_context_service.py
+src/tests/views/__init__.py
+src/tests/views/test_change_password_view.py
+src/tests/views/test_chat_token_request_view.py
+src/tests/views/test_chat_view.py
+src/tests/views/test_download_file_view.py
+src/tests/views/test_external_chat_login_view.py
+src/tests/views/test_external_login_view.py
+src/tests/views/test_file_store_view.py
+src/tests/views/test_forgot_password_view.py
+src/tests/views/test_history_view.py
+src/tests/views/test_home_view.py
+src/tests/views/test_llm_query_view.py
+src/tests/views/test_login_view.py
+src/tests/views/test_prompt_view.py
+src/tests/views/test_signup_view.py
+src/tests/views/test_tasks_review_view.py
+src/tests/views/test_tasks_view.py
+src/tests/views/test_user_feedback_view.py
+src/tests/views/test_verify_user_view.py

{iatoolkit-0.7.11 → iatoolkit-0.7.12}/src/iatoolkit.egg-info/top_level.txt

@@ -1 +1,2 @@
 iatoolkit
+tests

iatoolkit-0.7.12/src/tests/__init__.py

@@ -0,0 +1,5 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+

iatoolkit-0.7.12/src/tests/common/test_auth.py

@@ -0,0 +1,279 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import Flask, Response
+from unittest.mock import patch, MagicMock
+from iatoolkit.common.auth import IAuthentication
+import pytest
+from datetime import datetime, timezone
+from iatoolkit.repositories.profile_repo import ProfileRepo
+from iatoolkit.services.jwt_service import JWTService
+from werkzeug.exceptions import HTTPException
+
+CURRENT_TIME = datetime.now(timezone.utc).timestamp()
+
+
+class TestAuth:
+
+    def setup_method(self):
+        self.app = Flask(__name__)
+        self.app.config['TESTING'] = True
+
+        self.profile_repo = MagicMock(spec=ProfileRepo)
+        self.jwt_service = MagicMock(spec=JWTService)
+        self.iauth_service = IAuthentication(
+            profile_repo=self.profile_repo,
+            jwt_service=self.jwt_service
+        )
+
+        @self.app.route('/<company_short_name>/protected')
+        def protected(company_short_name):
+            with self.app.test_request_context():
+                auth_result = self.iauth_service.check_if_user_is_logged_in(company_short_name)
+            if isinstance(auth_result, Response):
+                return auth_result
+            return "Access Granted", 200
+
+        @self.app.route('/login')
+        def login():
+            return "Login Page", 200
+
+        self.client = self.app.test_client()
+
+    def teardown_method(self):
+        patch.stopall()
+
+    # --- Pruebas para check_if_user_is_logged_in (de TestAuth original) ---
+
+    def test_allows_access_if_user_authenticated_and_active(self):
+        with patch('iatoolkit.common.auth.SessionManager.get') as mock_get, \
+             patch('iatoolkit.common.auth.SessionManager.set') as mock_set:
+            mock_get.side_effect = lambda key, default=None: {
+                'user': {'id': 1, 'username': 'test_user'},
+                'company_short_name': 'test_company',
+                'last_activity': CURRENT_TIME
+            }.get(key, default)
+
+            response = self.client.get('/test_company/protected')
+
+            assert response.status_code == 200
+            assert response.data == b"Access Granted"
+            mock_set.assert_called_with("last_activity", pytest.approx(CURRENT_TIME, 1))
+
+    def test_redirect_if_user_not_authenticated(self):
+        with patch('iatoolkit.common.auth.SessionManager.get') as mock_get:
+            mock_get.side_effect = lambda key, default=None: None if key == "user" else default
+            response = self.client.get('/test_company/protected')
+            assert response.status_code == 302
+            assert "/login" in response.headers['Location']
+
+    def test_redirect_if_last_activity_missing(self):
+        with patch('iatoolkit.common.auth.SessionManager.get') as mock_get, \
+             patch('iatoolkit.common.auth.SessionManager.clear') as mock_clear:
+            mock_get.side_effect = lambda key, default=None: {
+                'user': {'id': 1, 'username': 'test_user'},
+                'company_short_name': 'test_company'
+            }.get(key, default)
+
+            response = self.client.get('/test_company/protected')
+
+            assert response.status_code == 302
+            assert "/login" in response.headers['Location']
+            mock_clear.assert_called_once()
+
+    # --- Pruebas para verify (de TestAuthVerify original) ---
+
+    def test_verify_with_jwt_success(self):
+        with patch.object(self.iauth_service, '_authenticate_via_chat_jwt') as mock_auth_jwt, \
+             patch.object(self.iauth_service, '_authenticate_via_api_key') as mock_auth_api_key:
+            mock_auth_jwt.return_value = (123, 'ext_user_1', None)
+
+            with self.app.test_request_context():
+                result = self.iauth_service.verify('some_company')
+
+            assert result['success'] is True
+            assert result['company_id'] == 123
+            assert result['external_user_id'] == 'ext_user_1'
+            mock_auth_jwt.assert_called_once_with('some_company')
+            mock_auth_api_key.assert_not_called()
+
+    def test_verify_with_jwt_failure(self):
+        with patch.object(self.iauth_service, '_authenticate_via_chat_jwt') as mock_auth_jwt, \
+             patch.object(self.iauth_service, '_authenticate_via_api_key') as mock_auth_api_key:
+            mock_auth_jwt.return_value = (None, None, "Error de JWT")
+
+            with self.app.test_request_context():
+                result = self.iauth_service.verify('some_company')
+
+            assert result == {"error_message": "Fallo de autenticación JWT"}
+            mock_auth_api_key.assert_not_called()
+
+    def test_verify_with_api_key_success(self):
+        with patch.object(self.iauth_service, '_authenticate_via_chat_jwt') as mock_auth_jwt, \
+             patch.object(self.iauth_service, '_authenticate_via_api_key') as mock_auth_api_key, \
+             patch.object(self.iauth_service, 'check_if_user_is_logged_in') as mock_check_if_logged_in:
+            mock_auth_jwt.return_value = (None, None, None)
+            mock_auth_api_key.return_value = (456, None)
+
+            with self.app.test_request_context():
+                result = self.iauth_service.verify('some_company', body_external_user_id='body_user_2')
+
+            assert result['success'] is True
+            assert result['company_id'] == 456
+            assert result['external_user_id'] == 'body_user_2'
+            mock_auth_api_key.assert_called_once_with('some_company')
+            mock_check_if_logged_in.assert_not_called()
+
+    def test_verify_with_api_key_failure(self):
+        with patch.object(self.iauth_service, '_authenticate_via_chat_jwt') as mock_auth_jwt, \
+             patch.object(self.iauth_service, '_authenticate_via_api_key') as mock_auth_api_key:
+            mock_auth_jwt.return_value = (None, None, None)
+            mock_auth_api_key.return_value = (None, "Error de API Key")
+
+            with self.app.test_request_context():
+                result = self.iauth_service.verify('some_company')
+
+            assert result == {"error_message": "Fallo de autenticación API Key"}
+
+    def test_verify_with_session_success(self):
+        with patch.object(self.iauth_service, '_authenticate_via_chat_jwt') as mock_auth_jwt, \
+             patch.object(self.iauth_service, '_authenticate_via_api_key') as mock_auth_api_key, \
+             patch.object(self.iauth_service, 'check_if_user_is_logged_in') as mock_check_if_logged_in, \
+             patch('iatoolkit.common.auth.SessionManager.get') as mock_session_get:
+            mock_auth_jwt.return_value = (None, None, None)
+            mock_auth_api_key.return_value = (None, None)
+            mock_check_if_logged_in.return_value = None
+            mock_session_get.side_effect = lambda key, default=None: {'user_id': 789, 'company_id': 999}.get(key,
+                                                                                                              default)
+
+            with self.app.test_request_context():
+                result = self.iauth_service.verify('some_company')
+
+            assert result['success'] is True
+            assert result['company_id'] == 999
+            assert result['local_user_id'] == 789
+            mock_check_if_logged_in.assert_called_once_with('some_company')
+
+    def test_verify_with_session_incomplete_data(self):
+        with patch.object(self.iauth_service, '_authenticate_via_chat_jwt') as mock_auth_jwt, \
+             patch.object(self.iauth_service, '_authenticate_via_api_key') as mock_auth_api_key, \
+             patch.object(self.iauth_service, 'check_if_user_is_logged_in') as mock_check_if_logged_in, \
+             patch('iatoolkit.common.auth.SessionManager.get') as mock_session_get:
+            mock_auth_jwt.return_value = (None, None, None)
+            mock_auth_api_key.return_value = (None, None)
+            mock_check_if_logged_in.return_value = None
+            mock_session_get.side_effect = lambda key, default=None: {'user_id': 789}.get(key, default)
+
+            with self.app.test_request_context():
+                result = self.iauth_service.verify('some_company')
+
+            assert result == {"error_message": "Fallo interno en la autenticación o no autenticado"}
+
+    def test_verify_with_session_check_fails_by_exception(self):
+        with patch.object(self.iauth_service, '_authenticate_via_chat_jwt') as mock_auth_jwt, \
+             patch.object(self.iauth_service, '_authenticate_via_api_key') as mock_auth_api_key, \
+             patch.object(self.iauth_service, 'check_if_user_is_logged_in') as mock_check_if_logged_in:
+            mock_auth_jwt.return_value = (None, None, None)
+            mock_auth_api_key.return_value = (None, None)
+            mock_check_if_logged_in.side_effect = HTTPException
+
+            with self.app.test_request_context(), pytest.raises(HTTPException):
+                self.iauth_service.verify('some_company')
+
+    # --- Nuevas pruebas para _authenticate_via_api_key ---
+
+    def test_authenticate_via_api_key_success(self):
+        with self.app.test_request_context(headers={'Authorization': 'Bearer valid_key'}):
+            mock_api_entry = MagicMock()
+            mock_api_entry.company.short_name = 'test_company'
+            mock_api_entry.company_id = 123
+            self.profile_repo.get_active_api_key_entry.return_value = mock_api_entry
+
+            company_id, error = self.iauth_service._authenticate_via_api_key('test_company')
+
+            assert company_id == 123
+            assert error is None
+            self.profile_repo.get_active_api_key_entry.assert_called_once_with('valid_key')
+
+    def test_authenticate_via_api_key_no_header(self):
+        with self.app.test_request_context(headers={}):
+            company_id, error = self.iauth_service._authenticate_via_api_key('test_company')
+            assert company_id is None
+            assert error is None
+            self.profile_repo.get_active_api_key_entry.assert_not_called()
+
+    def test_authenticate_via_api_key_wrong_scheme(self):
+        with self.app.test_request_context(headers={'Authorization': 'Basic some_token'}):
+            company_id, error = self.iauth_service._authenticate_via_api_key('test_company')
+            assert company_id is None
+            assert error is None
+            self.profile_repo.get_active_api_key_entry.assert_not_called()
+
+    def test_authenticate_via_api_key_inactive_key(self):
+        with self.app.test_request_context(headers={'Authorization': 'Bearer inactive_key'}):
+            self.profile_repo.get_active_api_key_entry.return_value = None
+            company_id, error = self.iauth_service._authenticate_via_api_key('test_company')
+            assert company_id is None
+            assert error == "API Key inválida o inactiva"
+
+    def test_authenticate_via_api_key_company_mismatch(self):
+        with self.app.test_request_context(headers={'Authorization': 'Bearer valid_key'}):
+            mock_api_entry = MagicMock()
+            mock_api_entry.company.short_name = 'other_company'
+            self.profile_repo.get_active_api_key_entry.return_value = mock_api_entry
+
+            company_id, error = self.iauth_service._authenticate_via_api_key('test_company')
+            assert company_id is None
+            assert error == "API Key no es válida para la compañía test_company"
+
+    def test_authenticate_via_api_key_repo_exception(self):
+        with self.app.test_request_context(headers={'Authorization': 'Bearer any_key'}):
+            self.profile_repo.get_active_api_key_entry.side_effect = Exception("DB error")
+            company_id, error = self.iauth_service._authenticate_via_api_key('test_company')
+            assert company_id is None
+            assert error == "Error interno del servidor al validar API Key"
+
+    # --- Nuevas pruebas para _authenticate_via_chat_jwt ---
+
+    def test_authenticate_via_chat_jwt_success(self):
+        with self.app.test_request_context(headers={'X-Chat-Token': 'valid_jwt'}):
+            self.jwt_service.validate_chat_jwt.return_value = {'company_id': 123, 'external_user_id': 'ext_user'}
+            company_id, external_user_id, error = self.iauth_service._authenticate_via_chat_jwt('test_company')
+            assert company_id == 123
+            assert external_user_id == 'ext_user'
+            assert error is None
+            self.jwt_service.validate_chat_jwt.assert_called_once_with('valid_jwt', 'test_company')
+
+    def test_authenticate_via_chat_jwt_no_header(self):
+        with self.app.test_request_context(headers={}):
+            company_id, external_user_id, error = self.iauth_service._authenticate_via_chat_jwt('test_company')
+            assert company_id is None
+            assert external_user_id is None
+            assert error is None
+            self.jwt_service.validate_chat_jwt.assert_not_called()
+
+    def test_authenticate_via_chat_jwt_validation_fails(self):
+        with self.app.test_request_context(headers={'X-Chat-Token': 'invalid_jwt'}):
+            self.jwt_service.validate_chat_jwt.return_value = None
+            company_id, external_user_id, error = self.iauth_service._authenticate_via_chat_jwt('test_company')
+            assert company_id is None
+            assert external_user_id is None
+            assert error == "Token de chat expirado, debes reingresar al chat"
+
+    def test_authenticate_via_chat_jwt_incomplete_payload_no_company_id(self):
+        with self.app.test_request_context(headers={'X-Chat-Token': 'valid_jwt'}):
+            self.jwt_service.validate_chat_jwt.return_value = {'external_user_id': 'ext_user'}
+            company_id, external_user_id, error = self.iauth_service._authenticate_via_chat_jwt('test_company')
+            assert company_id is None
+            assert external_user_id is None
+            assert error == "Token de chat con formato interno incorrecto"
+
+    def test_authenticate_via_chat_jwt_incomplete_payload_no_external_id(self):
+        with self.app.test_request_context(headers={'X-Chat-Token': 'valid_jwt'}):
+            self.jwt_service.validate_chat_jwt.return_value = {'company_id': 123}
+            company_id, external_user_id, error = self.iauth_service._authenticate_via_chat_jwt('test_company')
+            assert company_id is None
+            assert external_user_id is None
+            assert error == "Token de chat con formato interno incorrecto"

iatoolkit-0.7.12/src/tests/common/test_routes.py

@@ -0,0 +1,42 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import Flask
+from unittest.mock import patch
+
+
+class TestRoutes:
+    def setup_method(self):
+        # Configurar la aplicación Flask para pruebas
+        self.app = Flask(__name__)
+        self.app.config['VERSION'] = "1.0.0"
+        self.app.secret_key = 'test_secret'
+
+        # Concentramos todos los mocks y patches aquí:
+
+        # Patch para el SessionManager (por ejemplo para evitar lecturas de sesión reales)
+        self.session_manager_patch = patch("common.routes.SessionManager")
+        self.mock_session_manager = self.session_manager_patch.start()
+        # Para este ejemplo, no es necesario configurar un return_value, pero se podría definir según se requiera
+
+        # Patch para la función flash, reemplazándola para que no intente enviar mensajes reales
+        self.flash_patch = patch("common.routes.flash")
+        self.mock_flash = self.flash_patch.start()
+
+        # Patch para render_template, que es usado por la ruta "/about" y otros
+        self.render_template_patch = patch("common.routes.render_template", return_value="<html>About</html>")
+        self.mock_render_template = self.render_template_patch.start()
+
+        # Registrar las rutas en la aplicación
+        from iatoolkit.common.routes import register_routes
+        register_routes(self.app)
+
+        # Crear el cliente de pruebas
+        self.client = self.app.test_client()
+
+    def teardown_method(self, method):
+        patch.stopall()
+
+

iatoolkit-0.7.12/src/tests/common/test_session_manager.py

@@ -0,0 +1,59 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+import unittest
+from unittest.mock import patch
+from iatoolkit.common.session_manager import SessionManager
+
+
+class TestSessionManager(unittest.TestCase):
+
+    def setUp(self):
+        """
+        Configura los patches y mocks comunes para las pruebas.
+        """
+        # Parchear el objeto 'session' como un diccionario en todos los tests
+        self.session_patcher = patch("iatoolkit.common.session_manager.session", new_callable=dict)
+        self.mock_session = self.session_patcher.start()  # Iniciar el patch y obtener el mock
+
+    def tearDown(self):
+        """
+        Detiene todos los patches después de cada prueba.
+        """
+        patch.stopall()  # Detener cualquier parche activo y limpiar el entorno
+
+    def test_set(self):
+        """Prueba que el método set almacena un valor en la sesión."""
+        SessionManager.set("key", "value")
+        self.assertEqual(self.mock_session["key"], "value")
+
+    def test_get_existing_key(self):
+        """Prueba que el método get devuelve el valor correcto si la clave existe."""
+        self.mock_session["key"] = "value"
+        result = SessionManager.get("key")
+        self.assertEqual(result, "value")
+
+    def test_get_non_existing_key_with_default(self):
+        """Prueba que el método get devuelve el valor predeterminado si la clave no existe."""
+        result = SessionManager.get("non_existing_key", default="default_value")
+        self.assertEqual(result, "default_value")
+
+    def test_remove_existing_key(self):
+        """Prueba que el método remove elimina correctamente una clave existente."""
+        self.mock_session["key"] = "value"
+        SessionManager.remove("key")
+        self.assertNotIn("key", self.mock_session)
+
+    def test_remove_non_existing_key(self):
+        """Prueba que el método remove no lanza errores si la clave no existe."""
+        SessionManager.remove("non_existing_key")
+        # No debería hacer nada, y la sesión debe permanecer vacía
+        self.assertEqual(len(self.mock_session), 0)
+
+    def test_clear(self):
+        """Prueba que el método clear elimina todos los elementos de la sesión."""
+        self.mock_session.update({"key1": "value1", "key2": "value2"})
+        SessionManager.clear()
+        self.assertEqual(len(self.mock_session), 0)