Fix stamp file creation for zap unit test. (#27296)

unittest.main by default calls sys.exit so we did not get a chance to create the stamp file. Fix this and handle sys exit logic separately. Co-authored-by: Andrei Litvin <[email protected]>
project-chip · Jun 16, 2023 · 90f4f7c · 90f4f7c
1 parent 470dd12
commit 90f4f7c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scripts/tools/zap/test_generate.py b/scripts/tools/zap/test_generate.py
@@ -197,7 +197,9 @@ def process_arguments():
 
     PROGRAM_ARGUMENTS = process_args
 
-    unittest.main(argv=unittest_args)
+    test_results = unittest.main(argv=unittest_args, exit=False)
+    if test_results.result.failures:
+        sys.exit(1)
 
     if process_args.stamp_file:
         open(process_args.stamp_file, "wb").close()