First Commit

test_fusohorario.py

@@ -0,0 +1,187 @@
+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"
+
+import monica_agent as ma
+from monica_agent import (
+    fusohorario_command,
+    message_handler,
+    load_user_profile,
+    save_user_profile,
+    get_now,
+)
+
+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_fusohorario_command_starts():
+    print("=== Testando /fusohorario inicia fluxo ===")
+    ma.is_setting_timezone = False
+    ma.timezone_step = 0
+    update = FakeUpdate("/fusohorario")
+    context = FakeContext()
+    try:
+        await fusohorario_command(update, context)
+        assert ma.is_setting_timezone is True
+        assert ma.timezone_step == 1
+        print("✅ /fusohorario iniciou o fluxo corretamente")
+    except Exception as e:
+        print(f"❌ Erro: {e}")
+        import traceback
+        traceback.print_exc()
+
+async def test_fusohorario_valid_timezone():
+    print("\n=== Testando definicao de fuso horario valido ===")
+    profile = load_user_profile()
+    profile["timezone"] = ""
+    save_user_profile(profile)
+    ma.is_setting_timezone = False
+    ma.timezone_step = 0
+    update = FakeUpdate("/fusohorario")
+    context = FakeContext()
+    await fusohorario_command(update, context)
+
+    update = FakeUpdate("America/Sao_Paulo")
+    context = FakeContext()
+    await message_handler(update, context)
+    assert ma.is_setting_timezone is False
+    assert ma.timezone_step == 0
+
+    profile = load_user_profile()
+    assert profile.get("timezone") == "America/Sao_Paulo"
+    print(f"✅ Fuso horario salvo: {profile.get('timezone')}")
+
+    now = get_now()
+    assert now.tzinfo is not None
+    print(f"✅ get_now() retorna datetime com timezone: {now}")
+
+async def test_fusohorario_invalid_timezone():
+    print("\n=== Testando fuso horario invalido ===")
+    profile = load_user_profile()
+    profile["timezone"] = ""
+    save_user_profile(profile)
+    ma.is_setting_timezone = False
+    ma.timezone_step = 0
+    update = FakeUpdate("/fusohorario")
+    context = FakeContext()
+    await fusohorario_command(update, context)
+
+    update = FakeUpdate("Fuso/Invalido")
+    context = FakeContext()
+    await message_handler(update, context)
+    assert ma.is_setting_timezone is True
+
+    profile = load_user_profile()
+    assert profile.get("timezone") == ""
+    print("✅ Fuso horario invalido rejeitado e fluxo mantido")
+
+async def test_fusohorario_cancelar():
+    print("\n=== Testando cancelar durante /fusohorario ===")
+    profile = load_user_profile()
+    profile["timezone"] = ""
+    save_user_profile(profile)
+    ma.is_setting_timezone = False
+    ma.timezone_step = 0
+    update = FakeUpdate("/fusohorario")
+    context = FakeContext()
+    await fusohorario_command(update, context)
+
+    update = FakeUpdate("cancelar")
+    context = FakeContext()
+    await message_handler(update, context)
+    assert ma.is_setting_timezone is False
+    assert ma.timezone_step == 0
+    print("✅ Cancelar encerrou o fluxo de fuso horario")
+
+async def test_fusohorario_affects_time():
+    print("\n=== Testando que get_now usa o fuso horario ===")
+    profile = load_user_profile()
+    profile["timezone"] = "UTC"
+    save_user_profile(profile)
+
+    now = get_now()
+    assert now.tzinfo is not None
+    print(f"✅ get_now() com UTC: {now}")
+
+    profile["timezone"] = "America/Sao_Paulo"
+    save_user_profile(profile)
+    now_sp = get_now()
+    assert now_sp.tzinfo is not None
+    print(f"✅ get_now() com America/Sao_Paulo: {now_sp}")
+
+    profile["timezone"] = ""
+    save_user_profile(profile)
+    now_naive = get_now()
+    assert now_naive.tzinfo is None
+    print(f"✅ get_now() sem timezone retorna naive: {now_naive}")
+
+async def test_fusohorario_in_help():
+    print("\n=== Verificando /help contem /fusohorario ===")
+    from monica_agent import help_command
+    update = FakeUpdate("/help")
+    context = FakeContext()
+    responses = []
+
+    original_reply = FakeMessage.reply_text
+    async def capture_reply(self, text, **kwargs):
+        responses.append(text)
+        print(f"BOT RESPONSE: {text[:300]}")
+
+    FakeMessage.reply_text = capture_reply
+    try:
+        await help_command(update, context)
+        help_text = responses[0] if responses else ""
+        if "/fusohorario" in help_text:
+            print("✅ /help menciona /fusohorario")
+        else:
+            print("❌ /help nao menciona /fusohorario")
+            print(f"   Help text: {help_text[:200]}")
+    finally:
+        FakeMessage.reply_text = original_reply
+
+async def main():
+    await test_fusohorario_command_starts()
+    await test_fusohorario_valid_timezone()
+    await test_fusohorario_invalid_timezone()
+    await test_fusohorario_cancelar()
+    await test_fusohorario_affects_time()
+    await test_fusohorario_in_help()
+    print("\n=== Todos os testes de fusohorario concluidos ===")
+
+if __name__ == "__main__":
+    asyncio.run(main())