From fcfcba98fbbabbaf2f0d897d9809e041384d3f0f Mon Sep 17 00:00:00 2001
From: Erich Gubler <erichdongubler@gmail.com>
Date: Thu, 11 Apr 2024 16:49:06 -0400
Subject: [PATCH] feat(fmt): apply timeout tainting

---
 moz-webgpu-cts/src/main.rs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/moz-webgpu-cts/src/main.rs b/moz-webgpu-cts/src/main.rs
index 4bd2885..8bb1b20 100644
--- a/moz-webgpu-cts/src/main.rs
+++ b/moz-webgpu-cts/src/main.rs
@@ -800,7 +800,17 @@ fn run(cli: Cli) -> ExitCode {
                         fmt_err_found = true;
                         render_metadata_parse_errors(&path, &file_contents, errors);
                     }
-                    Ok(file) => {
+                    Ok(mut file) => {
+                        for (_name, test) in &mut file.tests {
+                            for (_name, subtest) in &mut test.subtests {
+                                if let Some(expected) = subtest.properties.expectations.as_mut() {
+                                    for (_, expected) in expected.iter_mut() {
+                                        taint_subtest_timeouts_by_suspicion(expected);
+                                    }
+                                }
+                            }
+                        }
+
                         match write_to_file(&path, metadata::format_file(&file)) {
                             Ok(()) => (),
                             Err(AlreadyReportedToCommandline) => {