handle case when no state is passed

Author: aldogonzalez8

airbyte_cdk/sources/declarative/migrations/legacy_to_per_partition_state_migration.py

@@ -78,19 +78,20 @@ def should_migrate(self, stream_state: Mapping[str, Any]) -> bool:
           "<cursor_field>" : "<cursor_value>"
         }
         """
-        if stream_state:
-            for key, value in stream_state.items():
-                if isinstance(value, dict):
-                    keys = list(value.keys())
-                    if len(keys) != 1:
-                        # The input partitioned state should only have one key
-                        return False
-                    if keys[0] != self._cursor_field:
-                        # Unexpected key. Found {keys[0]}. Expected {self._cursor.cursor_field}
-                        return False
-                # it is expected the internal value to be a dictionary according to docstring
-                else:
-                    return False
+        if not stream_state:
+            return False
+        for key, value in stream_state.items():
+            # it is expected the internal value to be a dictionary according to docstring
+            if not isinstance(value, dict):
+                return False
+            keys = list(value.keys())
+            if len(keys) != 1:
+                # The input partitioned state should only have one key
+                return False
+            if keys[0] != self._cursor_field:
+                # Unexpected key. Found {keys[0]}. Expected {self._cursor.cursor_field}
+                return False
+
         return True
 
     def migrate(self, stream_state: Mapping[str, Any]) -> Mapping[str, Any]:

unit_tests/sources/declarative/migrations/test_legacy_to_per_partition_migration.py

@@ -193,6 +193,10 @@ def test_migrate_a_valid_legacy_state_to_per_partition():
             {"last_changed": "2022-12-27T08:34:39+00:00"},
             id="test_should_not_migrate_if_the_partitioned_state_is_not_in_correct_format",
         ),
+        pytest.param(
+            {},
+            id="test_should_not_migrate_if_not_state_is_passed",
+        ),
     ],
 )
 def test_should_not_migrate(input_state):