Skip to content

Commit a26c00e

Browse files
committed
feat: add minBalance param to wallet tokens
1 parent 05e0ccd commit a26c00e

2 files changed

Lines changed: 32 additions & 1 deletion

File tree

client/api/wallets.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,10 +50,11 @@ def votes(self, wallet_id, page=None, limit=100):
5050
}
5151
return self.with_endpoint('api').request_get(f'wallets/{wallet_id}/votes', params)
5252

53-
def tokens(self, wallet_id, page=None, limit=100):
53+
def tokens(self, wallet_id, page=None, limit=100, min_balance=None):
5454
params = {
5555
'page': page,
5656
'limit': limit,
57+
'minBalance': min_balance,
5758
}
5859
return self.with_endpoint('api').request_get(f'wallets/{wallet_id}/tokens', params)
5960

tests/api/test_wallets.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -278,6 +278,36 @@ def test_tokens_calls_correct_url_with_passed_in_params():
278278
assert 'limit=69' in responses.calls[0].request.url
279279

280280

281+
def test_tokens_calls_correct_url_with_min_balance():
282+
wallet_id = '12345'
283+
responses.add(
284+
responses.GET,
285+
'http://127.0.0.1:4002/api/wallets/{}/tokens'.format(wallet_id),
286+
json={'success': True},
287+
status=200
288+
)
289+
290+
client = ArkClient('http://127.0.0.1:4002/api')
291+
client.wallets.tokens(wallet_id, min_balance=0.01)
292+
assert len(responses.calls) == 1
293+
assert 'minBalance=0.01' in responses.calls[0].request.url
294+
295+
296+
def test_tokens_calls_correct_url_with_min_balance_zero():
297+
wallet_id = '12345'
298+
responses.add(
299+
responses.GET,
300+
'http://127.0.0.1:4002/api/wallets/{}/tokens'.format(wallet_id),
301+
json={'success': True},
302+
status=200
303+
)
304+
305+
client = ArkClient('http://127.0.0.1:4002/api')
306+
client.wallets.tokens(wallet_id, min_balance=0)
307+
assert len(responses.calls) == 1
308+
assert 'minBalance=0' in responses.calls[0].request.url
309+
310+
281311
def test_token_addresses_calls_correct_url_with_default_params():
282312
responses.add(
283313
responses.GET,

0 commit comments

Comments
 (0)