Add tests and fix research tools

1 files changed, 102 insertions, 0 deletions

diff --git a/tests/test_main.py b/tests/test_main.py
new file mode 100644
index 0000000..0d7e051
--- /dev/null
+++ b/tests/test_main.py
@@ -0,0 +1,102 @@
+import unittest
+from unittest.mock import Mock, patch
+
+import main
+
+
+class SlopbotTests(unittest.TestCase):
+    def setUp(self):
+        self.old_client = main.client
+        self.old_executor = main.executor
+
+    def tearDown(self):
+        main.client = self.old_client
+        main.executor = self.old_executor
+
+    def stream_msg(self, content, **overrides):
+        msg = {
+            "id": 123,
+            "type": "stream",
+            "sender_email": "alice@example.com",
+            "sender_full_name": "Alice",
+            "display_recipient": "test stream",
+            "stream_id": 23,
+            "subject": "slop",
+            "content": content,
+        }
+        msg.update(overrides)
+        return msg
+
+    def test_ticker_regex_deduplicates_in_handle_message(self):
+        sent = []
+        main.client = Mock(email="slop-bot@example.com")
+        main.client.send_message.side_effect = sent.append
+
+        with patch.object(main, "fetch_price", side_effect=lambda ticker: f"${ticker}: $1.00") as fetch:
+            main.handle_message(self.stream_msg("$NVDA $AAPL $NVDA"))
+
+        self.assertEqual([call.args[0] for call in fetch.call_args_list], ["NVDA", "AAPL"])
+        self.assertEqual(len(sent), 1)
+        self.assertEqual(sent[0]["type"], "stream")
+        self.assertEqual(sent[0]["to"], 23)
+        self.assertEqual(sent[0]["topic"], "slop")
+        self.assertEqual(sent[0]["content"], "$NVDA: $1.00\n$AAPL: $1.00")
+
+    def test_mention_calls_agent_and_appends_prices(self):
+        sent = []
+        main.client = Mock(email="slop-bot@example.com")
+        main.client.send_message.side_effect = sent.append
+
+        with patch.object(main, "ask_agent", return_value="agent reply") as ask_agent, \
+             patch.object(main, "fetch_price", return_value="$NVDA: $1.00") as fetch_price:
+            main.handle_message(self.stream_msg("@**slop-bot** summarize $NVDA"))
+
+        ask_agent.assert_called_once()
+        self.assertEqual(ask_agent.call_args.args[1], "summarize $NVDA")
+        fetch_price.assert_called_once_with("NVDA")
+        self.assertEqual(sent[0]["content"], "agent reply\n\n$NVDA: $1.00")
+
+    def test_ignores_self_and_irrelevant_messages(self):
+        main.client = Mock(email="slop-bot@example.com")
+        main.handle_message(self.stream_msg("$NVDA", sender_email="slop-bot@example.com"))
+        main.handle_message(self.stream_msg("nothing to do here"))
+        main.client.send_message.assert_not_called()
+
+    def test_private_reply_goes_to_sender_email(self):
+        sent = []
+        main.client = Mock(email="slop-bot@example.com")
+        main.client.send_message.side_effect = sent.append
+        msg = self.stream_msg("$NVDA", type="private", sender_email="alice@example.com")
+
+        with patch.object(main, "fetch_price", return_value="$NVDA: $1.00"):
+            main.handle_message(msg)
+
+        self.assertEqual(sent[0]["type"], "private")
+        self.assertEqual(sent[0]["to"], "alice@example.com")
+
+    def test_handle_event_submits_message_work(self):
+        submitted = []
+        main.executor = Mock()
+        main.executor.submit.side_effect = lambda fn, msg: submitted.append((fn, msg))
+        msg = self.stream_msg("$NVDA")
+
+        main.handle_event({"type": "message", "message": msg})
+        main.handle_event({"type": "heartbeat"})
+
+        self.assertEqual(submitted, [(main.handle_message, msg)])
+
+    def test_agent_debug_streams_stderr_to_journal(self):
+        with patch.object(main.subprocess, "run") as run:
+            run.return_value = Mock(returncode=0, stdout="ok\n", stderr=None)
+            with patch.object(main, "AGENT_DEBUG", True):
+                rc, out, err = main._agent("hello")
+
+        self.assertEqual((rc, out, err), (0, "ok", ""))
+        kwargs = run.call_args.kwargs
+        command = run.call_args.args[0]
+        self.assertIn("--debug", command)
+        self.assertIsNone(kwargs["stderr"])
+
+
+if __name__ == "__main__":
+    unittest.main()