CMock/lib/cmock_unityhelper_parser.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 CMockUnityHelperParser
  attr_accessor :c_types

  def initialize(config)
    @config = config
    @fallback = @config.plugins.include?(:array) ? 'UNITY_TEST_ASSERT_EQUAL_MEMORY_ARRAY' : 'UNITY_TEST_ASSERT_EQUAL_MEMORY'
    @c_types = map_c_types.merge(import_source)
  end

  def get_helper(ctype)
    lookup = ctype.gsub(/(?:^|(\S?)(\s*)|(\W))const(?:$|(\s*)(\S)|(\W))/, '\1\3\5\6').strip.gsub(/\s+/, '_')
    return [@c_types[lookup], ''] if @c_types[lookup]

    if lookup =~ /\*$/
      lookup = lookup.gsub(/\*$/, '')
      return [@c_types[lookup], '*'] if @c_types[lookup]
    else
      lookup += '*'
      return [@c_types[lookup], '&'] if @c_types[lookup]
    end
    return ['UNITY_TEST_ASSERT_EQUAL_PTR', ''] if ctype =~ /cmock_\w+_ptr\d+/
    raise("Don't know how to test #{ctype} and memory tests are disabled!") unless @config.memcmp_if_unknown

    lookup =~ /\*$/ ? [@fallback, '&'] : [@fallback, '']
  end

  private ###########################

  def map_c_types
    c_types = {}
    @config.treat_as.each_pair do |ctype, expecttype|
      c_type = ctype.gsub(/\s+/, '_')
      if expecttype =~ /\*/
        c_types[c_type] = "UNITY_TEST_ASSERT_EQUAL_#{expecttype.delete('*')}_ARRAY"
      else
        c_types[c_type] = "UNITY_TEST_ASSERT_EQUAL_#{expecttype}"
        c_types[c_type + '*'] ||= "UNITY_TEST_ASSERT_EQUAL_#{expecttype}_ARRAY"
      end
    end
    c_types
  end

  def import_source
    source = @config.load_unity_helper
    return {} if source.nil?

    c_types = {}
    source = source.gsub(/\/\/.*$/, '') # remove line comments
    source = source.gsub(/\/\*.*?\*\//m, '') # remove block comments

    # scan for comparison helpers
    match_regex = Regexp.new('^\s*#define\s+(UNITY_TEST_ASSERT_EQUAL_(\w+))\s*\(' + Array.new(4, '\s*\w+\s*').join(',') + '\)')
    pairs = source.scan(match_regex).flatten.compact
    (pairs.size / 2).times do |i|
      expect = pairs[i * 2]
      ctype = pairs[(i * 2) + 1]
      c_types[ctype] = expect unless expect.include?('_ARRAY')
    end

    # scan for array variants of those helpers
    match_regex = Regexp.new('^\s*#define\s+(UNITY_TEST_ASSERT_EQUAL_(\w+_ARRAY))\s*\(' + Array.new(5, '\s*\w+\s*').join(',') + '\)')
    pairs = source.scan(match_regex).flatten.compact
    (pairs.size / 2).times do |i|
      expect = pairs[i * 2]
      ctype = pairs[(i * 2) + 1]
      c_types[ctype.gsub('_ARRAY', '*')] = expect
    end

    c_types
  end
end