Skip to content

Commit 7a12d5c

Browse files
authored
refactor: update delegates method arguments (#132)
1 parent 95090e8 commit 7a12d5c

2 files changed

Lines changed: 40 additions & 12 deletions

File tree

client/api/delegates.py

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -15,16 +15,8 @@ def all(self, page=None, limit=100, **kwargs):
1515
def get(self, delegate_id):
1616
return self.with_endpoint('api').request_get(f'delegates/{delegate_id}')
1717

18-
def blocks(self, delegate_id, page=None, limit=100):
19-
params = {
20-
'page': page,
21-
'limit': limit,
22-
}
23-
return self.with_endpoint('api').request_get(f'delegates/{delegate_id}/blocks', params)
18+
def blocks(self, delegate_id, **kwargs):
19+
return self.with_endpoint('api').request_get(f'delegates/{delegate_id}/blocks', kwargs)
2420

25-
def voters(self, delegate_id, page=None, limit=100):
26-
params = {
27-
'page': page,
28-
'limit': limit,
29-
}
30-
return self.with_endpoint('api').request_get(f'delegates/{delegate_id}/voters', params)
21+
def voters(self, delegate_id, **kwargs):
22+
return self.with_endpoint('api').request_get(f'delegates/{delegate_id}/voters', kwargs)

tests/api/test_delegates.py

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,3 +66,39 @@ def test_get_calls_correct_url():
6666

6767
assert len(responses.calls) == 1
6868
assert responses.calls[0].request.url == 'http://127.0.0.1:4002/api/delegates/12345'
69+
70+
71+
def test_blocks_calls_correct_url():
72+
delegate_id = '12345'
73+
responses.add(
74+
responses.GET,
75+
'http://127.0.0.1:4002/api/delegates/{}/blocks'.format(delegate_id),
76+
json={'success': True},
77+
status=200
78+
)
79+
80+
client = ArkClient('http://127.0.0.1:4002/api')
81+
client.delegates.blocks(delegate_id, limit=100, orderBy='timestamp:desc')
82+
83+
assert len(responses.calls) == 1
84+
assert responses.calls[0].request.url.startswith('http://127.0.0.1:4002/api/delegates/12345/blocks?')
85+
assert 'limit=100' in responses.calls[0].request.url
86+
assert 'orderBy=timestamp%3Adesc' in responses.calls[0].request.url
87+
88+
89+
def test_voters_calls_correct_url():
90+
delegate_id = '12345'
91+
responses.add(
92+
responses.GET,
93+
'http://127.0.0.1:4002/api/delegates/{}/voters'.format(delegate_id),
94+
json={'success': True},
95+
status=200
96+
)
97+
98+
client = ArkClient('http://127.0.0.1:4002/api')
99+
client.delegates.voters(delegate_id, limit=100, orderBy='timestamp:desc')
100+
101+
assert len(responses.calls) == 1
102+
assert responses.calls[0].request.url.startswith('http://127.0.0.1:4002/api/delegates/12345/voters?')
103+
assert 'limit=100' in responses.calls[0].request.url
104+
assert 'orderBy=timestamp%3Adesc' in responses.calls[0].request.url

0 commit comments

Comments
 (0)