1
0
mirror of synced 2025-12-23 21:03:15 -05:00

[source-google-analytics-data-api] - Migrate to CDK v5 (#47013)

This commit is contained in:
Patrick Nilan
2024-10-28 16:53:45 -07:00
committed by GitHub
parent 0dc4efe7f7
commit 5fda600d3c
8 changed files with 381 additions and 101 deletions

View File

@@ -5,10 +5,9 @@
from unittest.mock import MagicMock, patch
import pytest
from airbyte_cdk.models import AirbyteConnectionStatus, FailureType
from airbyte_cdk.models import AirbyteConnectionStatus, FailureType, Status
from airbyte_cdk.sources.streams.http.http import HttpStatusErrorHandler
from airbyte_cdk.utils import AirbyteTracedException
from airbyte_protocol.models import Status
from source_google_analytics_data_api import SourceGoogleAnalyticsDataApi
from source_google_analytics_data_api.api_quota import GoogleAnalyticsApiQuotaBase
from source_google_analytics_data_api.source import GoogleAnalyticsDatApiErrorHandler, MetadataDescriptor

View File

@@ -64,7 +64,8 @@ def test_request_params(patch_base_class):
def test_request_body_json(patch_base_class):
request_body_params = {"stream_state": MagicMock(), "stream_slice": MagicMock(), "next_page_token": None}
stream_slice = {"startDate": "2024-01-01", "endDate": "2024-01-31"}
request_body_params = {"stream_state": MagicMock(), "stream_slice": stream_slice, "next_page_token": None}
expected_body_json = {
"metrics": [
@@ -84,7 +85,10 @@ def test_request_body_json(patch_base_class):
{"name": "browser"},
],
"keepEmptyRows": True,
"dateRanges": [request_body_params["stream_slice"]],
"dateRanges": [{
"startDate": request_body_params["stream_slice"]["startDate"],
"endDate": request_body_params["stream_slice"]["endDate"]
}],
"returnPropertyQuota": True,
"offset": str(0),
"limit": "100000",