Skip to content

Commit

Permalink
update default endpoint and tests
Browse files Browse the repository at this point in the history
  • Loading branch information
digizeph committed Apr 20, 2023
1 parent 4272205 commit 7f4e6cd
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 14 deletions.
2 changes: 1 addition & 1 deletion bgpkit/bgpkit_broker.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ class BrokerItem:

class Broker:

def __init__(self, api_url: str = "https://api.broker.bgpkit.com/v2", page_size: int = 100, verify=True):
def __init__(self, api_url: str = "https://api.bgpkit.com/broker", page_size: int = 100, verify=True):
self.base_url = api_url.strip()
self.page_size = int(page_size)
self.verify = verify
Expand Down
21 changes: 8 additions & 13 deletions bgpkit/test_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,30 +13,25 @@ def test_parser(self):
assert len(elems) == 4227

def test_broker(self):
# filter by only time
broker = bgpkit.Broker()
items = broker.query(ts_start="1643760000", ts_end="2022-02-02T00:20:00")
for item in items:
print(json.dumps(item.__dict__))
print(len(items))
assert len(items) == 290

# elems = bgpkit.Parser(items[0].url).parse_all()
# assert len(elems) == 222467

def test_broker_2(self):
# filter by both time and collector
broker = bgpkit.Broker()
items = broker.query(ts_start="1643760000", ts_end="2022-02-02T00:20:00", collector_id="rrc00")
for item in items:
print(json.dumps(item.__dict__))
print(len(items))
assert len(items) == 7

# specify API endpoint and filter by time string with timezones (+ and - zones)
broker = bgpkit.Broker("https://api.bgpkit.com/broker")
items = broker.query(ts_start="2022-02-02T00:00:00-00:00", ts_end="2022-02-02T00:20:00.123000+00:00",
collector_id="rrc00")
assert len(items) == 7

def test_broker_no_verify(self):
broker = bgpkit.Broker(verify=False)
items = broker.query(ts_start="1643760000", ts_end="2022-02-02T00:20:00", collector_id="rrc00")
for item in items:
print(json.dumps(item.__dict__))
print(len(items))
assert len(items) == 7

def test_roas(self):
Expand Down

0 comments on commit 7f4e6cd

Please sign in to comment.