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'