iatoolkit 0.7.11__py3-none-any.whl → 0.7.14__py3-none-any.whl

iatoolkit/cli_commands.py

@@ -6,7 +6,7 @@
 import click
 import logging
 from iatoolkit import IAToolkit
-from iatoolkit.services import ProfileService
+from iatoolkit.services.profile_service import ProfileService
 
 def register_core_commands(app):
     """Registra los comandos CLI del núcleo de IAToolkit."""

iatoolkit/iatoolkit.py

@@ -264,15 +264,15 @@ class IAToolkit:
         binder.bind(TaskRepo, to=TaskRepo)
 
     def _bind_services(self, binder: Binder):
-        from iatoolkit.services import QueryService
+        from iatoolkit.services.query_service import QueryService
         from iatoolkit.services.tasks_service import TaskService
-        from iatoolkit.services import BenchmarkService
+        from iatoolkit.services.benchmark_service import BenchmarkService
         from iatoolkit.services.document_service import DocumentService
         from iatoolkit.services.prompt_manager_service import PromptService
         from iatoolkit.services.excel_service import ExcelService
-        from iatoolkit.services import MailService
+        from iatoolkit.services.mail_service import MailService
         from iatoolkit.services.load_documents_service import LoadDocumentsService
-        from iatoolkit.services import ProfileService
+        from iatoolkit.services.profile_service import ProfileService
         from iatoolkit.services.jwt_service import JWTService
         from iatoolkit.services.dispatcher_service import Dispatcher
 
@@ -290,10 +290,10 @@ class IAToolkit:
 
     def _bind_infrastructure(self, binder: Binder):
         from iatoolkit.infra.llm_client import llmClient
-        from iatoolkit.infra import LLMProxy
-        from iatoolkit.infra import GoogleChatApp
+        from iatoolkit.infra.llm_proxy import LLMProxy
+        from iatoolkit.infra.google_chat_app import GoogleChatApp
         from iatoolkit.infra.mail_app import MailApp
-        from iatoolkit.common import IAuthentication
+        from iatoolkit.common.auth import IAuthentication
         from iatoolkit.common.util import Utility
 
 
@@ -307,10 +307,10 @@ class IAToolkit:
 
     def _bind_views(self, binder: Binder):
         """Vincula las vistas después de que el injector ha sido creado"""
-        from iatoolkit.views import LLMQueryView
-        from iatoolkit.views import HomeView
-        from iatoolkit.views import ChatView
-        from iatoolkit.views import ChangePasswordView
+        from iatoolkit.views.llmquery_view import LLMQueryView
+        from iatoolkit.views.home_view import HomeView
+        from iatoolkit.views.chat_view import ChatView
+        from iatoolkit.views.change_password_view import ChangePasswordView
 
         binder.bind(HomeView, to=HomeView)
         binder.bind(ChatView, to=ChatView)
@@ -355,7 +355,7 @@ class IAToolkit:
         # Configura context processors para templates
         @self.app.context_processor
         def inject_globals():
-            from iatoolkit.common import SessionManager
+            from iatoolkit.common.session_manager import SessionManager
             return {
                 'url_for': url_for,
                 'iatoolkit_version': self.version,
@@ -367,16 +367,14 @@ class IAToolkit:
     def _get_default_static_folder(self) -> str:
         try:
             current_dir = os.path.dirname(os.path.abspath(__file__))  # .../src/iatoolkit
-            src_dir = os.path.dirname(current_dir)  # .../src
-            return os.path.join(src_dir, "static")
+            return os.path.join(current_dir, "static")
         except:
             return 'static'
 
     def _get_default_template_folder(self) -> str:
         try:
             current_dir = os.path.dirname(os.path.abspath(__file__))  # .../src/iatoolkit
-            src_dir = os.path.dirname(current_dir)  # .../src
-            return os.path.join(src_dir, "templates")
+            return os.path.join(current_dir, "templates")
         except:
             return 'templates'
 

{iatoolkit-0.7.11.dist-info → iatoolkit-0.7.14.dist-info}/METADATA

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

{iatoolkit-0.7.11.dist-info → iatoolkit-0.7.14.dist-info}/RECORD

@@ -1,8 +1,8 @@
 iatoolkit/__init__.py,sha256=yi7Uw68KajDWUUo-v4xV31Nt7hjr_-2gCXbkhdiDl44,1887
 iatoolkit/base_company.py,sha256=gM3X7xedNaVLodOObvjUN9eJKPO5-HmSLaBFa9jowCk,4277
-iatoolkit/cli_commands.py,sha256=lMOz5ZCuYhkIXde86Gg0cxm530lPZ7H8D1qpYj5jPPI,2286
+iatoolkit/cli_commands.py,sha256=G5L9xQXZ0lVFXQWBaE_KEZHyfuiT6PL1nTQRoSdnBzc,2302
 iatoolkit/company_registry.py,sha256=tduqt3oV8iDX_IB1eA7KIgvIxE4edTcy-3qZIXh3Lzw,2549
-iatoolkit/iatoolkit.py,sha256=bdw5VrMyfYWGeFGLxDR4ZJYyMJQwEXNW6BUUZxTTc3w,15841
+iatoolkit/iatoolkit.py,sha256=UTixRTy0HoOMfO01H58IhC6BJFfZFZU_cbyOA25gkmM,15888
 iatoolkit/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 iatoolkit/common/auth.py,sha256=8NH6MQXfddLQd1GxrO2op3IYrrP4SMQKoKzj1o1jZmc,8486
 iatoolkit/common/exceptions.py,sha256=EXx40n5htp7UiOM6P1xfJ9U6NMcADqm62dlFaKz7ICU,1154
@@ -122,7 +122,73 @@ iatoolkit/views/tasks_review_view.py,sha256=keLsLCyOTTlcoIapnB_lbuSvLwrPVZVpBiFC
 iatoolkit/views/tasks_view.py,sha256=a3anTXrJTTvbQuc6PSpOzidLKQFL4hWa7PI2Cppcz8w,4110
 iatoolkit/views/user_feedback_view.py,sha256=G37zmP8P4LvZrSymNJ5iFXhLZg1A3BEwRfTpH1Iam5w,2652
 iatoolkit/views/verify_user_view.py,sha256=a3q4wHJ8mKAEmgbNTOcnX4rMikROjOR3mHvCr30qGGA,2351
-iatoolkit-0.7.11.dist-info/METADATA,sha256=aW2_npSGL4nUTH8Iq5RnAvKBmT38Xa9RlPO5cngT80Y,9301
-iatoolkit-0.7.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.7.11.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
-iatoolkit-0.7.11.dist-info/RECORD,,
+tests/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
+tests/conftest.py,sha256=S000Ry-rWczEDsndTtOvsj2zf2EYgjHi78oVn60hoOU,1094
+tests/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/common/test_auth.py,sha256=WViWPG3E8OldgZyPbl-lp5rosjgPQb9MlADGLr1rq4k,14338
+tests/common/test_routes.py,sha256=_Fz8mYqxp8YZ39CSng0CQOIRs9fOfgC42iU9XiHLMzU,1560
+tests/common/test_session_manager.py,sha256=8FLa8Z7ueLYNnJeJyTgCm8pp1DY6QvC0b0ZxZC2vpKc,2388
+tests/common/test_util.py,sha256=SMh0pSIv8K-bCeMZeYBbCD70o6MpOUXHQH6kZFLLN5c,20653
+tests/companies/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
+tests/infra/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
+tests/infra/test_call_service.py,sha256=p5kFqXFojSmHucLTyika2u3wvpQwJRCrAlVOPeohoVE,3747
+tests/infra/test_database_manager.py,sha256=SMU_pcCsW89b6GyykAU2LEO8UaHL9gO5rdeWGFP3A6g,2503
+tests/infra/test_gemini_adapter.py,sha256=gMjSgxh854sC08vYQpH6QSIX9w921U3g_Jrt_HU1i1Y,6292
+tests/infra/test_google_chat_app.py,sha256=D1iUaGy8Z-PHufJuQ607lGEUOFueedoziXR8J9b7qmM,2363
+tests/infra/test_llm_client.py,sha256=i5uRPy3k2csoNOg16dlVAaZq93MDTk5OxLgRPrsyDow,7619
+tests/infra/test_llm_proxy.py,sha256=Y9F8ozVsnPLOiIt_Minr_VvANEz0x0JIPvFaWJBCLgk,5753
+tests/infra/test_mail_app.py,sha256=6Mhj3HbnQ_sVaQc7wgTpIEFb2GAzc4kkFNWgPWBCmp0,3727
+tests/infra/test_openai_adapter.py,sha256=4CLLIOqQLaC_zpc3clyoA9fiMu5uJmfDoPqs7hEw4Qo,4319
+tests/infra/test_redis_session_manager_service.py,sha256=NbrUVxmEmWIZc9TtnFd1IiWwm1N0Ih0y7chP7z5Mx34,4914
+tests/infra/connectors/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
+tests/infra/connectors/test_google_drive_connector.py,sha256=UxbR64xiDpFyeE18--BHiKgvwEuaWt9b4Cf6bL6LreM,4434
+tests/infra/connectors/test_local_file_connector.py,sha256=-1wpjXt2DOn_NbSUerdGFd-g-NI3rWS5vDoL3PFqxGA,3459
+tests/infra/connectors/test_s3_connector.py,sha256=2hsJvKPyTWQj6kkNjtpRwsLNdvW3O5N5JlhzUM67x5Q,3512
+tests/repositories/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
+tests/repositories/test_database_manager.py,sha256=C6pqzxPCtk0DcZE7VNaFCT5ahONO7mWFKDo1bxb-9sU,3990
+tests/repositories/test_document_repo.py,sha256=sqNL5yVM_8phPaxcYR3FupuTWN5RNG8yLfeDxR07uIM,2543
+tests/repositories/test_llm_query_repo.py,sha256=Vdu85MI_L7Nh5ksNz9X3d4hORp9hXaXPwH1AUbuN5YE,13228
+tests/repositories/test_models.py,sha256=CJbz3U89BgPKmP61pwGdwDW8vHvPROKwYfqcg4lx-_Q,1627
+tests/repositories/test_profile_repo.py,sha256=Ex9R5JH2uyam8dt8Nl-1yiNYYj72Ma9A3Jlo_JkwktI,4981
+tests/repositories/test_tasks_repo.py,sha256=ZtOOjJTb_j8dKEpirZTQAnR9fhDDO7xQWZU710So9Bk,2769
+tests/repositories/test_vs_repo.py,sha256=PGN5PqQBtlwMC_UFU3qMSqxtoBSNz1YvZK4yYj0O0UU,4595
+tests/services/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
+tests/services/test_dispatcher_service.py,sha256=iALQx0I94SehG4eXqwvdswyYrPVShXFIlk-GwwbWyzw,13086
+tests/services/test_document_service.py,sha256=Z62VjjIn1EPgDT2_2jEVI6AQoO3yOdskc_FDqlvML1Q,8085
+tests/services/test_excel_service.py,sha256=7Gyy82MFqOVFIGL7RXMIOmal22ne-lsgAfY16azDAMM,8434
+tests/services/test_file_processor_service.py,sha256=XzdA581hoAhevp8atqgI7jxTY1gmHE7rnMllihRS9S0,4794
+tests/services/test_history_service.py,sha256=VF6IgogsgaFe64LNDeXMFcLrWjuGLf3Vj_srnFfkneM,7273
+tests/services/test_jwt_service.py,sha256=3TYtuPk1ps0aa5TKGvd30Q4F7Q1FEsqK6Gr1wjW8-q8,10325
+tests/services/test_load_documents_service.py,sha256=zDcHSXVQAdtAZCVWecTl-pYlkA4Sp8JLxKpu82LB69A,4640
+tests/services/test_mail_service.py,sha256=PPy5XA2lYFFBNdmU1Ea-cfqszwQZrrqZSZU1OQ26lDE,2295
+tests/services/test_profile_service.py,sha256=WzvVxeK5AYUDWJLBRB43vVDLHi64cIkPYPcynMEGZzM,14831
+tests/services/test_prompt_manager_service.py,sha256=B6j12RNjljAKZnd7vy1CNAdZxwKiTWD5l4uxOGKVuvE,9395
+tests/services/test_query_service.py,sha256=vq8dhteJOFiA0eGo1s5eBzasx7noZFrv96XdKFKhmS8,11331
+tests/services/test_search_service.py,sha256=n8C5mjDDCVX4NAJvS3mKgw-WsLWu5qUA8eTOxy-g9NU,1183
+tests/services/test_sql_service.py,sha256=bHDiPzjfM7-mbM4eEXlMVk3vcvefhlGCx3bdIrlriVM,7753
+tests/services/test_tasks_service.py,sha256=V_WcglSX4CAtOkkKhZKl7Q92LsODHLRu66TqydiSQP4,10598
+tests/services/test_user_feedback_service.py,sha256=HMU75j48jT-Chls78QEXoq92Hpq7TXP-m8xXIAL0rK0,15859
+tests/services/test_user_session_context_service.py,sha256=c49RqHuoY6W_u-eR-UR05SNwDfqvRpRQWAJ6Ghyh7sQ,6799
+tests/views/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
+tests/views/test_change_password_view.py,sha256=5dIjx6_-rNHqj9yrkz00ajMxmZENBoTwN0gNFGUzxRE,9213
+tests/views/test_chat_token_request_view.py,sha256=jwgGaNmBPTK9irKTqsDTTTIZawpYBxfGXBUv0xxvCQ0,9850
+tests/views/test_chat_view.py,sha256=24dySNWlpu7EXkIeFJwUUXwb5xAq3Kdv2IVzPSVSWuY,3798
+tests/views/test_download_file_view.py,sha256=l0o5DnI29ByV7N5kK4UoTKhPmStKuiAOZKFPxCSZszE,5969
+tests/views/test_external_chat_login_view.py,sha256=1PLOPqVJjR-7d1DV24uuA7QOYEeKsIusjw6poBYTFB8,5405
+tests/views/test_external_login_view.py,sha256=Lf4VHxTyddx_RSTvYoxcZr8cG82bvJ0pdVqU1CJpwcg,4394
+tests/views/test_file_store_view.py,sha256=hT2k5-ppGNphKnQwueTvyzHQUGJ1umVXpPNOpTL8GOI,4837
+tests/views/test_forgot_password_view.py,sha256=GeaGMQpY20T9_cM_kBTrwRcx-Z1zsDbytB-KoyM9p2o,6224
+tests/views/test_history_view.py,sha256=AxNNjVEEYVdLDNa6aZaWdSE3FMquiqqDGxvl57UYdD8,12322
+tests/views/test_home_view.py,sha256=uqPCQnDyTm25l6YLrwzINrHMd16oyY3yrqXui1LChjg,2438
+tests/views/test_llm_query_view.py,sha256=3Mf_mfZNW9wTG7BivwEW0OL-QXplV2zRakl5rIUOvu8,5121
+tests/views/test_login_view.py,sha256=P-6n6LGnQ87pC9a48FMz0okVm7eukpSeIKDnMUABs3I,4812
+tests/views/test_prompt_view.py,sha256=y8mIXc0AqOoSe2QJJX3UdiFgA1a2TYFCcIwGiupxR0U,4563
+tests/views/test_signup_view.py,sha256=WVh11rzH2MbJWS9rvYIn9R1aCWJLnpIhIK_k7MQxi0I,6534
+tests/views/test_tasks_review_view.py,sha256=HXbAY99rbcdg-ZoaU3kxp-jNTPgOhgiTyKmV9rUcFyQ,3743
+tests/views/test_tasks_view.py,sha256=TNdV69Vt17uc-vBI37A7qtRNqyPho2MoW7cgGxQKecU,4551
+tests/views/test_user_feedback_view.py,sha256=IfKOpsncmf5ciMDsGuPtro9GdKXkxhRzsysVlEEM_HA,9181
+tests/views/test_verify_user_view.py,sha256=dxQ3ibOETEuP8M50Gmwbmzj2L6UU7PAyC91Wp9Xs0K0,4772
+iatoolkit-0.7.14.dist-info/METADATA,sha256=dKOyYj3Eqor1XBk1Q51M-CpSbTFnEnaXDsKvE_imzho,9301
+iatoolkit-0.7.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-0.7.14.dist-info/top_level.txt,sha256=or0Ar3Su6BhTy86zRrUwMAWtsR8Nk-tFEwdC0CZpKCs,16
+iatoolkit-0.7.14.dist-info/RECORD,,

{iatoolkit-0.7.11.dist-info → iatoolkit-0.7.14.dist-info}/top_level.txt

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

tests/__init__.py

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

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"

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()
+
+

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)