major refactor (untested)

tests/test_telegram_bot.py

@@ -0,0 +1,62 @@
+import os
+import pytest
+from unittest.mock import MagicMock, patch
+from telegram_bot import TelegramBot
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
+
+# Explicitly pass token and chat ID to ensure they are set
+@pytest.fixture(autouse=True)
+def mock_env_vars():
+    os.environ["TELEGRAM_BOT_TOKEN"] = "test_token"
+    os.environ["TELEGRAM_CHAT_ID"] = "test_chat_id"
+
+@pytest.fixture
+def mock_monitor():
+    monitor = MagicMock()
+    monitor.load_state.return_value = {"autopilot": True}
+    monitor.load_applications.return_value = {
+        "app1": {"company": "CompanyA"},
+        "app2": {"company": "CompanyB"},
+        "app3": {"company": "CompanyA"},
+    }
+    return monitor
+
+@pytest.fixture
+def telegram_bot(mock_monitor):
+    return TelegramBot(mock_monitor, bot_token="test_token", chat_id="test_chat_id")
+
+@patch("telegram_bot.requests.post")
+def test_send_message(mock_post, telegram_bot):
+    mock_post.return_value.ok = True
+    telegram_bot._send_message("Test message")
+    mock_post.assert_called_once()
+    assert mock_post.call_args[1]["json"]["text"] == "Test message"
+
+@patch("telegram_bot.requests.post")
+def test_send_photo(mock_post, telegram_bot):
+    mock_post.return_value.ok = True
+    with patch("builtins.open", create=True):
+        telegram_bot._send_photo("/path/to/photo.jpg", "Test caption")
+    mock_post.assert_called_once()
+    assert mock_post.call_args[1]["data"]["caption"] == "Test caption"
+
+@patch("telegram_bot.TelegramBot._send_message")
+def test_handle_status_command(mock_send_message, telegram_bot):
+    telegram_bot._handle_status_command()
+    mock_send_message.assert_called_once()
+    assert "Autopilot" in mock_send_message.call_args[0][0]
+
+@patch("telegram_bot.TelegramBot._send_message")
+def test_handle_help_command(mock_send_message, telegram_bot):
+    telegram_bot._handle_help_command()
+    mock_send_message.assert_called_once()
+    assert "InBerlin Monitor Commands" in mock_send_message.call_args[0][0]
+
+@patch("telegram_bot.TelegramBot._send_message")
+def test_handle_unknown_command(mock_send_message, telegram_bot):
+    telegram_bot._handle_unknown_command("/unknown")
+    mock_send_message.assert_called_once()
+    assert "Unknown command" in mock_send_message.call_args[0][0]