mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
305 lines
11 KiB
Ruby
305 lines
11 KiB
Ruby
require "cases/helper"
|
|
require 'models/topic'
|
|
require 'models/minimalistic'
|
|
|
|
class AttributeMethodsTest < ActiveRecord::TestCase
|
|
fixtures :topics
|
|
def setup
|
|
@old_suffixes = ActiveRecord::Base.send(:attribute_method_suffixes).dup
|
|
@target = Class.new(ActiveRecord::Base)
|
|
@target.table_name = 'topics'
|
|
end
|
|
|
|
def teardown
|
|
ActiveRecord::Base.send(:attribute_method_suffixes).clear
|
|
ActiveRecord::Base.attribute_method_suffix *@old_suffixes
|
|
end
|
|
|
|
def test_match_attribute_method_query_returns_match_data
|
|
assert_not_nil md = @target.match_attribute_method?('title=')
|
|
assert_equal 'title', md.pre_match
|
|
assert_equal ['='], md.captures
|
|
|
|
%w(_hello_world ist! _maybe?).each do |suffix|
|
|
@target.class_eval "def attribute#{suffix}(*args) args end"
|
|
@target.attribute_method_suffix suffix
|
|
|
|
assert_not_nil md = @target.match_attribute_method?("title#{suffix}")
|
|
assert_equal 'title', md.pre_match
|
|
assert_equal [suffix], md.captures
|
|
end
|
|
end
|
|
|
|
def test_declared_attribute_method_affects_respond_to_and_method_missing
|
|
topic = @target.new(:title => 'Budget')
|
|
assert topic.respond_to?('title')
|
|
assert_equal 'Budget', topic.title
|
|
assert !topic.respond_to?('title_hello_world')
|
|
assert_raise(NoMethodError) { topic.title_hello_world }
|
|
|
|
%w(_hello_world _it! _candidate= able?).each do |suffix|
|
|
@target.class_eval "def attribute#{suffix}(*args) args end"
|
|
@target.attribute_method_suffix suffix
|
|
|
|
meth = "title#{suffix}"
|
|
assert topic.respond_to?(meth)
|
|
assert_equal ['title'], topic.send(meth)
|
|
assert_equal ['title', 'a'], topic.send(meth, 'a')
|
|
assert_equal ['title', 1, 2, 3], topic.send(meth, 1, 2, 3)
|
|
end
|
|
end
|
|
|
|
def test_should_unserialize_attributes_for_frozen_records
|
|
myobj = {:value1 => :value2}
|
|
topic = Topic.create("content" => myobj)
|
|
topic.freeze
|
|
assert_equal myobj, topic.content
|
|
end
|
|
|
|
def test_typecast_attribute_from_select_to_false
|
|
topic = Topic.create(:title => 'Budget')
|
|
topic = Topic.find(:first, :select => "topics.*, 1=2 as is_test")
|
|
assert !topic.is_test?
|
|
end
|
|
|
|
def test_typecast_attribute_from_select_to_true
|
|
topic = Topic.create(:title => 'Budget')
|
|
topic = Topic.find(:first, :select => "topics.*, 2=2 as is_test")
|
|
assert topic.is_test?
|
|
end
|
|
|
|
def test_kernel_methods_not_implemented_in_activerecord
|
|
%w(test name display y).each do |method|
|
|
assert !ActiveRecord::Base.instance_method_already_implemented?(method), "##{method} is defined"
|
|
end
|
|
end
|
|
|
|
def test_primary_key_implemented
|
|
assert Class.new(ActiveRecord::Base).instance_method_already_implemented?('id')
|
|
end
|
|
|
|
def test_defined_kernel_methods_implemented_in_model
|
|
%w(test name display y).each do |method|
|
|
klass = Class.new ActiveRecord::Base
|
|
klass.class_eval "def #{method}() 'defined #{method}' end"
|
|
assert klass.instance_method_already_implemented?(method), "##{method} is not defined"
|
|
end
|
|
end
|
|
|
|
def test_defined_kernel_methods_implemented_in_model_abstract_subclass
|
|
%w(test name display y).each do |method|
|
|
abstract = Class.new ActiveRecord::Base
|
|
abstract.class_eval "def #{method}() 'defined #{method}' end"
|
|
abstract.abstract_class = true
|
|
klass = Class.new abstract
|
|
assert klass.instance_method_already_implemented?(method), "##{method} is not defined"
|
|
end
|
|
end
|
|
|
|
def test_raises_dangerous_attribute_error_when_defining_activerecord_method_in_model
|
|
%w(save create_or_update).each do |method|
|
|
klass = Class.new ActiveRecord::Base
|
|
klass.class_eval "def #{method}() 'defined #{method}' end"
|
|
assert_raise ActiveRecord::DangerousAttributeError do
|
|
klass.instance_method_already_implemented?(method)
|
|
end
|
|
end
|
|
end
|
|
|
|
def test_only_time_related_columns_are_meant_to_be_cached_by_default
|
|
expected = %w(datetime timestamp time date).sort
|
|
assert_equal expected, ActiveRecord::Base.attribute_types_cached_by_default.map(&:to_s).sort
|
|
end
|
|
|
|
def test_declaring_attributes_as_cached_adds_them_to_the_attributes_cached_by_default
|
|
default_attributes = Topic.cached_attributes
|
|
Topic.cache_attributes :replies_count
|
|
expected = default_attributes + ["replies_count"]
|
|
assert_equal expected.sort, Topic.cached_attributes.sort
|
|
Topic.instance_variable_set "@cached_attributes", nil
|
|
end
|
|
|
|
def test_time_related_columns_are_actually_cached
|
|
column_types = %w(datetime timestamp time date).map(&:to_sym)
|
|
column_names = Topic.columns.select{|c| column_types.include?(c.type) }.map(&:name)
|
|
|
|
assert_equal column_names.sort, Topic.cached_attributes.sort
|
|
assert_equal time_related_columns_on_topic.sort, Topic.cached_attributes.sort
|
|
end
|
|
|
|
def test_accessing_cached_attributes_caches_the_converted_values_and_nothing_else
|
|
t = topics(:first)
|
|
cache = t.instance_variable_get "@attributes_cache"
|
|
|
|
assert_not_nil cache
|
|
assert cache.empty?
|
|
|
|
all_columns = Topic.columns.map(&:name)
|
|
cached_columns = time_related_columns_on_topic
|
|
uncached_columns = all_columns - cached_columns
|
|
|
|
all_columns.each do |attr_name|
|
|
attribute_gets_cached = Topic.cache_attribute?(attr_name)
|
|
val = t.send attr_name unless attr_name == "type"
|
|
if attribute_gets_cached
|
|
assert cached_columns.include?(attr_name)
|
|
assert_equal val, cache[attr_name]
|
|
else
|
|
assert uncached_columns.include?(attr_name)
|
|
assert !cache.include?(attr_name)
|
|
end
|
|
end
|
|
end
|
|
|
|
def test_time_attributes_are_retrieved_in_current_time_zone
|
|
in_time_zone "Pacific Time (US & Canada)" do
|
|
utc_time = Time.utc(2008, 1, 1)
|
|
record = @target.new
|
|
record[:written_on] = utc_time
|
|
assert_equal utc_time, record.written_on # record.written on is equal to (i.e., simultaneous with) utc_time
|
|
assert_kind_of ActiveSupport::TimeWithZone, record.written_on # but is a TimeWithZone
|
|
assert_equal ActiveSupport::TimeZone["Pacific Time (US & Canada)"], record.written_on.time_zone # and is in the current Time.zone
|
|
assert_equal Time.utc(2007, 12, 31, 16), record.written_on.time # and represents time values adjusted accordingly
|
|
end
|
|
end
|
|
|
|
def test_setting_time_zone_aware_attribute_to_utc
|
|
in_time_zone "Pacific Time (US & Canada)" do
|
|
utc_time = Time.utc(2008, 1, 1)
|
|
record = @target.new
|
|
record.written_on = utc_time
|
|
assert_equal utc_time, record.written_on
|
|
assert_equal ActiveSupport::TimeZone["Pacific Time (US & Canada)"], record.written_on.time_zone
|
|
assert_equal Time.utc(2007, 12, 31, 16), record.written_on.time
|
|
end
|
|
end
|
|
|
|
def test_setting_time_zone_aware_attribute_in_other_time_zone
|
|
utc_time = Time.utc(2008, 1, 1)
|
|
cst_time = utc_time.in_time_zone("Central Time (US & Canada)")
|
|
in_time_zone "Pacific Time (US & Canada)" do
|
|
record = @target.new
|
|
record.written_on = cst_time
|
|
assert_equal utc_time, record.written_on
|
|
assert_equal ActiveSupport::TimeZone["Pacific Time (US & Canada)"], record.written_on.time_zone
|
|
assert_equal Time.utc(2007, 12, 31, 16), record.written_on.time
|
|
end
|
|
end
|
|
|
|
def test_setting_time_zone_aware_attribute_with_string
|
|
utc_time = Time.utc(2008, 1, 1)
|
|
(-11..13).each do |timezone_offset|
|
|
time_string = utc_time.in_time_zone(timezone_offset).to_s
|
|
in_time_zone "Pacific Time (US & Canada)" do
|
|
record = @target.new
|
|
record.written_on = time_string
|
|
assert_equal Time.zone.parse(time_string), record.written_on
|
|
assert_equal ActiveSupport::TimeZone["Pacific Time (US & Canada)"], record.written_on.time_zone
|
|
assert_equal Time.utc(2007, 12, 31, 16), record.written_on.time
|
|
end
|
|
end
|
|
end
|
|
|
|
def test_setting_time_zone_aware_attribute_to_blank_string_returns_nil
|
|
in_time_zone "Pacific Time (US & Canada)" do
|
|
record = @target.new
|
|
record.written_on = ' '
|
|
assert_nil record.written_on
|
|
end
|
|
end
|
|
|
|
def test_setting_time_zone_aware_attribute_interprets_time_zone_unaware_string_in_time_zone
|
|
time_string = 'Tue Jan 01 00:00:00 2008'
|
|
(-11..13).each do |timezone_offset|
|
|
in_time_zone timezone_offset do
|
|
record = @target.new
|
|
record.written_on = time_string
|
|
assert_equal Time.zone.parse(time_string), record.written_on
|
|
assert_equal ActiveSupport::TimeZone[timezone_offset], record.written_on.time_zone
|
|
assert_equal Time.utc(2008, 1, 1), record.written_on.time
|
|
end
|
|
end
|
|
end
|
|
|
|
def test_setting_time_zone_aware_attribute_in_current_time_zone
|
|
utc_time = Time.utc(2008, 1, 1)
|
|
in_time_zone "Pacific Time (US & Canada)" do
|
|
record = @target.new
|
|
record.written_on = utc_time.in_time_zone
|
|
assert_equal utc_time, record.written_on
|
|
assert_equal ActiveSupport::TimeZone["Pacific Time (US & Canada)"], record.written_on.time_zone
|
|
assert_equal Time.utc(2007, 12, 31, 16), record.written_on.time
|
|
end
|
|
end
|
|
|
|
def test_setting_time_zone_conversion_for_attributes_should_write_value_on_class_variable
|
|
Topic.skip_time_zone_conversion_for_attributes = [:field_a]
|
|
Minimalistic.skip_time_zone_conversion_for_attributes = [:field_b]
|
|
|
|
assert_equal [:field_a], Topic.skip_time_zone_conversion_for_attributes
|
|
assert_equal [:field_b], Minimalistic.skip_time_zone_conversion_for_attributes
|
|
end
|
|
|
|
def test_read_attributes_respect_access_control
|
|
privatize("title")
|
|
|
|
topic = @target.new(:title => "The pros and cons of programming naked.")
|
|
assert !topic.respond_to?(:title)
|
|
exception = assert_raise(NoMethodError) { topic.title }
|
|
assert_equal "Attempt to call private method", exception.message
|
|
assert_equal "I'm private", topic.send(:title)
|
|
end
|
|
|
|
def test_write_attributes_respect_access_control
|
|
privatize("title=(value)")
|
|
|
|
topic = @target.new
|
|
assert !topic.respond_to?(:title=)
|
|
exception = assert_raise(NoMethodError) { topic.title = "Pants"}
|
|
assert_equal "Attempt to call private method", exception.message
|
|
topic.send(:title=, "Very large pants")
|
|
end
|
|
|
|
def test_question_attributes_respect_access_control
|
|
privatize("title?")
|
|
|
|
topic = @target.new(:title => "Isaac Newton's pants")
|
|
assert !topic.respond_to?(:title?)
|
|
exception = assert_raise(NoMethodError) { topic.title? }
|
|
assert_equal "Attempt to call private method", exception.message
|
|
assert topic.send(:title?)
|
|
end
|
|
|
|
def test_bulk_update_respects_access_control
|
|
privatize("title=(value)")
|
|
|
|
assert_raise(ActiveRecord::UnknownAttributeError) { topic = @target.new(:title => "Rants about pants") }
|
|
assert_raise(ActiveRecord::UnknownAttributeError) { @target.new.attributes = { :title => "Ants in pants" } }
|
|
end
|
|
|
|
private
|
|
def time_related_columns_on_topic
|
|
Topic.columns.select{|c| [:time, :date, :datetime, :timestamp].include?(c.type)}.map(&:name)
|
|
end
|
|
|
|
def in_time_zone(zone)
|
|
old_zone = Time.zone
|
|
old_tz = ActiveRecord::Base.time_zone_aware_attributes
|
|
|
|
Time.zone = zone ? ActiveSupport::TimeZone[zone] : nil
|
|
ActiveRecord::Base.time_zone_aware_attributes = !zone.nil?
|
|
yield
|
|
ensure
|
|
Time.zone = old_zone
|
|
ActiveRecord::Base.time_zone_aware_attributes = old_tz
|
|
end
|
|
|
|
def privatize(method_signature)
|
|
@target.class_eval <<-private_method
|
|
private
|
|
def #{method_signature}
|
|
"I'm private"
|
|
end
|
|
private_method
|
|
end
|
|
end
|