require "cases/helper" require 'active_support/core_ext/object/inclusion' module ActiveRecord module AttributeMethods class ReadTest < ActiveRecord::TestCase class FakeColumn < Struct.new(:name) def type_cast_code(var) var end def type; :integer; end end def setup @klass = Class.new do def self.superclass; Base; end def self.active_record_super; Base; end def self.base_class; self; end include ActiveRecord::AttributeMethods def self.column_names %w{ one two three } end def self.primary_key end def self.columns column_names.map { FakeColumn.new(name) } end def self.columns_hash Hash[column_names.map { |name| [name, FakeColumn.new(name)] }] end end end def test_define_attribute_methods instance = @klass.new @klass.column_names.each do |name| assert !name.in?(instance.methods.map(&:to_s)) end @klass.define_attribute_methods @klass.column_names.each do |name| assert name.in?(instance.methods.map(&:to_s)), "#{name} is not defined" end end def test_attribute_methods_generated? assert(!@klass.attribute_methods_generated?, 'attribute_methods_generated?') @klass.define_attribute_methods assert(@klass.attribute_methods_generated?, 'attribute_methods_generated?') end end end end