fix test to not access full table stream key in bookmark

singer-io · Jan 13, 2025 · 3c34f1e · 3c34f1e
1 parent 852b9e5
commit 3c34f1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_github_interrupted_sync_remove_stream.py b/tests/test_github_interrupted_sync_remove_stream.py
@@ -189,8 +189,8 @@ def run_interrupted_sync(self, removed_stream):
 
                         else:
                             # Verify full table streams do not save bookmarked values after a successful sync
-                            self.assertNotIn(stream, full_sync_bookmark.keys())
-                            self.assertNotIn(stream, final_bookmark.keys())
+                            self.assertNotIn(stream, full_sync_state["bookmarks"].keys())
+                            self.assertNotIn(stream, final_state["bookmarks"].keys())
 
                             # Verify first and second sync have the same records
                             self.assertEqual(full_record_count, interrupted_record_count)