tests/gen-tests-mk: allow overriding the set of attacks via --attacks

Fixes #2.

Signed-off-by: Tim Janik <timj@gnu.org>

Author: tim-janik

tests/gen-tests-mk

@@ -127,6 +127,7 @@ def main():
   parser.add_argument ("--seed", type = int, default = None, help = "Seed for random number generators")
   parser.add_argument ("--without-original", action = "store_true", help = "Disable use of original image for detection.")
   parser.add_argument ("--cornersync", choices = ['on', 'off', 'auto'], default = 'auto', help = "Set cornersync aided detection mode (on|off|auto).")
+  parser.add_argument ("--attacks", type = str, default = 'all', help = "Comma-separated list of attacks to run.")
   args = parser.parse_args()
   seed = args.seed if args.seed is not None else random.randint (0, 2**32 - 1)
   random.seed (seed)
@@ -151,6 +152,8 @@ def main():
     content = re.sub (r'^SRCDIR\s*\?=.*', f'SRCDIR := {SRCDIR}', content, 0, re.M)
     content = re.sub (r'^WITHOUT_ORIGINAL\s*\?=.*', f'WITHOUT_ORIGINAL := {str (args.without_original).lower()}', content, 0, re.M)
     content = re.sub (r'^CORNERSYNC_MODE\s*\?=.*', f'CORNERSYNC_MODE := {str (args.cornersync).lower()}', content, 0, re.M)
+    if args.attacks and 'all' != args.attacks:
+      content = re.sub (r'^(ATTACKS\s*\?=.*)', f'# \\1\nATTACKS ?= { " ".join (args.attacks.split (",")) }', content, 0, re.M)
     dst.write (content)
 
   # crawl imagedir for files with an image extension