fix test cases

MrRoudyk · Nov 26, 2024 · 3d6f8c7 · 3d6f8c7
1 parent 1317737
commit 3d6f8c7
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/apps/data_handler/handler_tools/data_parser/serializers/zklend.py b/apps/data_handler/handler_tools/data_parser/serializers/zklend.py
@@ -181,9 +181,7 @@ class CollateralEnabledDisabledEventData(BaseModel):
     Attributes:
         user: The user address for whom collateral is being enabled/disabled.
         token: The token address being enabled/disabled as collateral.
-        enabled: Boolean indicating if collateral is being enabled (True) or disabled (False).
     """
 
     user: str
     token: str
-    enabled: bool
diff --git a/apps/data_handler/tests/test_zklend_transformer.py b/apps/data_handler/tests/test_zklend_transformer.py
@@ -376,10 +376,9 @@ def test_save_collateral_enabled_event(transformer, sample_collateral_enabled_ev
     Test saving a collateral enabled event.
     """
     transformer.api_connector.get_data.return_value = [sample_collateral_enabled_event_data]
-
     expected_parsed_data = CollateralEnabledDisabledEventData(
         user=sample_collateral_enabled_event_data['data'][0],
-        token=sample_collateral_enabled_event_data['data'][1]
+        token=sample_collateral_enabled_event_data['data'][1],
     )
 
     transformer.fetch_and_transform_events(