iatoolkit 0.8.0__tar.gz → 0.8.2__tar.gz

{iatoolkit-0.8.0 → iatoolkit-0.8.2}/PKG-INFO

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

{iatoolkit-0.8.0 → iatoolkit-0.8.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "iatoolkit"
-version = "0.8.0"
+version = "0.8.2"
 requires-python = ">=3.11"
 description = "IAToolkit"
 readme = "readme.md"
@@ -19,7 +19,7 @@ package-dir = {"" = "src"}
 
 [tool.setuptools.packages.find]
 where = ["src"]
-exclude = ["iatoolkit.tests*", "app.py"]
+exclude = ["tests*"]
 
 
 [tool.setuptools.dynamic]

{iatoolkit-0.8.0 → iatoolkit-0.8.2}/src/iatoolkit/iatoolkit.py

@@ -409,7 +409,7 @@ class IAToolkit:
 
     def _setup_download_dir(self):
         # 1. set the default download directory
-        default_download_dir = os.path.join(self.app.instance_path, 'downloads')
+        default_download_dir = os.path.join(os.getcwd(), 'downloads')
 
         # 3. if user specified one, use it
         download_dir = self.app.config.get('IATOOLKIT_DOWNLOAD_DIR', default_download_dir)

{iatoolkit-0.8.0 → iatoolkit-0.8.2}/src/iatoolkit.egg-info/PKG-INFO

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

{iatoolkit-0.8.0 → iatoolkit-0.8.2}/src/iatoolkit.egg-info/SOURCES.txt

@@ -110,70 +110,4 @@ 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/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
+src/iatoolkit/views/verify_user_view.py

{iatoolkit-0.8.0 → iatoolkit-0.8.2}/src/iatoolkit.egg-info/top_level.txt

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

iatoolkit-0.8.0/src/tests/__init__.py

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

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

@@ -1,279 +0,0 @@
-# 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.8.0/src/tests/common/test_routes.py

@@ -1,42 +0,0 @@
-# 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.8.0/src/tests/common/test_session_manager.py

@@ -1,59 +0,0 @@
-# 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)