X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2Fautogen%2FRegenerator.py;h=30dc7b7e89252147aa037c480116b244509c99a7;hp=85e8b60dd5d7ee989b9e3973da66498d6c1a50f5;hb=f1ec30ef61d430fb47d2e1a8bcc1305975500699;hpb=ac6cc22910ea0d3bda42c227d799f2d4e89bc9d0

diff --git a/resources/libraries/python/autogen/Regenerator.py b/resources/libraries/python/autogen/Regenerator.py
index 85e8b60dd5..30dc7b7e89 100644
--- a/resources/libraries/python/autogen/Regenerator.py
+++ b/resources/libraries/python/autogen/Regenerator.py
@@ -56,6 +56,13 @@ class Regenerator(object):
         :type tc_kwargs_list: list of tuple or None
         """
 
+        def get_suite_id(filename):
+            dash_split = filename.split("-", 1)
+            if len(dash_split[0]) <= 4:
+                # It was something like "2n1l", we need one more split.
+                dash_split = dash_split[1].split("-", 1)
+            return dash_split[1].split(".", 1)[0]
+
         def add_testcase(file_out, num, **kwargs):
             file_out.write(testcase.generate(num=num, **kwargs))
             return num + 1
@@ -86,7 +93,7 @@ class Regenerator(object):
                 text = file_in.read()
                 text_prolog = "".join(text.partition("*** Test Cases ***")[:-1])
             # TODO: Make the following work for 2n suites.
-            suite_id = filename.split("-", 1)[1].split(".", 1)[0]
+            suite_id = get_suite_id(filename)
             print "Regenerating suite_id:", suite_id
             testcase = self.testcase_class(suite_id)
             with open(filename, "w") as file_out: