blob: 7665f1c12eaf1052ad9135b8fa620267e11025e4 (
plain) (
tree)
|
|
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.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
|