diff --git a/lib/cmock.rb b/lib/cmock.rb index c1d80af..c224fa1 100644 --- a/lib/cmock.rb +++ b/lib/cmock.rb @@ -27,9 +27,9 @@ class CMock @silent = (cm_config.verbosity < 2) end - def setup_mocks(files) + def setup_mocks(files, folder = nil) [files].flatten.each do |src| - generate_mock src + generate_mock(src, folder) end end @@ -41,11 +41,11 @@ class CMock private ############################### - def generate_mock(src) + def generate_mock(src, folder) name = File.basename(src, '.*') ext = File.extname(src) puts "Creating mock for #{name}..." unless @silent - @cm_generator.create_mock(name, @cm_parser.parse(name, File.read(src)), ext) + @cm_generator.create_mock(name, @cm_parser.parse(name, File.read(src)), ext, folder) end def generate_skeleton(src) diff --git a/lib/cmock_generator.rb b/lib/cmock_generator.rb index f08cadd..b442895 100644 --- a/lib/cmock_generator.rb +++ b/lib/cmock_generator.rb @@ -21,6 +21,7 @@ class CMockGenerator @fail_on_unexpected_calls = @config.fail_on_unexpected_calls @subdir = @config.subdir + @folder = nil @includes_h_pre_orig_header = (@config.includes || @config.includes_h_pre_orig_header || []).map { |h| h =~ /\n" file << "#include \n"