26
26
import pytest
27
27
from six import StringIO
28
28
29
- import adobe_umapi_client .helper
30
- from adobe_umapi_client import UMAPI
31
- from adobe_umapi_client .error import UMAPIError , UMAPIRetryError
29
+ import umapi_client .helper
30
+ from umapi_client import UMAPI
31
+ from umapi_client .error import UMAPIError , UMAPIRetryError
32
32
33
33
34
34
# This method will be used by the mock to replace UMAPI.users
@@ -59,53 +59,53 @@ def __init__(self, status_code, headers):
59
59
60
60
61
61
# test the pagination on success
62
- @mock .patch ('adobe_umapi_client .UMAPI.users' , side_effect = mocked_users )
62
+ @mock .patch ('umapi_client .UMAPI.users' , side_effect = mocked_users )
63
63
def test_helper_success (_ ):
64
64
api = UMAPI ('' , None )
65
- assert adobe_umapi_client .helper .paginate (api .users , 'success' ) == ["user0" , "user1" , "user2" ]
65
+ assert umapi_client .helper .paginate (api .users , 'success' ) == ["user0" , "user1" , "user2" ]
66
66
67
67
68
68
@pytest .fixture
69
69
def reduce_attempts ():
70
70
# reduce the number of retry attempts
71
- adobe_umapi_client .helper .retry_max_attempts = 3
71
+ umapi_client .helper .retry_max_attempts = 3
72
72
73
73
74
74
# test the retry logic with seconds in the header
75
- @mock .patch ('adobe_umapi_client .UMAPI.users' , side_effect = mocked_users )
75
+ @mock .patch ('umapi_client .UMAPI.users' , side_effect = mocked_users )
76
76
def test_helper_retry_seconds (_ , reduce_attempts ):
77
77
api = UMAPI ('' , None )
78
- assert adobe_umapi_client .helper .paginate (api .users , 'retrySecs' ) == ["user1" , "user2" ]
78
+ assert umapi_client .helper .paginate (api .users , 'retrySecs' ) == ["user1" , "user2" ]
79
79
80
80
81
81
# test the retry logic with a time in the header
82
- @mock .patch ('adobe_umapi_client .UMAPI.users' , side_effect = mocked_users )
82
+ @mock .patch ('umapi_client .UMAPI.users' , side_effect = mocked_users )
83
83
def test_helper_fail_date (_ , reduce_attempts ):
84
84
api = UMAPI ('' , None )
85
- assert adobe_umapi_client .helper .paginate (api .users , 'retryTime' ) == ["user1" , "user2" ]
85
+ assert umapi_client .helper .paginate (api .users , 'retryTime' ) == ["user1" , "user2" ]
86
86
87
87
88
88
# the default retry waits are really long, so don't use them while testing
89
89
@pytest .fixture
90
90
def reduce_delay ():
91
91
# reduce the wait on retry attempts
92
- adobe_umapi_client .helper .retry_max_attempts = 3
93
- adobe_umapi_client .helper .retry_exponential_backoff_factor = 1
94
- adobe_umapi_client .helper .retry_random_delay_max = 3
92
+ umapi_client .helper .retry_max_attempts = 3
93
+ umapi_client .helper .retry_exponential_backoff_factor = 1
94
+ umapi_client .helper .retry_random_delay_max = 3
95
95
96
96
97
97
# test the retry logic with a zero delay in the header
98
- @mock .patch ('adobe_umapi_client .UMAPI.users' , side_effect = mocked_users )
98
+ @mock .patch ('umapi_client .UMAPI.users' , side_effect = mocked_users )
99
99
def test_helper_retry_zero (_ , reduce_delay , reduce_attempts ):
100
100
api = UMAPI ('' , None )
101
- assert adobe_umapi_client .helper .paginate (api .users , 'retryNone' ) == ["user1" , "user2" ]
101
+ assert umapi_client .helper .paginate (api .users , 'retryNone' ) == ["user1" , "user2" ]
102
102
103
103
104
104
# test the retry logic with no header
105
- @mock .patch ('adobe_umapi_client .UMAPI.users' , side_effect = mocked_users )
105
+ @mock .patch ('umapi_client .UMAPI.users' , side_effect = mocked_users )
106
106
def test_helper_retry_noheader (_ , reduce_delay , reduce_attempts ):
107
107
api = UMAPI ('' , None )
108
- assert adobe_umapi_client .helper .paginate (api .users , 'retryNull' ) == ["user1" , "user2" ]
108
+ assert umapi_client .helper .paginate (api .users , 'retryNull' ) == ["user1" , "user2" ]
109
109
110
110
111
111
# py.test doesn't divert string logging, so use it
@@ -116,18 +116,18 @@ def log_stream():
116
116
logger = logging .getLogger ('test_logger' )
117
117
logger .setLevel (logging .WARNING )
118
118
logger .addHandler (handler )
119
- prior_logger = adobe_umapi_client .helper .logger
120
- adobe_umapi_client .helper .logger = logger
119
+ prior_logger = umapi_client .helper .logger
120
+ umapi_client .helper .logger = logger
121
121
yield stream
122
- adobe_umapi_client .helper .logger = prior_logger
122
+ umapi_client .helper .logger = prior_logger
123
123
handler .close ()
124
124
125
125
126
126
# test the retry logic with a custom logger
127
- @mock .patch ('adobe_umapi_client .UMAPI.users' , side_effect = mocked_users )
127
+ @mock .patch ('umapi_client .UMAPI.users' , side_effect = mocked_users )
128
128
def test_helper_retry_logging (_ , log_stream , reduce_attempts ):
129
129
api = UMAPI ('' , None )
130
- assert adobe_umapi_client .helper .paginate (api .users , 'retrySecs' ) == ["user1" , "user2" ]
130
+ assert umapi_client .helper .paginate (api .users , 'retrySecs' ) == ["user1" , "user2" ]
131
131
log_stream .flush ()
132
132
log = log_stream .getvalue () # save as a local so can do pytest -l to see exact log
133
133
assert log == '''UMAPI service temporarily unavailable (attempt 1) -- 429
0 commit comments