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)
