chore(source-woocommerce): migrate woocommerce integration tests to use new mocking utils (#44037)

Author: strosek

airbyte-integrations/connectors/source-woocommerce/unit_tests/integration/common.py

@@ -0,0 +1,42 @@
+# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
+
+from typing import Any, MutableMapping, Optional
+
+import source_woocommerce
+from airbyte_cdk import AbstractSource
+from airbyte_cdk.connector_builder.connector_builder_handler import resolve_manifest
+
+
+def config() -> MutableMapping[str, Any]:
+    return {
+        "api_key": "test_api_key",
+        "api_secret": "test_api_secret",
+        "shop": "airbyte.store",
+        "start_date": "2017-01-01",
+    }
+
+
+def source() -> AbstractSource:
+    return source_woocommerce.SourceWoocommerce()
+
+
+def url_base() -> str:
+    url = resolve_manifest(source()).record.data["manifest"]["definitions"]["requester"]["url_base"]
+    url = url.replace("{{ config['shop'] }}", config()["shop"])
+    return url
+
+
+def common_params():
+    return "orderby=id&order=asc&dates_are_gmt=true&per_page=100"
+
+
+def build_url(resource_path: str, is_regex: bool = False, modified_after: str = None, modified_before: str = None) -> str:
+    """Build a URL for a WooCommerce API endpoint."""
+    separator = "." if is_regex else "?"
+    url = f"{url_base()}/{resource_path}{separator}{common_params()}"
+    if modified_after:
+        url = f"{url}&modified_after={modified_after}"
+    if modified_before:
+        url = f"{url}&modified_before={modified_before}"
+
+    return url

airbyte-integrations/connectors/source-woocommerce/unit_tests/integration/conftest.py

@@ -0,0 +1,134 @@
+# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
+
+from airbyte_cdk.test.utils.data import read_resource_file_contents
+
+from .common import build_url
+
+
+def request_response_mapping(stream_name, method="GET", custom_resource=None, is_regex=False, modified_after=None, modified_before=None,
+                             custom_json_filename=None, status_code=200, response=None):
+    """Create an HTTP request-response mapping for a stream."""
+    json_filename = f"{stream_name}.json" if custom_json_filename is None else custom_json_filename
+    resource = stream_name if custom_resource is None else custom_resource
+
+    response_map = response
+    if response is None:
+        response_map = {"text": read_resource_file_contents(json_filename, __file__), "status_code": status_code}
+
+    return {
+        "request": {
+            "url": build_url(resource, modified_after=modified_after, modified_before=modified_before, is_regex=is_regex),
+            "is_regex": is_regex,
+            "method": method,
+        },
+        "response": response_map,
+    }
+
+
+def orders_http_calls():
+    return [request_response_mapping("orders")]
+
+
+def coupons_http_calls():
+    return [request_response_mapping("coupons")]
+
+
+def customers_http_calls():
+    return [request_response_mapping("customers")]
+
+
+def payment_gateways_http_calls():
+    return [request_response_mapping("payment_gateways")]
+
+
+def product_attributes_http_calls():
+    return [request_response_mapping("product_attributes", custom_resource="products/attributes")]
+
+
+def product_categories_http_calls():
+    return [request_response_mapping("product_categories", custom_resource="products/categories")]
+
+
+def product_reviews_http_calls():
+    return [request_response_mapping("product_reviews", custom_resource="products/reviews")]
+
+
+def products_http_calls():
+    return [request_response_mapping("products")]
+
+
+def product_shipping_classes_http_calls():
+    return [request_response_mapping("product_shipping_classes", custom_resource="products/shipping_classes")]
+
+
+def product_tags_http_calls():
+    return [request_response_mapping("product_tags", custom_resource="products/tags")]
+
+
+def shipping_methods_http_calls():
+    return [request_response_mapping("shipping_methods", custom_resource="shipping_methods")]
+
+
+def shipping_zones_http_calls():
+    return [request_response_mapping("shipping_zones", custom_resource="shipping/zones")]
+
+
+def system_status_tools_http_calls():
+    return [request_response_mapping("system_status_tools", custom_resource="system_status/tools")]
+
+
+def order_notes_http_calls():
+    return [
+        request_response_mapping("orders", modified_after="2017-01-01.+", modified_before="2017-01-29.+", is_regex=True),
+        request_response_mapping("order_notes", custom_resource="orders/(727|723)/notes", is_regex=True),
+    ]
+
+
+def product_attribute_terms_http_calls():
+    return [
+        request_response_mapping("product_attributes", custom_resource="products/attributes"),
+        request_response_mapping("product_attribute_terms", custom_resource="products/attributes/.+/terms", is_regex=True),
+    ]
+
+
+def product_variations_http_calls():
+    return [
+        request_response_mapping("products"),
+        request_response_mapping("product_variations", custom_resource="products/(799|794)/variations", is_regex=True),
+    ]
+
+
+def refunds_http_calls():
+    return [
+        request_response_mapping("orders"),
+        request_response_mapping("refunds", custom_resource="orders/(727|723)/refunds", is_regex=True),
+    ]
+
+
+def shipping_zone_locations_http_calls():
+    return [
+        request_response_mapping("shipping_zones", custom_resource="shipping/zones"),
+        request_response_mapping("shipping_zone_locations", custom_resource="shipping/zones/(0|5)/locations", is_regex=True),
+    ]
+
+
+def shipping_zone_methods_http_calls():
+    return [
+        request_response_mapping("shipping_zones", custom_resource="shipping/zones"),
+        request_response_mapping("shipping_zone_methods", custom_resource="shipping/zones/(0|5)/methods", is_regex=True),
+    ]
+
+
+def tax_classes_http_calls():
+    return [request_response_mapping("tax_classes", custom_resource="taxes/classes")]
+
+
+def tax_rates_http_calls():
+    return [request_response_mapping("tax_rates", custom_resource="taxes")]
+
+
+def orders_empty_last_page():
+    return [
+        request_response_mapping("orders", is_regex=True, modified_after=".+", modified_before="2017-01-30.+"),
+        request_response_mapping("orders", is_regex=True, modified_after=".+", modified_before="2017-02-10.+", response={"text": "[]"}),
+    ]