def test_refresh_token_passes_refresh_token_as_header(self, patch_post): post = patch_post({ "data": { "refreshToken": { "accessToken": "ACCESS_TOKEN", "expiresAt": "2100-01-01", "refreshToken": "REFRESH_TOKEN", } } }) client = Client() client._refresh_token = "refresh" client._refresh_access_token() assert post.call_args[1]["headers"] == dict( Authorization="Bearer refresh")
def test_refresh_token_passes_access_token_as_arg(self, patch_post): post = patch_post({ "data": { "refreshToken": { "accessToken": "ACCESS_TOKEN", "expiresAt": "2100-01-01", "refreshToken": "REFRESH_TOKEN", } } }) client = Client() client._access_token = "access" client._refresh_access_token() variables = json.loads(post.call_args[1]["json"]["variables"]) assert variables["input"]["accessToken"] == "access"
def test_refresh_token_passes_refresh_token_as_header(self, patch_post): post = patch_post({ "data": { "refresh_token": { "access_token": "ACCESS_TOKEN", "expires_at": "2100-01-01", "refresh_token": "REFRESH_TOKEN", } } }) client = Client() client._refresh_token = "refresh" client._refresh_access_token() assert post.call_args[1]["headers"] == { "Authorization": "Bearer refresh", "X-PREFECT-CORE-VERSION": str(prefect.__version__), }
def test_refresh_token_sets_attributes(self, patch_post): patch_post({ "data": { "refreshToken": { "accessToken": "ACCESS_TOKEN", "expiresIn": 600, "refreshToken": "REFRESH_TOKEN", } } }) client = Client() assert client._access_token is None assert client._refresh_token is None assert client._access_token_expires_at < pendulum.now() client._refresh_access_token() assert client._access_token is "ACCESS_TOKEN" assert client._refresh_token is "REFRESH_TOKEN" assert client._access_token_expires_at > pendulum.now().add( seconds=599)
def test_refresh_token_sets_attributes(self, patch_post): patch_post({ "data": { "refreshToken": { "accessToken": "ACCESS_TOKEN", "expiresAt": "2100-01-01", "refreshToken": "REFRESH_TOKEN", } } }) client = Client() assert client._access_token is None assert client._refresh_token is None # add buffer because Windows doesn't compare milliseconds assert client._access_token_expires_at < pendulum.now().add(seconds=1) client._refresh_access_token() assert client._access_token is "ACCESS_TOKEN" assert client._refresh_token is "REFRESH_TOKEN" assert client._access_token_expires_at > pendulum.now().add( seconds=599)