espressif/Unity
1
0
Fork 0
mirror of https://github.com/ThrowTheSwitch/Unity.git synced 2025-10-13 10:16:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

support cmock's need to destroy all of its memory at the end of the suite

Browse Source
This commit is contained in:
Mark VanderVoord
2014-05-21 19:29:44 -04:00
parent 124bfd8027
commit 1cf02f8bf3
9 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
auto/generate_test_runner.rb
View File

@ -62,7 +62,7 @@ class UnityTestRunnerGenerator
create_mock_management(output, used_mocks)
create_suite_setup_and_teardown(output)
create_reset(output, used_mocks)
create_main(output, input_file, tests)
create_main(output, input_file, tests, used_mocks)
end
end
@ -142,11 +142,11 @@ class UnityTestRunnerGenerator
output.puts('#include <setjmp.h>')
output.puts('#include <stdio.h>')
output.puts('#include "CException.h"') if @options[:plugins].include?(:cexception)
testfile_includes.delete("unity").delete("cmock")
testrunner_includes = testfile_includes - mocks
testrunner_includes.each do |inc|
output.puts("#include #{inc.include?('<') ? inc : "\"#{inc.gsub('.h','')}.h\""}")
end
testfile_includes.delete("unity").delete("cmock")
testrunner_includes = testfile_includes - mocks
testrunner_includes.each do |inc|
output.puts("#include #{inc.include?('<') ? inc : "\"#{inc.gsub('.h','')}.h\""}")
end
mocks.each do |mock|
output.puts("#include \"#{mock.gsub('.h','')}.h\"")
end
@ -261,7 +261,7 @@ class UnityTestRunnerGenerator
output.puts("}")
end
def create_main(output, filename, tests)
def create_main(output, filename, tests, used_mocks)
output.puts("\n\n//=======MAIN=====")
output.puts("int main(void)")
output.puts("{")
@ -280,6 +280,7 @@ class UnityTestRunnerGenerator
tests.each { |test| output.puts(" RUN_TEST(#{test[:test]}, #{test[:line_number]});") }
end
output.puts()
output.puts(" CMock_Guts_MemFreeFinal();") unless used_mocks.empty?
output.puts(" return #{@options[:suite_teardown].nil? ? "" : "suite_teardown"}(UnityEnd());")
output.puts("}")
end