Skip to content

Commit 3c0754d

Browse files
committed
fixing unittests
1 parent 8d560d8 commit 3c0754d

File tree

2 files changed

+6
-54
lines changed

2 files changed

+6
-54
lines changed

airbyte-integrations/connectors/source-facebook-marketing/unit_tests/test_async_job_manager.py

+6-7
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,11 @@
22
# Copyright (c) 2021 Airbyte, Inc., all rights reserved.
33
#
44

5-
import math
65
from unittest.mock import MagicMock, PropertyMock, patch
76

87
import pendulum
98
import pytest
10-
from source_facebook_marketing.streams.async_job_manager import InsightsAsyncJobManager
9+
from source_facebook_marketing.streams.async_job_manager import InsightAsyncJobManager
1110

1211

1312
@pytest.fixture(autouse=False)
@@ -45,7 +44,7 @@ def test_async_job_manager(job_mock, from_date, to_date):
4544
from_date, to_date = pendulum.parse(from_date), pendulum.parse(to_date)
4645
assert from_date <= to_date
4746
api_mock = make_api_mock()
48-
job_manager = InsightsAsyncJobManager(
47+
job_manager = InsightAsyncJobManager(
4948
api=api_mock,
5049
job_params={"breakdowns": []},
5150
from_date=from_date,
@@ -65,7 +64,7 @@ def test_async_job_manager_to_date_greater_from(job_mock):
6564
from_date, to_date = pendulum.parse("2020-10-10"), pendulum.parse("2019-10-10")
6665
assert from_date > to_date
6766
api_mock = make_api_mock()
68-
job_manager = InsightsAsyncJobManager(
67+
job_manager = InsightAsyncJobManager(
6968
api=api_mock,
7069
job_params={"breakdowns": []},
7170
from_date=from_date,
@@ -78,7 +77,7 @@ def test_async_job_manager_to_date_greater_from(job_mock):
7877
def test_job_failed(job_mock):
7978
from_date, to_date = pendulum.parse("2019-10-10"), pendulum.parse("2019-10-10")
8079
api_mock = make_api_mock()
81-
job_manager = InsightsAsyncJobManager(
80+
job_manager = InsightAsyncJobManager(
8281
api=api_mock,
8382
job_params={"breakdowns": []},
8483
from_date=from_date,
@@ -95,7 +94,7 @@ def test_job_failed(job_mock):
9594
def test_job_failed_two_times(job_mock):
9695
from_date, to_date = pendulum.parse("2019-10-10"), pendulum.parse("2019-10-10")
9796
api_mock = make_api_mock()
98-
job_manager = InsightsAsyncJobManager(
97+
job_manager = InsightAsyncJobManager(
9998
api=api_mock,
10099
job_params={"breakdowns": []},
101100
from_date=from_date,
@@ -113,7 +112,7 @@ def test_job_failed_two_times(job_mock):
113112
def test_job_wait_unitll_completed(job_mock, time_sleep_mock):
114113
from_date, to_date = pendulum.parse("2019-10-10"), pendulum.parse("2019-10-10")
115114
api_mock = make_api_mock()
116-
job_manager = InsightsAsyncJobManager(
115+
job_manager = InsightAsyncJobManager(
117116
api=api_mock,
118117
job_params={"breakdowns": []},
119118
from_date=from_date,

airbyte-integrations/connectors/source-facebook-marketing/unit_tests/test_streams.py

-47
This file was deleted.

0 commit comments

Comments
 (0)