fix: Drops check for .json when testing a single file (bluealloy#1301)

taikoxyz · Apr 13, 2024 · 5d25c4d · 5d25c4d
1 parent 7957c02
commit 5d25c4d
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/bins/revme/src/cmd/statetest/runner.rs b/bins/revme/src/cmd/statetest/runner.rs
@@ -60,12 +60,16 @@ pub enum TestErrorKind {
 }
 
 pub fn find_all_json_tests(path: &Path) -> Vec<PathBuf> {
-    WalkDir::new(path)
-        .into_iter()
-        .filter_map(Result::ok)
-        .filter(|e| e.path().extension() == Some("json".as_ref()))
-        .map(DirEntry::into_path)
-        .collect()
+    if path.is_file() {
+        vec![path.to_path_buf()]
+    } else {
+        WalkDir::new(path)
+            .into_iter()
+            .filter_map(Result::ok)
+            .filter(|e| e.path().extension() == Some("json".as_ref()))
+            .map(DirEntry::into_path)
+            .collect()
+    }
 }
 
 fn skip_test(path: &Path) -> bool {