From bfa1993840f5c73804980f852e41f85d7bc221e5 Mon Sep 17 00:00:00 2001
From: Ronny Pfannschmidt <ronny.pfannschmidt@redhat.com>
Date: Thu, 25 Oct 2018 18:39:54 +0200
Subject: [PATCH] add script to generate a example folder for collecting many
 files

---
 .../perf_examples/collect_stats/.gitignore    |  1 +
 .../collect_stats/generate_folders.py         | 28 +++++++++++++++++++
 .../collect_stats/template_test.py            |  2 ++
 3 files changed, 31 insertions(+)
 create mode 100644 testing/example_scripts/perf_examples/collect_stats/.gitignore
 create mode 100644 testing/example_scripts/perf_examples/collect_stats/generate_folders.py
 create mode 100644 testing/example_scripts/perf_examples/collect_stats/template_test.py

diff --git a/testing/example_scripts/perf_examples/collect_stats/.gitignore b/testing/example_scripts/perf_examples/collect_stats/.gitignore
new file mode 100644
index 000000000..1c45c2ea3
--- /dev/null
+++ b/testing/example_scripts/perf_examples/collect_stats/.gitignore
@@ -0,0 +1 @@
+foo_*
diff --git a/testing/example_scripts/perf_examples/collect_stats/generate_folders.py b/testing/example_scripts/perf_examples/collect_stats/generate_folders.py
new file mode 100644
index 000000000..94b3f013f
--- /dev/null
+++ b/testing/example_scripts/perf_examples/collect_stats/generate_folders.py
@@ -0,0 +1,28 @@
+import argparse
+import pathlib
+
+
+HERE = pathlib.Path(__file__).parent
+TEST_CONTENT = (HERE / "template_test.py").read_bytes()
+
+parser = argparse.ArgumentParser()
+parser.add_argument("numbers", nargs="*", type=int)
+
+
+def generate_folders(root, elements, *more_numbers):
+    fill_len = len(str(elements))
+    if more_numbers:
+        for i in range(elements):
+            new_folder = root.joinpath(f"foo_{i:0>{fill_len}}")
+            new_folder.mkdir()
+            new_folder.joinpath("__init__.py").write_bytes(TEST_CONTENT)
+            generate_folders(new_folder, *more_numbers)
+    else:
+        for i in range(elements):
+            new_test = root.joinpath(f"test_{i:0<{fill_len}}.py")
+            new_test.write_bytes(TEST_CONTENT)
+
+
+if __name__ == "__main__":
+    args = parser.parse_args()
+    generate_folders(HERE, *(args.numbers or (10, 100)))
diff --git a/testing/example_scripts/perf_examples/collect_stats/template_test.py b/testing/example_scripts/perf_examples/collect_stats/template_test.py
new file mode 100644
index 000000000..064ade190
--- /dev/null
+++ b/testing/example_scripts/perf_examples/collect_stats/template_test.py
@@ -0,0 +1,2 @@
+def test_x():
+    pass