[incubator-ponymail-unit-tests] branch master updated: Allow generator to work with earlier versions

[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 dae7474  Allow generator to work with earlier versions
dae7474 is described below

commit dae74749fa3288c6511240eea026cb9d522e8873
Author: Sebb <se...@apache.org>
AuthorDate: Sun Aug 23 22:49:30 2020 +0100

    Allow generator to work with earlier versions
---
 tests/test-generators.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/tests/test-generators.py b/tests/test-generators.py
index 3224b5c..0b6875a 100755
--- a/tests/test-generators.py
+++ b/tests/test-generators.py
@@ -21,9 +21,13 @@ def generate_specs(args):
     try:
         import generators
     except:
-        import plugins.generators as generators
+        try:
+            import plugins.generators as generators
+        except:
+            generators=None
     yml = {}
-    for gen_type in generators.generator_names():
+    generator_names = generators.generator_names() if hasattr(generators, 'generator_names') else ['full', 'medium', 'cluster', 'legacy']
+    for gen_type in generator_names:
         test_args = collections.namedtuple('testargs', ['parse_html', 'generator'])(parse_html, gen_type)
         archie = interfacer.Archiver(archiver, test_args)
         sys.stderr.write("Generating specs for type '%s'...\n" % gen_type)