From e9ade405b75652b68661081be97a50e62ba449e9 Mon Sep 17 00:00:00 2001 From: Sergey Chvalyuk Date: Thu, 24 Mar 2022 11:27:13 +0200 Subject: [PATCH] mock sleep, pendulum fix Signed-off-by: Sergey Chvalyuk --- .../source_amazon_ads/streams/report_streams/report_streams.py | 2 +- .../source-amazon-ads/unit_tests/test_report_streams.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/airbyte-integrations/connectors/source-amazon-ads/source_amazon_ads/streams/report_streams/report_streams.py b/airbyte-integrations/connectors/source-amazon-ads/source_amazon_ads/streams/report_streams/report_streams.py index f1ce09f2f2b24..1de2a404cfa9f 100644 --- a/airbyte-integrations/connectors/source-amazon-ads/source_amazon_ads/streams/report_streams/report_streams.py +++ b/airbyte-integrations/connectors/source-amazon-ads/source_amazon_ads/streams/report_streams/report_streams.py @@ -373,7 +373,7 @@ def _calc_report_generation_date(report_date: str, profile) -> str: report_date = pendulum.from_format(report_date, ReportStream.REPORT_DATE_FORMAT) profile_tz = pytz.timezone(profile.timezone) profile_time = report_date.astimezone(profile_tz) - return profile_time.strftime(ReportStream.REPORT_DATE_FORMAT) + return profile_time.format(ReportStream.REPORT_DATE_FORMAT) @backoff.on_exception( backoff.expo, diff --git a/airbyte-integrations/connectors/source-amazon-ads/unit_tests/test_report_streams.py b/airbyte-integrations/connectors/source-amazon-ads/unit_tests/test_report_streams.py index 561b2c4b834dd..3c4ca65f3f89a 100644 --- a/airbyte-integrations/connectors/source-amazon-ads/unit_tests/test_report_streams.py +++ b/airbyte-integrations/connectors/source-amazon-ads/unit_tests/test_report_streams.py @@ -270,6 +270,7 @@ def test_display_report_stream_init_too_many_requests(mocker, test_config): ) @responses.activate def test_display_report_stream_backoff(mocker, test_config, modifiers, expected): + mocker.patch("time.sleep") setup_responses(init_response=REPORT_INIT_RESPONSE, metric_response=METRIC_RESPONSE) with freeze_time("2021-01-02 03:04:05") as frozen_time: