update abnormal test

airbytehq · May 7, 2024 · 92655a8 · 92655a8
1 parent e510be1
commit 92655a8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 23 deletions.
diff --git a/airbyte-integrations/connectors/source-mixpanel/integration_tests/abnormal_state.json b/airbyte-integrations/connectors/source-mixpanel/integration_tests/abnormal_state.json
@@ -3,8 +3,10 @@
     "type": "STREAM",
     "stream": {
       "stream_state": {
+        "36152117": { "date": "2030-01-01" },
         "41833532": { "date": "2030-01-01" },
-        "36152117": { "date": "2030-01-01" }
+        "41833755": { "date": "2030-01-01" },
+        "41833700": { "date": "2030-01-01" }
       },
       "stream_descriptor": { "name": "funnels" }
     }

diff --git a/airbyte-integrations/connectors/source-mixpanel/source_mixpanel/components.py b/airbyte-integrations/connectors/source-mixpanel/source_mixpanel/components.py
@@ -253,28 +253,6 @@ def stream_slices(self) -> Iterable[StreamSlice]:
                         yield from []
 
 
-class FunnelsLegacyToPerPartitionStateMigration(LegacyToPerPartitionStateMigration):
-    """
-    Gor error when use custom StateMigration:
-        custom_component_class(**kwargs):
-            TypeError: LegacyToPerPartitionStateMigration.__init__() missing 2 required positional arguments: 'partition_router', 'cursor'
-
-    """
-
-    partition_router: SubstreamPartitionRouter = None
-    cursor: DatetimeBasedCursor = None
-    config: Mapping[str, Any]
-    parameters: Mapping[str, Any]
-
-    def migrate(self, stream_state: Mapping[str, Any]) -> Mapping[str, Any]:
-        state = super().migrate(stream_state)
-        for partition_state in state.get("states", []):
-            # add empty parent_slice attr to partition
-            if "parent_slice" not in partition_state.get("partition", {}):
-                partition_state["partition"]["parent_slice"] = {}
-        return state
-
-
 @dataclass
 class EngagePaginationStrategy(PageIncrement):
     """