iatoolkit 0.8.1__py3-none-any.whl → 0.9.0__py3-none-any.whl

iatoolkit/__init__.py

@@ -4,37 +4,21 @@ IAToolkit Package
 
 # Expose main classes and functions at the top level of the package
 
-# Assuming 'toolkit.py' contains the IAToolkit class
-from .iatoolkit import IAToolkit, create_app
-from .iatoolkit import current_iatoolkit
+# main IAToolkit class
+from .iatoolkit import IAToolkit, current_iatoolkit, create_app
 
-# Assuming 'app_factory.py' contains create_app and register_company
+# for registering the client companies
 from .company_registry import register_company
-
-# Assuming 'base_company.py' contains BaseCompany
 from .base_company import BaseCompany
+from iatoolkit.repositories.database_manager import DatabaseManager
 
 # --- Services ---
-# Assuming they are in a 'services' sub-package
 from iatoolkit.services.sql_service import SqlService
-from iatoolkit.services.excel_service import ExcelService
-from iatoolkit.services.dispatcher_service import Dispatcher
 from iatoolkit.services.document_service import DocumentService
 from iatoolkit.services.search_service import SearchService
 from iatoolkit.services.load_documents_service import LoadDocumentsService
-from iatoolkit.repositories.profile_repo import ProfileRepo
-from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
-
-from iatoolkit.services.query_service import QueryService
-from iatoolkit.services.prompt_manager_service import PromptService
-from iatoolkit.repositories.database_manager import DatabaseManager
-
+from iatoolkit.services.excel_service import ExcelService
 from iatoolkit.infra.call_service import CallServiceClient
-from iatoolkit.common.util import Utility
-
-
-from iatoolkit.repositories.models import Base, Company, Function, TaskType, Prompt, PromptCategory
-
 
 __all__ = [
     'IAToolkit',
@@ -42,23 +26,11 @@ __all__ = [
     'current_iatoolkit',
     'register_company',
     'BaseCompany',
+    'DatabaseManager',
     'SqlService',
     'ExcelService',
-    'Dispatcher',
     'DocumentService',
     'SearchService',
-    'QueryService',
     'LoadDocumentsService',
-    'ProfileRepo',
-    'LLMQueryRepo',
-    'PromptService',
-    'DatabaseManager',
     'CallServiceClient',
-    'Utility',
-    'Company',
-    'Function',
-    'TaskType',
-    'Base',
-    'Prompt',
-    'PromptCategory'
 ]

{iatoolkit-0.8.1.dist-info → iatoolkit-0.9.0.dist-info}/METADATA

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

{iatoolkit-0.8.1.dist-info → iatoolkit-0.9.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-iatoolkit/__init__.py,sha256=yi7Uw68KajDWUUo-v4xV31Nt7hjr_-2gCXbkhdiDl44,1887
+iatoolkit/__init__.py,sha256=Ob7CfSEh2idRDKFnCGx4NPZhDZyIenHibhcTtCsq-68,1042
 iatoolkit/base_company.py,sha256=gM3X7xedNaVLodOObvjUN9eJKPO5-HmSLaBFa9jowCk,4277
 iatoolkit/cli_commands.py,sha256=G5L9xQXZ0lVFXQWBaE_KEZHyfuiT6PL1nTQRoSdnBzc,2302
 iatoolkit/company_registry.py,sha256=tduqt3oV8iDX_IB1eA7KIgvIxE4edTcy-3qZIXh3Lzw,2549
@@ -103,73 +103,7 @@ 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
-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.8.1.dist-info/METADATA,sha256=3jM0LUTEQLziEwLaP2Cao3WePdsFE_UEPxGY4mEbCKw,9300
-iatoolkit-0.8.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.8.1.dist-info/top_level.txt,sha256=or0Ar3Su6BhTy86zRrUwMAWtsR8Nk-tFEwdC0CZpKCs,16
-iatoolkit-0.8.1.dist-info/RECORD,,
+iatoolkit-0.9.0.dist-info/METADATA,sha256=ekZoeb9neMd9mTJjRlLpwS_LUf2txpZOtMDTJk1wANM,9300
+iatoolkit-0.9.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-0.9.0.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
+iatoolkit-0.9.0.dist-info/RECORD,,

{iatoolkit-0.8.1.dist-info → iatoolkit-0.9.0.dist-info}/top_level.txt

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

tests/__init__.py

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

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"

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

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)