[source-google-analytics-data-api] - Migrate to CDK v5 (#47013)
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user