2
2
# Copyright (c) 2021 Airbyte, Inc., all rights reserved.
3
3
#
4
4
5
- import math
6
5
from unittest .mock import MagicMock , PropertyMock , patch
7
6
8
7
import pendulum
9
8
import pytest
10
- from source_facebook_marketing .streams .async_job_manager import InsightsAsyncJobManager
9
+ from source_facebook_marketing .streams .async_job_manager import InsightAsyncJobManager
11
10
12
11
13
12
@pytest .fixture (autouse = False )
@@ -45,7 +44,7 @@ def test_async_job_manager(job_mock, from_date, to_date):
45
44
from_date , to_date = pendulum .parse (from_date ), pendulum .parse (to_date )
46
45
assert from_date <= to_date
47
46
api_mock = make_api_mock ()
48
- job_manager = InsightsAsyncJobManager (
47
+ job_manager = InsightAsyncJobManager (
49
48
api = api_mock ,
50
49
job_params = {"breakdowns" : []},
51
50
from_date = from_date ,
@@ -65,7 +64,7 @@ def test_async_job_manager_to_date_greater_from(job_mock):
65
64
from_date , to_date = pendulum .parse ("2020-10-10" ), pendulum .parse ("2019-10-10" )
66
65
assert from_date > to_date
67
66
api_mock = make_api_mock ()
68
- job_manager = InsightsAsyncJobManager (
67
+ job_manager = InsightAsyncJobManager (
69
68
api = api_mock ,
70
69
job_params = {"breakdowns" : []},
71
70
from_date = from_date ,
@@ -78,7 +77,7 @@ def test_async_job_manager_to_date_greater_from(job_mock):
78
77
def test_job_failed (job_mock ):
79
78
from_date , to_date = pendulum .parse ("2019-10-10" ), pendulum .parse ("2019-10-10" )
80
79
api_mock = make_api_mock ()
81
- job_manager = InsightsAsyncJobManager (
80
+ job_manager = InsightAsyncJobManager (
82
81
api = api_mock ,
83
82
job_params = {"breakdowns" : []},
84
83
from_date = from_date ,
@@ -95,7 +94,7 @@ def test_job_failed(job_mock):
95
94
def test_job_failed_two_times (job_mock ):
96
95
from_date , to_date = pendulum .parse ("2019-10-10" ), pendulum .parse ("2019-10-10" )
97
96
api_mock = make_api_mock ()
98
- job_manager = InsightsAsyncJobManager (
97
+ job_manager = InsightAsyncJobManager (
99
98
api = api_mock ,
100
99
job_params = {"breakdowns" : []},
101
100
from_date = from_date ,
@@ -113,7 +112,7 @@ def test_job_failed_two_times(job_mock):
113
112
def test_job_wait_unitll_completed (job_mock , time_sleep_mock ):
114
113
from_date , to_date = pendulum .parse ("2019-10-10" ), pendulum .parse ("2019-10-10" )
115
114
api_mock = make_api_mock ()
116
- job_manager = InsightsAsyncJobManager (
115
+ job_manager = InsightAsyncJobManager (
117
116
api = api_mock ,
118
117
job_params = {"breakdowns" : []},
119
118
from_date = from_date ,
0 commit comments