summaryrefslogtreecommitdiff
path: root/sntp/unity/auto/generate_test_runner.rb
diff options
context:
space:
mode:
Diffstat (limited to 'sntp/unity/auto/generate_test_runner.rb')
-rw-r--r--sntp/unity/auto/generate_test_runner.rb6
1 files changed, 3 insertions, 3 deletions
diff --git a/sntp/unity/auto/generate_test_runner.rb b/sntp/unity/auto/generate_test_runner.rb
index 5b1d45190a56..22d10dbc1429 100644
--- a/sntp/unity/auto/generate_test_runner.rb
+++ b/sntp/unity/auto/generate_test_runner.rb
@@ -246,7 +246,7 @@ class UnityTestRunnerGenerator
def create_suite_setup_and_teardown(output)
unless (@options[:suite_setup].nil?)
output.puts("\n//=======Suite Setup=====")
- output.puts("static int suite_setup(void)")
+ output.puts("static void suite_setup(void)")
output.puts("{")
output.puts(@options[:suite_setup])
output.puts("}")
@@ -323,13 +323,13 @@ class UnityTestRunnerGenerator
output.puts(" progname = argv[0];\n")
-
+ modname = filename.split(/[\/\\]/).last
output.puts(" suite_setup();") unless @options[:suite_setup].nil?
- output.puts(" UnityBegin(\"#{filename}\");")
+ output.puts(" UnityBegin(\"#{modname}\");")
if (@options[:use_param_tests])
tests.each do |test|