run black

Author: dongwonmoon

data_scribe/components/writers/notion_writer.py

@@ -34,12 +34,12 @@ def _connect(self):
         try:
             self.notion = Client(auth=token)
             logger.info("Successfully connected to Notion API.")
-        
+
         except Exception as e:
             # This now correctly catches only Client() initialization errors
             logger.error(f"Failed to connect to Notion: {e}", exc_info=True)
             raise ConnectionError(f"Failed to connect to Notion: {e}")
-        
+
     def write(self, catalog_data: Dict[str, Any], **kwargs):
         self.params = kwargs
         self._connect()
@@ -48,9 +48,7 @@ def write(self, catalog_data: Dict[str, Any], **kwargs):
         if not parent_page_id:
             raise ConfigError("'parent_page_id' is required for NotionWriter.")
 
-        project_name = kwargs.get(
-            "project_name", "Data Catalog"
-        )
+        project_name = kwargs.get("project_name", "Data Catalog")
         page_title = f"Data Catalog - {project_name}"
 
         # --- 1. Generate Notion Blocks ---

data_scribe/tests/unit/writers/test_notion_writer.py

@@ -5,6 +5,7 @@
 from data_scribe.components.writers import NotionWriter
 from data_scribe.core.exceptions import ConfigError
 
+
 @pytest.fixture
 def mock_db_catalog_data():
     """Provides a mock catalog data structure for standard DB connections."""
@@ -50,7 +51,7 @@ def test_notion_writer_success(mock_notion_client, mock_db_catalog_data):
     mock_client_instance = MagicMock()
     mock_client_instance.pages.create.return_value = {
         "id": "new-page-id",
-        "url": "http://notion.so/new-page-id"
+        "url": "http://notion.so/new-page-id",
     }
     mock_notion_client.return_value = mock_client_instance
 
@@ -59,7 +60,7 @@ def test_notion_writer_success(mock_notion_client, mock_db_catalog_data):
     kwargs = {
         "api_token": "fake_token",
         "parent_page_id": "fake-parent-id",
-        "project_name": "test_db"
+        "project_name": "test_db",
     }
 
     # 3. Run
@@ -71,39 +72,51 @@ def test_notion_writer_success(mock_notion_client, mock_db_catalog_data):
     # 5. Assert Page Creation
     expected_title = "Data Catalog - test_db"
     expected_parent = {"page_id": "fake-parent-id"}
-    
+
     mock_client_instance.pages.create.assert_called_once()
-    
+
     # Get the arguments passed to pages.create
     create_args, create_kwargs = mock_client_instance.pages.create.call_args
-    
+
     assert create_kwargs["parent"] == expected_parent
-    assert create_kwargs["properties"]["title"][0]["text"]["content"] == expected_title
-    
+    assert (
+        create_kwargs["properties"]["title"][0]["text"]["content"]
+        == expected_title
+    )
+
     # Check that blocks were generated
     blocks = create_kwargs["children"]
     assert len(blocks) > 0
-    assert blocks[0]["type"] == "heading_2" # "🔎 Views"
-    assert blocks[0]["heading_2"]["rich_text"][0]["text"]["content"] == "🔎 Views"
+    assert blocks[0]["type"] == "heading_2"  # "🔎 Views"
+    assert (
+        blocks[0]["heading_2"]["rich_text"][0]["text"]["content"] == "🔎 Views"
+    )
     # Find the table H2
-    table_h2 = next(b for b in blocks if b.get("type") == "heading_2" and "Tables" in b["heading_2"]["rich_text"][0]["text"]["content"])
+    table_h2 = next(
+        b
+        for b in blocks
+        if b.get("type") == "heading_2"
+        and "Tables" in b["heading_2"]["rich_text"][0]["text"]["content"]
+    )
     assert table_h2 is not None
 
 
 @patch.dict(os.environ, {"NOTION_TEST_KEY": "env_key_value"})
 @patch("data_scribe.components.writers.notion_writer.Client")
-def test_notion_writer_resolves_env_var(mock_notion_client, mock_db_catalog_data):
+def test_notion_writer_resolves_env_var(
+    mock_notion_client, mock_db_catalog_data
+):
     """
     Tests that the writer correctly resolves API tokens from environment variables.
     """
-    mock_notion_client.return_value = MagicMock() # Basic mock
-    
+    mock_notion_client.return_value = MagicMock()  # Basic mock
+
     writer = NotionWriter()
     kwargs = {
-        "api_token": "${NOTION_TEST_KEY}", # Reference the env var
+        "api_token": "${NOTION_TEST_KEY}",  # Reference the env var
         "parent_page_id": "fake-parent-id",
     }
-    
+
     writer.write(mock_db_catalog_data, **kwargs)
 
     # Assert client was initialized with the *resolved* key
@@ -118,14 +131,8 @@ def test_notion_writer_config_errors(mock_db_catalog_data):
 
     # 1. Missing api_token
     with pytest.raises(ConfigError, match="'api_token'.*is required"):
-        writer.write(
-            mock_db_catalog_data,
-            parent_page_id="fake-parent-id"
-        )
+        writer.write(mock_db_catalog_data, parent_page_id="fake-parent-id")
 
     # 2. Missing parent_page_id
     with pytest.raises(ConfigError, match="'parent_page_id' is required"):
-        writer.write(
-            mock_db_catalog_data,
-            api_token="fake_token"
-        )
+        writer.write(mock_db_catalog_data, api_token="fake_token")