def test_get_hid(input_data, expected_output):
test_fw = Firmware(binary=b'foo')
test_fw.set_device_name('test_device')
test_fw.set_vendor('foo')
test_fw.set_firmware_version('1.0')
test_fw.set_part_name(input_data)
assert test_fw.get_hid() == expected_output
def test_get_hid(input_data, expected_output):
test_fw = Firmware(binary=b'foo')
test_fw.device_name = 'test_device'
test_fw.vendor = 'foo'
test_fw.version = '1.0'
test_fw.set_part_name(input_data)
assert test_fw.get_hid() == expected_output
def _init_current_analysis(fw_object: Firmware):
return {
'files_to_unpack': list(fw_object.files_included),
'files_to_analyze': [fw_object.uid],
'start_time': time(),
'unpacked_files_count': 1,
'analyzed_files_count': 0,
'total_files_count': 1 + len(fw_object.files_included),
'hid': fw_object.get_hid(),
}
def test_get_hid(self):
test_fw = Firmware(binary=b'foo')
test_fw.set_device_name('test_device')
test_fw.set_vendor('foo')
test_fw.set_firmware_version('1.0')
self.assertEqual(test_fw.get_hid(), 'foo test_device - 1.0', 'hid not correct')
def test_get_hid():
test_fw = Firmware(binary=b'foo')
test_fw.set_device_name('test_device')
test_fw.set_vendor('foo')
test_fw.set_firmware_version('1.0')
assert test_fw.get_hid() == 'foo test_device - 1.0'