From dc47656732d23dac86b0beb9e161ae23afac842c Mon Sep 17 00:00:00 2001
From: "alon.dotan" <alon.dotan@starkware.co>
Date: Tue, 2 Jul 2024 11:43:45 +0300
Subject: [PATCH] Fix(CI): Fix run tests flow

---
 crates/papyrus_storage/src/bin/README.md |  2 --
 scripts/run_tests.py                     | 39 ++++++++++++++++--------
 2 files changed, 27 insertions(+), 14 deletions(-)

diff --git a/crates/papyrus_storage/src/bin/README.md b/crates/papyrus_storage/src/bin/README.md
index 72496f4d19b..4b6e68fdf95 100644
--- a/crates/papyrus_storage/src/bin/README.md
+++ b/crates/papyrus_storage/src/bin/README.md
@@ -37,5 +37,3 @@ This tool allows you to dump the entire `declared_classes` table from Papyrus st
    ```
 
    The default value for file_path is `dump_declared_classes.json`.
-
-
diff --git a/scripts/run_tests.py b/scripts/run_tests.py
index 4cdd0647308..f497e33e944 100755
--- a/scripts/run_tests.py
+++ b/scripts/run_tests.py
@@ -1,5 +1,6 @@
 #!/bin/env python3
-from ast import arg
+
+import argparse
 import re
 import subprocess
 import os
@@ -59,21 +60,35 @@ def get_package_dependencies(package_name: str) -> Set[str]:
     return deps
 
 
-def run_test():
+def run_test(changes_only: bool):
     local_changes = get_local_changes(".")
     modified_packages = get_modified_packages(local_changes)
-    for p in modified_packages:
-        deps = get_package_dependencies(p)
-        print(f"Running tests for {deps}")
-        args = []
-        for d in deps:
-            args.extend(["--package", d])
+    args = []
+    if changes_only:
+        for p in modified_packages:
+            deps = get_package_dependencies(p)
+            print(f"Running tests for {deps}")
+            for d in deps:
+                args.extend(["--package", d])
+        if len(args) == 0:
+            print("No changes detected.")
+            return
     cmd = ["cargo", "test"] + args
+    print("Running tests...")
     print(cmd)
     subprocess.run(cmd)
+    print("Tests complete.")
+
+def parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(description="Presubmit script.")
+    parser.add_argument("--changes_only", action="store_true")
+    return parser.parse_args()
+
+
+def main():
+    args = parse_args()
+    run_test(changes_only=args.changes_only)
+
 
 if __name__ == "__main__":
-    print("Running tests...")
-    run_test()
-    # Run tests here
-    print("Tests complete.")
+    main()