fix bing_webmaster test

Author: djudjuu

sources/asana_dlt/__init__.py

@@ -47,8 +47,9 @@ def workspaces(fields: Iterable[str] = WORKSPACE_FIELDS) -> Iterable[TDataItem]:
         Yields:
             dict: The workspace data.
         """
-        yield from get_client(access_token).workspaces.find_all(opt_fields=",".join(fields))
-
+        yield from get_client(access_token).workspaces.find_all(
+            opt_fields=",".join(fields)
+        )
 
     @dlt.transformer(
         data_from=workspaces,
@@ -75,7 +76,6 @@ def projects(
             )
         )
 
-
     @dlt.transformer(
         data_from=projects,
         write_disposition="replace",
@@ -103,7 +103,6 @@ def sections(
             )
         ]
 
-
     @dlt.transformer(data_from=workspaces, write_disposition="replace")
     @dlt.defer
     def tags(
@@ -127,7 +126,6 @@ def tags(
             )
         ]
 
-
     @dlt.transformer(data_from=projects, write_disposition="merge", primary_key="gid")
     def tasks(
         project_array: t.List[TDataItem],
@@ -157,7 +155,6 @@ def tasks(
             )
         )
 
-
     @dlt.transformer(
         data_from=tasks,
         write_disposition="append",
@@ -184,7 +181,6 @@ def stories(
             )
         ]
 
-
     @dlt.transformer(
         data_from=workspaces,
         write_disposition="replace",
@@ -211,7 +207,6 @@ def teams(
             )
         ]
 
-
     @dlt.transformer(
         data_from=workspaces,
         write_disposition="replace",
@@ -249,4 +244,3 @@ def users(
         teams,
         users,
     ]
-

tests/bing_webmaster/test_bing_webmaster_source.py

@@ -17,7 +17,7 @@ def _make_pipeline(destination_name: str) -> dlt.Pipeline:
 @pytest.mark.parametrize("destination_name", ["duckdb"])
 def test_load_unauthorized_domain(destination_name: str) -> None:
     pipeline = _make_pipeline(destination_name)
-    data = source(site_urls=["wikipedia.org"])
+    data = source(site_urls=["wikipedia.org"], site_url_pages=[])
     with pytest.raises(Exception):
         pipeline.run(data.with_resources("page_stats"))
 
@@ -28,7 +28,7 @@ def test_load_page_stats(destination_name: str) -> None:
     table_name = "bing_page_stats"
 
     # Note: If this test fails: replace this with a site_url you can access
-    data = source(site_urls=["dlthub.com"])
+    data = source(site_urls=["dlthub.com"], site_url_pages=[])
 
     info = pipeline.run(data.with_resources("page_stats"))
     assert_load_info(info)
@@ -63,11 +63,12 @@ def test_load_page_query_stats(destination_name: str) -> None:
     table_name = "bing_page_query_stats"
 
     data = source(
+        site_urls=["dlthub.com"],
         # Note: If this test fails: replace this with site_url and pages you can access
         site_url_pages=[
             {"site_url": "dlthub.com", "page": "https://dlthub.com/docs/intro"},
             {"site_url": "dlthub.com", "page": "https://dlthub.com/why/"},
-        ]
+        ],
     )
     info = pipeline.run(data.with_resources("page_query_stats"))
     assert_load_info(info)