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])
