fixed tests

fylein · Aug 5, 2024 · c384240 · c384240
1 parent 08c9bfc
commit c384240
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/apps/mappings/connector.py b/apps/mappings/connector.py
@@ -112,7 +112,9 @@ def sync_projects(self, source_type: str):
         'order': 'updated_at.desc'
         }
         projects_generator = self.platform.v1beta.admin.projects.list_all(query)
-        existing_projects = QBDMapping.objects.filter(workspace_id=self.workspace_id, attribute_type=source_type).values_list('source_value', flat=True)
+        existing_projects = QBDMapping.objects.filter(
+            workspace_id=self.workspace_id,
+            attribute_type=source_type).values_list('source_value', flat=True)
 
         source_attributes = []
 
@@ -140,7 +142,9 @@ def sync_cost_center(self, source_type: str):
         }
 
         cost_center_generator = self.platform.v1beta.admin.cost_centers.list_all(query)
-        existing_cost_centers = QBDMapping.objects.filter(workspace_id=self.workspace_id, attribute_type=source_type).values_list('source_value', flat=True)
+        existing_cost_centers = QBDMapping.objects.filter(
+            workspace_id=self.workspace_id,
+            attribute_type=source_type).values_list('source_value', flat=True)
 
         source_attributes = []
 

diff --git a/tests/test_mapping/test_connector.py b/tests/test_mapping/test_connector.py
@@ -61,7 +61,6 @@ def test_sync_cost_center(create_temp_workspace, add_fyle_credentials, mocker):
     qbd_mappings = QBDMapping.objects.filter(workspace_id=workspace_id, attribute_type=source_type)
     assert len(qbd_mappings) == len(mock_response[0]['data'])
     for i, mapping in enumerate(qbd_mappings):
-        cost_center = mock_response[0]['data'][i]
         assert mapping.source_value == 'Cost Center 1'