From 43141ddb55c715e90b60d2fa2d9d5e01cc66b511 Mon Sep 17 00:00:00 2001
From: Paul Sokolovsky <pfalcon@users.sourceforge.net>
Date: Wed, 13 Dec 2017 18:32:25 +0200
Subject: [PATCH] tools/tinytest-codegen: Take --target= option for test set
 selection.

Gets passed to run-tests --list-tests to get actual list of tests to use.
If --target= is not given, legacy set hardcoded in tinytest-codegen itself
is used.

Also, get rid of tinytest test groups - they aren't really used for
anything, and only complicate processing. Besides, one of the next
step is to limit number of tests per a generated file to control
the binary size, which also will require "flat" list of tests.
---
 tools/tinytest-codegen.py | 28 ++++++++++++++++++++--------
 1 file changed, 20 insertions(+), 8 deletions(-)

diff --git a/tools/tinytest-codegen.py b/tools/tinytest-codegen.py
index d7761cd5ad..c6ba1867af 100755
--- a/tools/tinytest-codegen.py
+++ b/tools/tinytest-codegen.py
@@ -3,6 +3,8 @@
 import os, sys
 from glob import glob
 from re import sub
+import argparse
+
 
 def escape(s):
     s = s.decode()
@@ -17,7 +19,7 @@ def escape(s):
     return "\"\"\n\"{}\"".format(''.join([lookup[x] if x in lookup else x for x in s]))
 
 def chew_filename(t):
-    return { 'func': "test_{}_fn".format(sub(r'/|\.|-', '_', t)), 'desc': t.split('/')[1] }
+    return { 'func': "test_{}_fn".format(sub(r'/|\.|-', '_', t)), 'desc': t }
 
 def script_to_map(test_file):
     r = {"name": chew_filename(test_file)["func"]}
@@ -47,7 +49,7 @@ testgroup_struct = (
     "struct testgroup_t groups[] = {{\n{body}\n  END_OF_GROUPS\n}};"
 )
 testgroup_member = (
-    "  {{ \"{name}/\", {name}_tests }},"
+    "  {{ \"{name}\", {name}_tests }},"
 )
 
 ## XXX: may be we could have `--without <groups>` argument...
@@ -82,14 +84,24 @@ exclude_tests = (
 )
 
 output = []
+tests = []
 
-for group in test_dirs:
-    tests = [test for test in glob('{}/*.py'.format(group)) if test not in exclude_tests]
-    output.extend([test_function.format(**script_to_map(test)) for test in tests])
-    testcase_members = [testcase_member.format(**chew_filename(test)) for test in tests]
-    output.append(testcase_struct.format(name=group, body='\n'.join(testcase_members)))
+argparser = argparse.ArgumentParser(description='Convert native MicroPython tests to tinytest/upytesthelper C code')
+argparser.add_argument('--target', help='the target platform')
+args = argparser.parse_args()
 
-testgroup_members = [testgroup_member.format(name=group) for group in test_dirs]
+if not args.target:
+    for group in test_dirs:
+        tests += [test for test in glob('{}/*.py'.format(group)) if test not in exclude_tests]
+else:
+    for l in os.popen("./run-tests --list-tests --target=%s" % args.target, "r"):
+        tests.append(l.rstrip())
+
+output.extend([test_function.format(**script_to_map(test)) for test in tests])
+testcase_members = [testcase_member.format(**chew_filename(test)) for test in tests]
+output.append(testcase_struct.format(name="", body='\n'.join(testcase_members)))
+
+testgroup_members = [testgroup_member.format(name=group) for group in [""]]
 
 output.append(testgroup_struct.format(body='\n'.join(testgroup_members)))