fix: Write a valid final state message at the end of each stream sync (…

…#1164) Co-authored-by: Eric Boucher <[email protected]>
meltano · Nov 10, 2022 · ff3d65e · ff3d65e
1 parent 0a7f086
commit ff3d65e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/singer_sdk/tap_base.py b/singer_sdk/tap_base.py
@@ -380,6 +380,7 @@ def sync_all(self) -> None:
 
             stream.sync()
             stream.finalize_state_progress_markers()
+            stream._write_state_message()
 
         # this second loop is needed for all streams to print out their costs
         # including child streams which are otherwise skipped in the loop above

diff --git a/tests/core/test_countries_sync.py b/tests/core/test_countries_sync.py
@@ -130,4 +130,4 @@ def tally_messages(messages: list) -> Counter:
     assert counter["SCHEMA", "countries"] == 1
     assert counter["BATCH", "countries"] == 1
 
-    assert counter[("STATE",)] == 2
+    assert counter[("STATE",)] == 4