format

Author: pedroslopez

unit_tests/manifest_server/command_processor/test_utils.py

@@ -32,9 +32,7 @@ def test_build_catalog_creates_correct_structure(self):
         assert configured_stream.destination_sync_mode == DestinationSyncMode.overwrite
 
     @patch("airbyte_cdk.manifest_server.command_processor.utils.ConcurrentDeclarativeSource")
-    def test_build_source_creates_manifest_declarative_source(
-        self, mock_source_class
-    ):
+    def test_build_source_creates_manifest_declarative_source(self, mock_source_class):
         """Test that build_source creates a ConcurrentDeclarativeSource with correct parameters."""
         # Setup mocks
         mock_source = Mock()
@@ -85,9 +83,7 @@ def test_build_source_creates_manifest_declarative_source(
         assert result == mock_source
 
     @patch("airbyte_cdk.manifest_server.command_processor.utils.ConcurrentDeclarativeSource")
-    def test_build_source_with_normalize_flag(
-        self, mock_source_class
-    ):
+    def test_build_source_with_normalize_flag(self, mock_source_class):
         """Test build_source when normalize flag is set."""
         mock_source = Mock()
         mock_source_class.return_value = mock_source

unit_tests/manifest_server/routers/test_manifest.py

@@ -110,13 +110,13 @@ def test_test_read_endpoint_success(
         assert response.status_code == 200
         # Verify build_source was called with correct arguments
         mock_build_source.assert_called_once_with(
-            sample_manifest, 
-            mock_build_catalog.return_value, 
-            sample_config, 
-            [], 
+            sample_manifest,
+            mock_build_catalog.return_value,
+            sample_config,
+            [],
             100,  # record_limit
-            5,    # page_limit
-            5     # slice_limit
+            5,  # page_limit
+            5,  # slice_limit
         )
         mock_build_catalog.assert_called_once_with("products")
         mock_runner.test_read.assert_called_once()
@@ -167,16 +167,16 @@ def test_test_read_with_custom_components(
         assert config_arg["__injected_components_py"] == custom_code
         assert "__injected_components_py_checksums" in config_arg
         assert config_arg["__injected_components_py_checksums"]["md5"] == expected_checksum
-        
+
         # Verify other arguments
         mock_build_source.assert_called_once_with(
             sample_manifest,
             mock_build_catalog.return_value,
             config_arg,
             [],
             50,  # record_limit
-            3,   # page_limit
-            2    # slice_limit
+            3,  # page_limit
+            2,  # slice_limit
         )
 
     @patch("airbyte_cdk.manifest_server.routers.manifest.AirbyteStateMessageSerializer")
@@ -245,13 +245,13 @@ def test_resolve_endpoint_success(self, sample_manifest, mock_source):
             assert "manifest" in data
             assert data["manifest"] == mock_source.resolved_manifest
             mock_build_source.assert_called_once_with(
-                sample_manifest, 
+                sample_manifest,
                 None,  # catalog
-                {},    # config
+                {},  # config
                 None,  # state
                 None,  # record_limit
                 None,  # page_limit
-                None   # slice_limit
+                None,  # slice_limit
             )
 
     def test_resolve_invalid_manifest(self):
@@ -417,7 +417,7 @@ def test_check_endpoint_success(
             None,  # state
             None,  # record_limit
             None,  # page_limit
-            None   # slice_limit
+            None,  # slice_limit
         )
         mock_runner_class.assert_called_once_with(mock_source)
         mock_runner.check_connection.assert_called_once_with(sample_config)
@@ -445,15 +445,15 @@ def test_check_endpoint_failure(
         data = response.json()
         assert data["success"] is False
         assert data["message"] == "Invalid API key"
-        
+
         mock_build_source.assert_called_once_with(
             sample_manifest,
             None,  # catalog
             sample_config,
             None,  # state
             None,  # record_limit
             None,  # page_limit
-            None   # slice_limit
+            None,  # slice_limit
         )
 
     @patch("airbyte_cdk.manifest_server.routers.manifest.ManifestCommandProcessor")
@@ -500,7 +500,7 @@ def test_discover_endpoint_success(
             None,  # state
             None,  # record_limit
             None,  # page_limit
-            None   # slice_limit
+            None,  # slice_limit
         )
         mock_runner_class.assert_called_once_with(mock_source)
         mock_runner.discover.assert_called_once_with(sample_config)