def _extract_product_code(all_files_contents: Iterable[str]) -> Optional[str]: """Return first product code found in files contents, None if not found.""" for contents in all_files_contents: product_code = read_product_code(contents) if product_code: return product_code return None
def test_reads_product_code_from_auth_attribute(self): auth = "101000000000000000000002F7F35E602eeb0bb9b632205c51f6c357aeee7bc9" file_contents = ( '<meta http-equiv="refresh" ' f'content="0; url=http://mbed.org/start?auth={auth}&loader=11972&firmware=16457&configuration=4" />' ) self.assertEqual(read_product_code(file_contents), auth[:4])
def test_none_if_no_product_code(self): self.assertIsNone(read_product_code(""))
def test_reads_product_code_from_code_attribute(self): code = "02400201B80ECE4A45F033F2" file_contents = f'<meta http-equiv="refresh" content="0; url=http://mbed.org/device/?code={code}"/>' self.assertEqual(read_product_code(file_contents), code[:4])