From 1b637277456caba72c84457a0c6f285532e4901f Mon Sep 17 00:00:00 2001 From: Lorenzo Venerandi Date: Sun, 1 Mar 2026 17:43:13 +0100 Subject: [PATCH] fix: Improve command-line argument parsing in test_insert_fake_ips.py --- tests/test_insert_fake_ips.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/test_insert_fake_ips.py b/tests/test_insert_fake_ips.py index 1eba765..a9a3ea2 100644 --- a/tests/test_insert_fake_ips.py +++ b/tests/test_insert_fake_ips.py @@ -548,12 +548,13 @@ def generate_fake_data( if __name__ == "__main__": import sys - # Allow command-line arguments for customization - num_ips = int(sys.argv[1]) if len(sys.argv) > 1 else 20 - logs_per_ip = int(sys.argv[2]) if len(sys.argv) > 2 else 15 - credentials_per_ip = int(sys.argv[3]) if len(sys.argv) > 3 else 3 # Add --no-cleanup flag to skip database cleanup cleanup = "--no-cleanup" not in sys.argv + # Filter out flags before parsing positional args + positional = [a for a in sys.argv[1:] if not a.startswith("--")] + num_ips = int(positional[0]) if len(positional) > 0 else 20 + logs_per_ip = int(positional[1]) if len(positional) > 1 else 15 + credentials_per_ip = int(positional[2]) if len(positional) > 2 else 3 generate_fake_data( num_ips,