system-probe: tasks: Save all dockers from docker-compose files in th…

…e protocols dir (#14873) * system-probe: tasks: Save all dockers from docker-compose files in the protocols dir * Fixed lint
DataDog · Dec 28, 2022 · bb428cb · bb428cb
1 parent e4e7d4b
commit bb428cb
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/tasks/system_probe.py b/tasks/system_probe.py
@@ -1227,12 +1227,11 @@ def save_test_dockers(ctx, output_dir, arch, windows=is_windows):
     if windows:
         return
 
-    docker_compose_paths = (
-        # Add relative docker-compose paths
-        # For example:
-        # "./pkg/network/protocols/dockers/testdata/docker-compose.yml",
-        "./pkg/network/protocols/amqp/testdata/docker-compose.yml",
-    )
+    docker_compose_paths = glob.glob("./pkg/network/protocols/*/testdata/docker-compose.yml")
+    # Add relative docker-compose paths
+    # For example:
+    #   docker_compose_paths.append("./pkg/network/protocols/dockers/testdata/docker-compose.yml")
+
     images = set()
     for docker_compose_path in docker_compose_paths:
         with open(docker_compose_path, "r") as f: