def test_from_web():
# only if internet is available
with contextlib.suppress(RuntimeError):
r = Registry.from_web()
assert isinstance(r, Registry)
assert len(r) > 2000
def test_from_web_with_filter():
_, pnp_id, name = PNP_DATA[0]
# only if internet is available
with contextlib.suppress(RuntimeError):
r = Registry.from_web(filter_by_id=pnp_id)
assert isinstance(r, Registry)
assert len(r) == 1
assert r.get_company_by_pnp(pnp_id) == name
def test_from_web_connection_error():
with patch('requests.get', side_effect=ConnectionError()):
with pytest.raises(RuntimeError):
Registry.from_web()
def test_from_web_bad_response():
with patch('requests.get', side_effect=HTTPError()):
r = Registry.from_web()
assert isinstance(r, Registry)
assert len(r) == 0
def test_from_web_bad_args():
with pytest.raises(TypeError):
Registry.from_web(filter_by_id=123)