stuff, fmt

tests/test_app.py

@@ -1,47 +1,31 @@
 import contextlib
-import importlib
 import io
-import json
 import re
-import subprocess
-import sys
 import unittest
 from unittest.mock import Mock, patch
 
+from ipython_demo import utils
 from ipython_demo.app import App
 from ipython_demo.models import CallOptions, CallResponse
 
 
 class AppTests(unittest.TestCase):
     def test_generate_good_names_uses_readable_random_words(self):
-        app_module = importlib.import_module("ipython_demo.app")
         generator = Mock()
+        assert isinstance(generator.word, Mock)
         generator.word.side_effect = ["luminous", "otter"]
 
         with patch.object(
-            app_module,
+            utils,
             "_WORD_GENERATOR",
             generator,
             create=True,
         ):
-            name = app_module.generate_good_names()
+            name = utils.generate_good_names()
 
         self.assertRegex(name, re.compile(r"^[a-z]+-[a-z]+-[0-9a-f]{4}$"))
-        generator.word.assert_any_call(include_parts_of_speech=["adjectives"])
-        generator.word.assert_any_call(include_parts_of_speech=["nouns"])
-
-    def test_main_emits_one_json_payload(self):
-        completed = subprocess.run(
-            [sys.executable, "-c", "from ipython_demo.app import main; main()"],
-            capture_output=True,
-            text=True,
-            check=True,
-        )
-
-        payload = json.loads(completed.stdout)
-        self.assertEqual(len(payload), 2)
-        self.assertEqual(payload[0]["events"][0]["kind"], "stdout")
-        self.assertTrue(payload[1]["options"]["collapsed"])
+        generator.word.assert_any_call(include_categories=["adjectives"])
+        generator.word.assert_any_call(include_categories=["nouns"])
 
     def test_run_code_returns_frontend_facing_call_response(self):
         app = App()