Security-59306.11.20.tar.gz

[apple/security.git] / tests / TrustTests / gen_test_plist.py

diff --git a/tests/TrustTests/gen_test_plist.py b/tests/TrustTests/gen_test_plist.py
new file mode 100644 (file)
index 0000000..ee7c730
--- /dev/null
+++ b/tests/TrustTests/gen_test_plist.py
@@ -0,0 +1,51 @@
+#!/usr/bin/python
+import sys
+import Foundation
+from glob import glob
+import re
+import os
+
+test_dir = sys.argv[1]
+outfile = sys.argv[2]
+
+test_plist = Foundation.NSMutableDictionary.dictionary()
+test_plist['Project'] = 'Security'
+test_list = Foundation.NSMutableArray.array()
+
+test_files = glob(test_dir + '/*/*.m')
+
+for filename in test_files:
+    f = open(filename, 'r')
+    for line in f:
+        match = re.search('@interface ([a-zA-Z0-9_]+) ?: ?Trust[a-zA-Z0-9_]+TestCase', line)
+        if match:
+            regex_string = test_dir + '/([a-zA-Z0-9_]+/)'
+            test_dir_match = re.search(regex_string, filename)
+            if not test_dir_match:
+                print 'failed to find test dir name for ' + filename + " with regex " + regex_string
+                sys.exit(1)
+            test_dictionary = Foundation.NSMutableDictionary.dictionary()
+            test_dictionary['TestName'] = test_dir_match.group(1) + match.group(1)
+            test_dictionary['ShowSubtestResults']= True
+            test_dictionary['WorkingDirectory'] = '/AppleInternal/XCTests/com.apple.security/'
+            test_dictionary['AsRoot'] = True
+
+            test_command = Foundation.NSMutableArray.array()
+            test_command.append('/AppleInternal/CoreOS/tests/Security/TrustTests')
+            test_command.append('-c ' + match.group(1))
+            test_command.append('-t TrustTests')
+
+            test_dictionary['Command'] = test_command
+
+            test_list.append(test_dictionary)
+    f.close()
+
+test_plist['Tests'] = test_list
+
+out_dir = os.path.dirname(outfile)
+if not os.path.exists(out_dir):
+    os.makedirs(out_dir)
+success = test_plist.writeToFile_atomically_(outfile, 1)
+if not success:
+    print "test plist failed to write, error!"
+    sys.exit(1)