Skip to content

Commit 4f622ba

Browse files
committed
add asserts to installation order dict unit test
1 parent a3a8a6e commit 4f622ba

File tree

1 file changed

+20
-1
lines changed

1 file changed

+20
-1
lines changed

src/core/tests/Test_StatusHandler.py

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -413,17 +413,36 @@ def test_installation_packages_map(self):
413413
test_packages, test_package_versions = self.__set_up_packages_func(patch_count_for_test)
414414

415415
status_handler.set_package_install_status(test_packages, test_package_versions, 'Installed', 'Other')
416+
with self.runtime.env_layer.file_system.open(self.runtime.execution_config.status_file_path, 'r') as file_handle:
417+
substatus_file_data = json.load(file_handle)[0]["status"]["substatus"][0]
418+
419+
formatted_message = json.loads(substatus_file_data['formattedMessage']['message'])
420+
self.assertEqual(len(formatted_message['patches']), 50)
421+
self.assertEqual(formatted_message['patches'][0]['classifications'], ['Other'])
422+
self.assertEqual(formatted_message['patches'][0], expected_value_other)
423+
self.assertEqual(len(status_handler._StatusHandler__installation_packages), 50)
416424
self.assertIsNotNone(status_handler._StatusHandler__installation_packages_map)
417425
self.assertEqual(status_handler._StatusHandler__installation_packages_map[patch_id_other]['name'], 'python-samba0')
418426
self.assertEqual(status_handler._StatusHandler__installation_packages_map[patch_id_other], expected_value_other)
427+
self.assertEqual(status_handler._StatusHandler__installation_packages_map[patch_id_other]['classifications'], ['Other'])
419428
self.assertEqual(len(status_handler._StatusHandler__installation_packages_map), patch_count_for_test)
420429

430+
# Update the classification from Other to Critical
421431
status_handler.set_package_install_status_classification(test_packages, test_package_versions, 'Critical')
432+
with self.runtime.env_layer.file_system.open(self.runtime.execution_config.status_file_path, 'r') as file_handle:
433+
substatus_file_data = json.load(file_handle)[0]["status"]["substatus"][0]
434+
435+
formatted_message = json.loads(substatus_file_data['formattedMessage']['message'])
436+
self.assertEqual(len(formatted_message['patches']), 50)
437+
self.assertEqual(formatted_message['patches'][0]['classifications'], ['Critical'])
438+
self.assertEqual(formatted_message['patches'][0], expected_value_critical)
439+
self.assertEqual(len(status_handler._StatusHandler__installation_packages), 50)
422440
self.assertIsNotNone(status_handler._StatusHandler__installation_packages_map)
423441
self.assertEqual(status_handler._StatusHandler__installation_packages_map[patch_id_critical]['name'], 'python-samba0')
442+
self.assertEqual(status_handler._StatusHandler__installation_packages_map[patch_id_critical]['classifications'], ['Critical'])
424443
self.assertEqual(status_handler._StatusHandler__installation_packages_map[patch_id_critical], expected_value_critical)
425444
self.assertEqual(len(status_handler._StatusHandler__installation_packages_map), patch_count_for_test)
426-
445+
427446
# Setup functions to populate packages and versions for truncation
428447
def __set_up_packages_func(self, val):
429448
test_packages = []

0 commit comments

Comments
 (0)