CMock/lib/cmock_generator_plugin_expect_any_args.rb

# ==========================================
#   CMock Project - Automatic Mock Generation for C
#   Copyright (c) 2007 Mike Karlesky, Mark VanderVoord, Greg Williams
#   [Released under MIT License. Please refer to license.txt for details]
# ==========================================

class CMockGeneratorPluginExpectAnyArgs

  attr_reader :priority
  attr_reader :config, :utils

  def initialize(config, utils)
    @config = config
    @utils = utils
    @priority = 3
  end

  def instance_structure(function)
    if (function[:return][:void?]) || (@config.plugins.include? :ignore)
      ""
    else
      "  #{function[:return][:type]} #{function[:name]}_FinalReturn;\n"
    end
  end

  def instance_typedefs(function)
    "  CMOCK_ARG_MODE IgnoreMode;\n"
  end

  def mock_function_declarations(function)

    if (function[:return][:void?])
      return "#define #{function[:name]}_ExpectAnyArgs() #{function[:name]}_CMockExpectAnyArgs(__LINE__)\n" +
             "void #{function[:name]}_CMockExpectAnyArgs(UNITY_LINE_TYPE cmock_line);\n"
    else
      return "#define #{function[:name]}_ExpectAnyArgsAndReturn(cmock_retval) #{function[:name]}_CMockExpectAnyArgsAndReturn(__LINE__, cmock_retval)\n" +
             "void #{function[:name]}_CMockExpectAnyArgsAndReturn(UNITY_LINE_TYPE cmock_line, #{function[:return][:str]});\n"
    end
  end

  # def mock_implementation(function)
  #   lines = "  if (cmock_call_instance->IgnoreMode == CMOCK_ARG_NONE)\n  {\n"
  #   if (function[:return][:void?])
  #     lines << "    return;\n  }\n"
  #   else
  #     retval = function[:return].merge( { :name => "cmock_call_instance->ReturnVal"} )
  #     lines << "  " + @utils.code_assign_argument_quickly("Mock.#{function[:name]}_FinalReturn", retval) unless (retval[:void?])
  #     return_type = function[:return][:const?] ? "(const #{function[:return][:type]})" : ((function[:return][:type] =~ /cmock/) ? "(#{function[:return][:type]})" : '')
  #     lines << "    return #{return_type}cmock_call_instance->ReturnVal;\n  }\n"
  #   end
  #   lines
  # end

  def mock_interfaces(function)
    lines = ""
    if (function[:return][:void?])
      lines << "void #{function[:name]}_CMockExpectAnyArgs(UNITY_LINE_TYPE cmock_line)\n{\n"
    else
      lines << "void #{function[:name]}_CMockExpectAnyArgsAndReturn(UNITY_LINE_TYPE cmock_line, #{function[:return][:str]})\n{\n"
    end
    lines << @utils.code_add_base_expectation(function[:name], true)
    unless (function[:return][:void?])
      lines << "  cmock_call_instance->ReturnVal = cmock_to_return;\n"
    end
    lines << "  cmock_call_instance->IgnoreMode = CMOCK_ARG_NONE;\n"
    lines << "}\n\n"
  end
end