2626import pytest
2727from six import StringIO
2828
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
3232
3333
3434# This method will be used by the mock to replace UMAPI.users
@@ -59,53 +59,53 @@ def __init__(self, status_code, headers):
5959
6060
6161# 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 )
6363def test_helper_success (_ ):
6464 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" ]
6666
6767
6868@pytest .fixture
6969def reduce_attempts ():
7070 # reduce the number of retry attempts
71- adobe_umapi_client .helper .retry_max_attempts = 3
71+ umapi_client .helper .retry_max_attempts = 3
7272
7373
7474# 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 )
7676def test_helper_retry_seconds (_ , reduce_attempts ):
7777 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" ]
7979
8080
8181# 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 )
8383def test_helper_fail_date (_ , reduce_attempts ):
8484 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" ]
8686
8787
8888# the default retry waits are really long, so don't use them while testing
8989@pytest .fixture
9090def reduce_delay ():
9191 # 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
9595
9696
9797# 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 )
9999def test_helper_retry_zero (_ , reduce_delay , reduce_attempts ):
100100 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" ]
102102
103103
104104# 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 )
106106def test_helper_retry_noheader (_ , reduce_delay , reduce_attempts ):
107107 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" ]
109109
110110
111111# py.test doesn't divert string logging, so use it
@@ -116,18 +116,18 @@ def log_stream():
116116 logger = logging .getLogger ('test_logger' )
117117 logger .setLevel (logging .WARNING )
118118 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
121121 yield stream
122- adobe_umapi_client .helper .logger = prior_logger
122+ umapi_client .helper .logger = prior_logger
123123 handler .close ()
124124
125125
126126# 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 )
128128def test_helper_retry_logging (_ , log_stream , reduce_attempts ):
129129 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" ]
131131 log_stream .flush ()
132132 log = log_stream .getvalue () # save as a local so can do pytest -l to see exact log
133133 assert log == '''UMAPI service temporarily unavailable (attempt 1) -- 429
0 commit comments