fix: better errors for generate all models

1 files changed, 8 insertions, 0 deletions

diff --git a/app/tests/test_audiobook_cli.py b/app/tests/test_audiobook_cli.py
index f5dddf4..f1eb8e8 100644
--- a/app/tests/test_audiobook_cli.py
+++ b/app/tests/test_audiobook_cli.py
@@ -494,7 +494,15 @@ class AllModelsConvertTests(unittest.TestCase):
         self.assertEqual(events[4],
                          {"kind": "book", "index": 2, "total": 2,
                           "name": "book.txt", "model": "m2"})
+        # Passthrough events are stamped with the model too, so the run
+        # view can attribute a chunk failure to its model.
+        self.assertEqual(events[1],
+                         {"kind": "chunks", "total": 3, "model": "m1"})
+        self.assertEqual(events[2],
+                         {"kind": "chunk_done", "chunk": 1, "total": 3,
+                          "model": "m1"})
         self.assertEqual(events[3]["model"], "m1")
+        self.assertEqual(events[5]["model"], "m2")
         self.assertEqual(events[7]["model"], "m2")
         self.assertEqual(events[8],
                          {"kind": "done", "ok": 2, "total": 2,