udpate unittests

singer-io · Dec 11, 2024 · d14a435 · d14a435
1 parent c0e835e
commit d14a435
Showing 1 changed file with 17 additions and 19 deletions.
diff --git a/tests/unittests/test_get_streams_and_state_translate.py b/tests/unittests/test_get_streams_and_state_translate.py
@@ -33,11 +33,10 @@ class TestTranslateState(unittest.TestCase):
     def test_newer_format_state_with_repo_name(self):
         """Verify that `translate_state` return the state itself if a newer format bookmark is found."""
         state = {
-            "bookmarks": {
-                "org/test-repo" : {
-                        "comments": {"since": "2019-01-01T00:00:00Z"}
+            "bookmarks" : {
+                "comments" : {
+                    "org/test-repo": {"since": "2019-01-01T00:00:00Z"},
                     },
-                "org/test-repo2" : {}
             }
         }
 
@@ -47,11 +46,6 @@ def test_newer_format_state_with_repo_name(self):
     def test_older_format_state_without_repo_name(self):
         """Verify that `translate_state` migrate each stream's bookmark into the repo name"""
         older_format_state = {
-            "bookmarks": {
-                "comments": {"since": "2019-01-01T00:00:00Z"}
-            }
-        }
-        expected_state =  {
             "bookmarks": {
                 "org/test-repo" : {
                         "comments": {"since": "2019-01-01T00:00:00Z"}
@@ -61,6 +55,12 @@ def test_older_format_state_without_repo_name(self):
                     }
             }
         }
+        expected_state =  {
+            "bookmarks": {
+                "comments": {"org/test-repo" : {"since": "2019-01-01T00:00:00Z"},
+                            "org/test-repo2" : {"since": "2019-01-01T00:00:00Z"}},
+            }
+        }
         final_state = translate_state(older_format_state, self.catalog, ["org/test-repo", "org/test-repo2"])
         self.assertEqual(expected_state, dict(final_state))
 
@@ -75,10 +75,9 @@ def test_state_with_no_previous_repo_name_newer_format_bookmark(self):
         """Verify that `translate_state` return the existing state if all existing repo unselected in the current sync."""
         newer_format_state = {
             "bookmarks": {
-                "org/test-repo" : {
-                        "comments": {"since": "2019-01-01T00:00:00Z"}
+                "comments" : {
+                        "org/test-repo": {"since": "2019-01-01T00:00:00Z"}
                     },
-                "org/test-repo2" : {}
             }
         }
         final_state = translate_state(newer_format_state, self.catalog, ["org/test-repo3", "org/test-repo4"])
@@ -88,17 +87,16 @@ def test_state_with_no_previous_repo_name_old_format_bookmark(self):
         """Verify that `translate_state` migrate each stream's bookmark into the repo name"""
         older_format_state = {
             "bookmarks": {
-                "comments": {"since": "2019-01-01T00:00:00Z"}
+                "org/test-repo3": {
+                    "comments": {"since": "2019-01-01T00:00:00Z"}
+                 }
             }
         }
         expected_state = {
             "bookmarks": {
-                "org/test-repo3" : {
-                        "comments": {"since": "2019-01-01T00:00:00Z"}
-                    },
-                "org/test-repo4" : {
-                        "comments": {"since": "2019-01-01T00:00:00Z"}
-                    }
+                "comments" : {
+                    "org/test-repo3": {"since": "2019-01-01T00:00:00Z"},
+                },
             }
         }
         final_state = translate_state(older_format_state, self.catalog, ["org/test-repo3", "org/test-repo4"])