Update unittest

Author: rickwu666666

contrib/checkbox-ce-oem/checkbox-provider-ce-oem/tests/test_gst_v4l2_video_decoder_md5_checksum_comparison.py

@@ -174,8 +174,12 @@ def tearDownClass(cls):
         new_callable=mock_open,
         read_data="golden_md5_checksum",
     )
+    @patch(
+        "gst_v4l2_video_decoder_md5_checksum_comparison.manage_test_file_by_name"
+    )
     def test_validate_video_decoder_md5_checksum_success(
         self,
+        mock_manage_test_file_by_name,
         mock_open,
         mock_get_md5_checksum_from_command,
         mock_build_gst_command,
@@ -196,7 +200,7 @@ def test_validate_video_decoder_md5_checksum_success(
         mock_get_md5_checksum_from_command.return_value = "golden_md5_checksum"
         mock_build_gst_command.return_value = "my testing command"
 
-        validate_video_decoder_md5_checksum(args)
+        validate_video_decoder_md5_checksum(args) # type: ignore
 
         mock_os_path_exists.assert_has_calls(
             [call("golden_sample.mp4"), call("my_test.md5")]
@@ -214,8 +218,12 @@ def test_validate_video_decoder_md5_checksum_success(
         "get_md5_checksum_from_command"
     )
     @patch("builtins.open", new_callable=mock_open)
+    @patch(
+        "gst_v4l2_video_decoder_md5_checksum_comparison.manage_test_file_by_name"
+    )
     def test_validate_video_decoder_md5_checksum_golden_sample_not_found(
         self,
+        mock_manage_test_file_by_name,
         mock_open,
         mock_get_md5_checksum_from_command,
         mock_os_path_exists,
@@ -236,9 +244,9 @@ def test_validate_video_decoder_md5_checksum_golden_sample_not_found(
 
         # Act and Assert
         with self.assertRaises(SystemExit) as cm:
-            validate_video_decoder_md5_checksum(args)
+            validate_video_decoder_md5_checksum(args) # type: ignore
         self.assertEqual(
-            cm.exception.code,
+            str(cm.exception),
             "Golden Sample 'non_exist_golden_sample.mp4' doesn't exist",
         )
         mock_os_path_exists.assert_has_calls(
@@ -253,8 +261,12 @@ def test_validate_video_decoder_md5_checksum_golden_sample_not_found(
         "get_md5_checksum_from_command"
     )
     @patch("builtins.open", new_callable=mock_open)
+    @patch(
+        "gst_v4l2_video_decoder_md5_checksum_comparison.manage_test_file_by_name"
+    )
     def test_validate_video_decoder_md5_checksum_golden_md5_checksum_not_found(
         self,
+        mock_manage_test_file_by_name,
         mock_open,
         mock_get_md5_checksum_from_command,
         mock_os_path_exists,
@@ -275,16 +287,16 @@ def test_validate_video_decoder_md5_checksum_golden_md5_checksum_not_found(
         mock_os_path_exists.side_effect = [True, False]
 
         with self.assertRaises(SystemExit) as cm:
-            validate_video_decoder_md5_checksum(args)
+            validate_video_decoder_md5_checksum(args) # type: ignore
         self.assertEqual(
-            cm.exception.code,
+            str(cm.exception),
             (
                 "Golden Sample's MD5 checksum 'non_exist_golden_sample.md5'"
                 " doesn't exist"
             ),
         )
         mock_os_path_exists.assert_has_calls(
-            [call("golden_sample.mp4"), call("non_exist_golden_sample.md5")]
+            [call("non_exist_golden_sample.md5")]
         )
         mock_get_md5_checksum_from_command.assert_not_called()
         mock_open.assert_not_called()
@@ -302,8 +314,12 @@ def test_validate_video_decoder_md5_checksum_golden_md5_checksum_not_found(
         new_callable=mock_open,
         read_data="different_golden_md5_checksum",
     )
+    @patch(
+        "gst_v4l2_video_decoder_md5_checksum_comparison.manage_test_file_by_name"
+    )
     def test_validate_video_decoder_md5_checksum_failure(
         self,
+        mock_manage_test_file_by_name,
         mock_open,
         mock_get_md5_checksum_from_command,
         mock_build_gst_command,
@@ -326,12 +342,12 @@ def test_validate_video_decoder_md5_checksum_failure(
 
         # Act and Assert
         with self.assertRaises(SystemExit) as cm:
-            validate_video_decoder_md5_checksum(args)
+            validate_video_decoder_md5_checksum(args) # type: ignore
         self.assertEqual(
             cm.exception.code,
             "Failed. MD5 checksum is not same as Golden Sample",
         )
-        mock_os_path_exists.assert_has_calls(
+        mock_os_path_exists.assert_any_call(
             [call("golden_sample.mp4"), call("golden_sample.md5")]
         )
         mock_get_md5_checksum_from_command.assert_called_once_with(