def test_api_rate_limit_exceeded_outside_window():
# When I make an API request with a rate limit and I am have a previous
# request outside of the time window
def foobar():
pass
c = Client(account_id="foo", api_key="bar")
c.foobar_window = datetime.datetime.now() - datetime.timedelta(seconds=61)
# Then I should not receive a API Rate Limit Timeout
c._api_rate_limit_exceeded(foobar, window=60).should.be.false
def test_api_rate_limit_exceeded_outside_window():
# When I make an API request with a rate limit and I am have a previous
# request outside of the time window
def foobar():
pass
c = Client(account_id="foo", api_key="bar")
c.foobar_window = datetime.datetime.now() - datetime.timedelta(seconds=61)
# Then I should not receive a API Rate Limit Timeout
c._api_rate_limit_exceeded(foobar, window=60).should.be.false
def test_api_rate_limit_exceeded_inside_window():
# When I make an API request with a rate limit and I am have a previous
# request inside of the time window
def foobar():
pass
c = Client(account_id="foo", api_key="bar")
c.foobar_window = datetime.datetime.now() - datetime.timedelta(seconds=59)
# Then I should receive a wait time
c._api_rate_limit_exceeded.when.called_with(foobar, window=60)\
.should.throw(NewRelicApiRateLimitException)
def test_api_rate_limit_exceeded_inside_window():
# When I make an API request with a rate limit and I am have a previous
# request inside of the time window
def foobar():
pass
c = Client(account_id="foo", api_key="bar")
c.foobar_window = datetime.datetime.now() - datetime.timedelta(seconds=59)
# Then I should receive a wait time
c._api_rate_limit_exceeded.when.called_with(foobar, window=60)\
.should.throw(NewRelicApiRateLimitException)
