133 lines
3.7 KiB
Ruby
133 lines
3.7 KiB
Ruby
require 'cases/helper'
|
|
|
|
class ModelWithAttributes
|
|
include ActiveModel::AttributeMethods
|
|
|
|
attribute_method_suffix ''
|
|
|
|
class << self
|
|
define_method(:bar) do
|
|
'original bar'
|
|
end
|
|
end
|
|
|
|
def attributes
|
|
{ :foo => 'value of foo' }
|
|
end
|
|
|
|
private
|
|
def attribute(name)
|
|
attributes[name.to_sym]
|
|
end
|
|
end
|
|
|
|
class ModelWithAttributes2
|
|
include ActiveModel::AttributeMethods
|
|
|
|
attribute_method_suffix '_test'
|
|
end
|
|
|
|
class ModelWithAttributesWithSpaces
|
|
include ActiveModel::AttributeMethods
|
|
|
|
attribute_method_suffix ''
|
|
|
|
def attributes
|
|
{ :'foo bar' => 'value of foo bar'}
|
|
end
|
|
|
|
private
|
|
def attribute(name)
|
|
attributes[name.to_sym]
|
|
end
|
|
end
|
|
|
|
class ModelWithWeirdNamesAttributes
|
|
include ActiveModel::AttributeMethods
|
|
|
|
attribute_method_suffix ''
|
|
|
|
class << self
|
|
define_method(:'c?d') do
|
|
'original c?d'
|
|
end
|
|
end
|
|
|
|
def attributes
|
|
{ :'a?b' => 'value of a?b' }
|
|
end
|
|
|
|
private
|
|
def attribute(name)
|
|
attributes[name.to_sym]
|
|
end
|
|
end
|
|
|
|
class AttributeMethodsTest < ActiveModel::TestCase
|
|
test 'unrelated classes should not share attribute method matchers' do
|
|
assert_not_equal ModelWithAttributes.send(:attribute_method_matchers),
|
|
ModelWithAttributes2.send(:attribute_method_matchers)
|
|
end
|
|
|
|
test '#define_attribute_method generates attribute method' do
|
|
ModelWithAttributes.define_attribute_method(:foo)
|
|
|
|
assert_respond_to ModelWithAttributes.new, :foo
|
|
assert_equal "value of foo", ModelWithAttributes.new.foo
|
|
end
|
|
|
|
test '#define_attribute_method generates attribute method with invalid identifier characters' do
|
|
ModelWithWeirdNamesAttributes.define_attribute_method(:'a?b')
|
|
|
|
assert_respond_to ModelWithWeirdNamesAttributes.new, :'a?b'
|
|
assert_equal "value of a?b", ModelWithWeirdNamesAttributes.new.send('a?b')
|
|
end
|
|
|
|
test '#define_attribute_methods generates attribute methods' do
|
|
ModelWithAttributes.define_attribute_methods([:foo])
|
|
|
|
assert_respond_to ModelWithAttributes.new, :foo
|
|
assert_equal "value of foo", ModelWithAttributes.new.foo
|
|
end
|
|
|
|
test '#define_attribute_methods generates attribute methods with spaces in their names' do
|
|
ModelWithAttributesWithSpaces.define_attribute_methods([:'foo bar'])
|
|
|
|
assert_respond_to ModelWithAttributesWithSpaces.new, :'foo bar'
|
|
assert_equal "value of foo bar", ModelWithAttributesWithSpaces.new.send(:'foo bar')
|
|
end
|
|
|
|
test '#define_attr_method generates attribute method' do
|
|
ModelWithAttributes.define_attr_method(:bar, 'bar')
|
|
|
|
assert_respond_to ModelWithAttributes, :bar
|
|
assert_equal "original bar", ModelWithAttributes.original_bar
|
|
assert_equal "bar", ModelWithAttributes.bar
|
|
ModelWithAttributes.define_attr_method(:bar)
|
|
assert !ModelWithAttributes.bar
|
|
end
|
|
|
|
test '#define_attr_method generates attribute method with invalid identifier characters' do
|
|
ModelWithWeirdNamesAttributes.define_attr_method(:'c?d', 'c?d')
|
|
|
|
assert_respond_to ModelWithWeirdNamesAttributes, :'c?d'
|
|
assert_equal "original c?d", ModelWithWeirdNamesAttributes.send('original_c?d')
|
|
assert_equal "c?d", ModelWithWeirdNamesAttributes.send('c?d')
|
|
end
|
|
|
|
test '#alias_attribute works with attributes with spaces in their names' do
|
|
ModelWithAttributesWithSpaces.define_attribute_methods([:'foo bar'])
|
|
ModelWithAttributesWithSpaces.alias_attribute(:'foo_bar', :'foo bar')
|
|
|
|
assert_equal "value of foo bar", ModelWithAttributesWithSpaces.new.foo_bar
|
|
end
|
|
|
|
test '#undefine_attribute_methods removes attribute methods' do
|
|
ModelWithAttributes.define_attribute_methods([:foo])
|
|
ModelWithAttributes.undefine_attribute_methods
|
|
|
|
assert !ModelWithAttributes.new.respond_to?(:foo)
|
|
assert_raises(NoMethodError) { ModelWithAttributes.new.foo }
|
|
end
|
|
end
|