@@ -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