First Commit

test_start_context.py

@@ -0,0 +1,109 @@
+import asyncio
+import sys
+import os
+sys.path.insert(0, "/home/marco/MonicaAgentIA/app")
+
+os.environ["TELEGRAM_BOT_TOKEN"] = "test"
+os.environ["TELEGRAM_CHAT_ID"] = "123"
+
+from unittest.mock import patch, AsyncMock
+from monica_agent import (
+    start_command,
+    load_user_profile,
+    save_user_profile,
+    load_all_contexts,
+    add_to_context,
+    ensure_contexts_dir,
+    AIProvider,
+    DEFAULT_MODEL,
+)
+from pathlib import Path
+
+class FakeMessage:
+    def __init__(self, text):
+        self.text = text
+        self.chat_id = 7889499475
+
+    async def reply_text(self, text, **kwargs):
+        print(f"BOT RESPONSE: {text[:500]}")
+
+class FakeUpdate:
+    def __init__(self, text):
+        self.message = FakeMessage(text)
+        self.effective_chat = self
+        self.effective_user = self
+        self.effective_message = self.message
+
+    @property
+    def id(self):
+        return self.message.chat_id
+
+    @property
+    def username(self):
+        return "test_user"
+
+    @property
+    def first_name(self):
+        return "Marco"
+
+class FakeContext:
+    def __init__(self):
+        self.bot = self
+
+    async def send_message(self, **kwargs):
+        print(f"BOT SEND: {kwargs.get('text', '')[:500]}")
+
+async def test_start_without_context():
+    print("=== Testando /start SEM contextos ===")
+    
+    profile = load_user_profile()
+    profile["name"] = "Marco"
+    save_user_profile(profile)
+    
+    from monica_agent import CONTEXTS_DIR
+    contexts_dir = Path(CONTEXTS_DIR)
+    for file_path in contexts_dir.glob("*.json"):
+        file_path.unlink()
+    
+    update = FakeUpdate("/start")
+    context = FakeContext()
+    
+    try:
+        await start_command(update, context)
+        print("✅ start_command executou sem erros")
+    except Exception as e:
+        print(f"❌ Erro: {e}")
+        import traceback
+        traceback.print_exc()
+
+async def test_start_with_context():
+    print("\n=== Testando /start COM contextos ===")
+    
+    profile = load_user_profile()
+    profile["name"] = "Marco"
+    save_user_profile(profile)
+    
+    add_to_context("user", "Olá Mônica!")
+    add_to_context("assistant", "Oi, Marco! Como você está?")
+    
+    def mock_chat(model, messages, temperature=0.8, max_tokens=2000):
+        return "Que bom ver você de novo! Continue a conversa baseado no contexto anterior."
+    
+    with patch.object(AIProvider, 'chat', mock_chat):
+        update = FakeUpdate("/start")
+        context = FakeContext()
+        try:
+            await start_command(update, context)
+            print("✅ start_command executou sem erros")
+        except Exception as e:
+            print(f"❌ Erro: {e}")
+            import traceback
+            traceback.print_exc()
+
+async def main():
+    await test_start_without_context()
+    await test_start_with_context()
+    print("\n=== Todos os testes concluídos ===")
+
+if __name__ == "__main__":
+    asyncio.run(main())