1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
# frozen_string_literal: true
require 'octokit/rate_limit'
describe Octokit::RateLimit do
describe '.from_response' do
let(:response_headers) do
{
'X-RateLimit-Limit' => 60,
'X-RateLimit-Remaining' => 42,
'X-RateLimit-Reset' => (Time.now + 60).to_i
}
end
let(:response) { double('response') }
it 'parses rate limit info from response headers' do
expect(response).to receive(:response_headers)
.at_least(:once)
.and_return(response_headers)
info = described_class.from_response(response)
expect(info.limit).to eq(60)
expect(info.remaining).to eq(42)
expect(info.resets_in).to eq(59)
expect(info.resets_at).to be_kind_of(Time)
end
it 'returns a positive rate limit for Enterprise' do
expect(response).to receive(:response_headers)
.at_least(:once)
.and_return({})
info = described_class.from_response(response)
expect(info.limit).to eq(1)
expect(info.remaining).to eq(1)
expect(info.resets_in).to eq(0)
expect(info.resets_at).to be_kind_of(Time)
end
it 'handles nil responses' do
info = described_class.from_response(nil)
expect(info.limit).to be_nil
expect(info.remaining).to be_nil
expect(info.resets_in).to be_nil
expect(info.resets_at).to be_nil
end
it 'handles resets_in time in past' do
expect(response).to receive(:response_headers)
.at_least(:once)
.and_return(
response_headers.merge('X-RateLimit-Reset' => (Time.now - 60).to_i)
)
info = described_class.from_response(response)
expect(info.limit).to eq(60)
expect(info.remaining).to eq(42)
expect(info.resets_in).to eq(0)
expect(info.resets_at).to be_kind_of(Time)
end
end
end
|
