[incubator-ponymail-unit-tests] branch master updated: Allow override of generators

[se...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

sebb pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-ponymail-unit-tests.git


The following commit(s) were added to refs/heads/master by this push:
     new 5897eee  Allow override of generators
5897eee is described below

commit 5897eee2ba96e7b794251a8ee51ed9932faea169
Author: Sebb <se...@apache.org>
AuthorDate: Mon Aug 24 17:51:49 2020 +0100

    Allow override of generators
---
 tests/test-generators.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/tests/test-generators.py b/tests/test-generators.py
index 29476b7..700d1d4 100755
--- a/tests/test-generators.py
+++ b/tests/test-generators.py
@@ -18,12 +18,15 @@ fake_args = collections.namedtuple('fakeargs', ['verbose', 'ibody'])(False, None
 
 def generate_specs(args):
     import archiver
-    try:
-        import generators
-    except:
-        import plugins.generators as generators
+    if args.generators:
+        generator_names = args.generators
+    else:
+        try:
+            import generators
+        except:
+            import plugins.generators as generators
+        generator_names = generators.generator_names() if hasattr(generators, 'generator_names') else ['full', 'medium', 'cluster', 'legacy']
     yml = {}
-    generator_names = generators.generator_names() if hasattr(generators, 'generator_names') else ['full', 'medium', 'cluster', 'legacy']
     # sort so most recent generators come last to make comparisons easier
     for gen_type in sorted(generator_names, key=lambda s: s.replace('dkim','zkim')):
         test_args = collections.namedtuple('testargs', ['parse_html', 'generator'])(parse_html, gen_type)
@@ -105,6 +108,8 @@ def main():
     parser = argparse.ArgumentParser(description='Command line options.')
     parser.add_argument('--generate', dest='generate', type=str,
                         help='Generate a test yaml spec, output to file specified here')
+    parser.add_argument('--generators', dest='generators', nargs='+', type=str,
+                        help='Override the list of generator names')
     parser.add_argument('--load', dest='load', type=str,
                         help='Load and run tests from a yaml spec file')
     parser.add_argument('--mbox', dest='mboxfile', type=str,