Skip to content

Commit ede7b83

Browse files
committed
refactor: remove whitelist endpoints (TBD)
1 parent 258231f commit ede7b83

2 files changed

Lines changed: 1 addition & 76 deletions

File tree

client/api/tokens.py

Lines changed: 1 addition & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -3,24 +3,13 @@
33

44
class Tokens(Resource):
55

6-
def all(self, page=None, limit=100, **kwargs):
7-
extra_params = {k: v for k, v in kwargs.items() if v is not None}
6+
def all(self, page=None, limit=100):
87
params = {
98
'page': page,
109
'limit': limit,
11-
**extra_params
1210
}
1311
return self.with_endpoint('api').request_get('tokens', params)
1412

15-
def all_with_whitelist(self, whitelist, page=None, limit=100):
16-
params = {
17-
'page': page,
18-
'limit': limit,
19-
}
20-
return self.with_endpoint('api').request_post(
21-
'tokens', data={'whitelist': whitelist}, params=params
22-
)
23-
2413
def get(self, address):
2514
return self.with_endpoint('api').request_get(
2615
f'tokens/{address}'
@@ -53,5 +42,3 @@ def all_transfers(self, page=None, limit=100):
5342
'tokens/transfers', params
5443
)
5544

56-
def whitelist(self):
57-
return self.with_endpoint('api').request_get('tokens/whitelist')

tests/api/test_tokens.py

Lines changed: 0 additions & 62 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
import json
2-
31
import responses
42

53
from client import ArkClient
@@ -35,52 +33,6 @@ def test_all_calls_correct_url_with_passed_in_params():
3533
assert 'limit=69' in responses.calls[0].request.url
3634

3735

38-
def test_all_calls_correct_url_with_ignore_whitelist():
39-
responses.add(
40-
responses.GET,
41-
'http://127.0.0.1:4002/api/tokens',
42-
json={'success': True},
43-
status=200
44-
)
45-
46-
client = ArkClient('http://127.0.0.1:4002/api')
47-
client.tokens.all(ignoreWhitelist=True)
48-
assert len(responses.calls) == 1
49-
assert 'ignoreWhitelist=True' in responses.calls[0].request.url
50-
51-
52-
def test_all_with_whitelist_calls_correct_url():
53-
responses.add(
54-
responses.POST,
55-
'http://127.0.0.1:4002/api/tokens',
56-
json={'success': True},
57-
status=200
58-
)
59-
60-
client = ArkClient('http://127.0.0.1:4002/api')
61-
whitelist = ['0xabc', '0xdef']
62-
client.tokens.all_with_whitelist(whitelist)
63-
assert len(responses.calls) == 1
64-
assert responses.calls[0].request.url == 'http://127.0.0.1:4002/api/tokens?limit=100'
65-
body = json.loads(responses.calls[0].request.body)
66-
assert body == {'whitelist': ['0xabc', '0xdef']}
67-
68-
69-
def test_all_with_whitelist_calls_correct_url_with_params():
70-
responses.add(
71-
responses.POST,
72-
'http://127.0.0.1:4002/api/tokens',
73-
json={'success': True},
74-
status=200
75-
)
76-
77-
client = ArkClient('http://127.0.0.1:4002/api')
78-
client.tokens.all_with_whitelist(['0xabc'], page=2, limit=50)
79-
assert len(responses.calls) == 1
80-
assert 'page=2' in responses.calls[0].request.url
81-
assert 'limit=50' in responses.calls[0].request.url
82-
83-
8436
def test_get_calls_correct_url():
8537
address = '0x1234567890abcdef'
8638
responses.add(
@@ -204,17 +156,3 @@ def test_all_transfers_calls_correct_url_with_passed_in_params():
204156
assert 'limit=10' in responses.calls[0].request.url
205157

206158

207-
def test_whitelist_calls_correct_url():
208-
responses.add(
209-
responses.GET,
210-
'http://127.0.0.1:4002/api/tokens/whitelist',
211-
json={'success': True},
212-
status=200
213-
)
214-
215-
client = ArkClient('http://127.0.0.1:4002/api')
216-
client.tokens.whitelist()
217-
assert len(responses.calls) == 1
218-
assert responses.calls[0].request.url == (
219-
'http://127.0.0.1:4002/api/tokens/whitelist'
220-
)

0 commit comments

Comments
 (0)