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_csv(tmpdir):
tmp_csv_path = os.path.join(tmpdir.dirname, 'test_import.csv')
DEFAULT_REGISTRY.to_csv(tmp_csv_path)
r = Registry.from_csv(tmp_csv_path)
assert isinstance(r, Registry)
assert r == DEFAULT_REGISTRY
def test_from_csv_with_filter(tmpdir):
_, pnp_id, name = PNP_DATA[1]
tmp_csv_path = os.path.join(tmpdir.dirname, 'test_import_with_f.csv')
DEFAULT_REGISTRY.to_csv(tmp_csv_path)
r = Registry.from_csv(tmp_csv_path, filter_by_id=pnp_id)
assert isinstance(r, Registry)
assert len(r) == 1
assert r[pnp_id] == name
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 load_pnp_info(file_dir="displaymanagement/resources/pnp_info.csv"):
"""
Loads the PNP registry from a CSV file
Parameters
----------
file_dir : str, optional
The directory of the file from which the PNP info is loaded
"""
global PNP_REGISTRY
PNP_REGISTRY = Registry.from_csv(file_dir)
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)
def test_get_comany_by_pnp_nf():
# empty registry
r = Registry()
assert r.get_company_by_pnp('foo') == 'Unknown'
def test_from_csv_bad_args():
with pytest.raises(TypeError):
Registry.from_csv(12345)
with pytest.raises(TypeError):
Registry.from_csv('/foo/bar', filter_by_id=12345)
def test_parse_edid_custom_registry():
parsed = parse_edid(BASE_HEX_EDID, registry=Registry())
assert isinstance(parsed, Edid)
assert parsed.manufacturer == 'Unknown'