def test_before_request(self):
headers = {"other": "header-value"}
request = self.make_mock_request(status=http_client.OK,
data=SUCCESS_RESPONSE)
credentials = self.make_credentials()
# First call should call refresh, setting the token.
credentials.before_request(request, "POST", "https://example.com/api",
headers)
assert headers == {
"other": "header-value",
"authorization":
"Bearer {}".format(SUCCESS_RESPONSE["access_token"]),
}
# Second call shouldn't call refresh (request should be untouched).
credentials.before_request(mock.sentinel.request, "POST",
"https://example.com/api", headers)
assert headers == {
"other": "header-value",
"authorization":
"Bearer {}".format(SUCCESS_RESPONSE["access_token"]),
}
def test_before_request():
credentials = CredentialsImpl()
request = "token"
headers = {}
# First call should call refresh, setting the token.
credentials.before_request(request, "http://example.com", "GET", headers)
assert credentials.valid
assert credentials.token == "token"
assert headers["authorization"] == "Bearer token"
request = "token2"
headers = {}
# Second call shouldn't call refresh.
credentials.before_request(request, "http://example.com", "GET", headers)
assert credentials.valid
assert credentials.token == "token"
assert headers["authorization"] == "Bearer token"
def test_before_request():
credentials = CredentialsImpl()
request = 'token'
headers = {}
# First call should call refresh, setting the token.
credentials.before_request(request, 'http://example.com', 'GET', headers)
assert credentials.valid
assert credentials.token == 'token'
assert headers['authorization'] == 'Bearer token'
request = 'token2'
headers = {}
# Second call shouldn't call refresh.
credentials.before_request(request, 'http://example.com', 'GET', headers)
assert credentials.valid
assert credentials.token == 'token'
assert headers['authorization'] == 'Bearer token'
def test_before_request_expired(self, utcnow):
headers = {}
request = self.make_mock_request(status=http_client.OK,
data=SUCCESS_RESPONSE)
credentials = self.make_credentials()
credentials.token = "token"
utcnow.return_value = datetime.datetime.min
# Set the expiration to one second more than now plus the clock skew
# accommodation. These credentials should be valid.
credentials.expiry = (datetime.datetime.min +
_helpers.REFRESH_THRESHOLD +
datetime.timedelta(seconds=1))
assert credentials.valid
assert not credentials.expired
credentials.before_request(request, "POST", "https://example.com/api",
headers)
# Cached token should be used.
assert headers == {"authorization": "Bearer token"}
# Next call should simulate 1 second passed.
utcnow.return_value = datetime.datetime.min + datetime.timedelta(
seconds=1)
assert not credentials.valid
assert credentials.expired
credentials.before_request(request, "POST", "https://example.com/api",
headers)
# New token should be retrieved.
assert headers == {
"authorization":
"Bearer {}".format(SUCCESS_RESPONSE["access_token"])
}