mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
e7a28c3990
This adds `.attribute_names` and `#attribute_names` to `ActiveModel::Attributes` along the same lines as the corresponding methods in `ActiveRecord::AttributeMethods` (see [`.attribute_names`][class_method] and [`#attribute_names`][instance_method]. While I was here I also added documentation for '#attributes', which I added in043ce35b18. The whole class is still `#:nodoc:` so I don't think this will have any effect for now. [class_method]:cc834db1d0/activerecord/lib/active_record/attribute_methods.rb (L154-L160)[instance_method]:cc834db1d0/activerecord/lib/active_record/attribute_methods.rb (L299-L301)
111 lines
3.2 KiB
Ruby
111 lines
3.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "cases/helper"
|
|
|
|
module ActiveModel
|
|
class AttributesTest < ActiveModel::TestCase
|
|
class ModelForAttributesTest
|
|
include ActiveModel::Model
|
|
include ActiveModel::Attributes
|
|
|
|
attribute :integer_field, :integer
|
|
attribute :string_field, :string
|
|
attribute :decimal_field, :decimal
|
|
attribute :string_with_default, :string, default: "default string"
|
|
attribute :date_field, :date, default: -> { Date.new(2016, 1, 1) }
|
|
attribute :boolean_field, :boolean
|
|
end
|
|
|
|
class ChildModelForAttributesTest < ModelForAttributesTest
|
|
end
|
|
|
|
class GrandchildModelForAttributesTest < ChildModelForAttributesTest
|
|
attribute :integer_field, :string
|
|
end
|
|
|
|
test "properties assignment" do
|
|
data = ModelForAttributesTest.new(
|
|
integer_field: "2.3",
|
|
string_field: "Rails FTW",
|
|
decimal_field: "12.3",
|
|
boolean_field: "0"
|
|
)
|
|
|
|
assert_equal 2, data.integer_field
|
|
assert_equal "Rails FTW", data.string_field
|
|
assert_equal BigDecimal("12.3"), data.decimal_field
|
|
assert_equal "default string", data.string_with_default
|
|
assert_equal Date.new(2016, 1, 1), data.date_field
|
|
assert_equal false, data.boolean_field
|
|
|
|
data.integer_field = 10
|
|
data.string_with_default = nil
|
|
data.boolean_field = "1"
|
|
|
|
assert_equal 10, data.integer_field
|
|
assert_nil data.string_with_default
|
|
assert_equal true, data.boolean_field
|
|
end
|
|
|
|
test "reading attributes" do
|
|
data = ModelForAttributesTest.new(
|
|
integer_field: 1.1,
|
|
string_field: 1.1,
|
|
decimal_field: 1.1,
|
|
boolean_field: 1.1
|
|
)
|
|
|
|
expected_attributes = {
|
|
integer_field: 1,
|
|
string_field: "1.1",
|
|
decimal_field: BigDecimal("1.1"),
|
|
string_with_default: "default string",
|
|
date_field: Date.new(2016, 1, 1),
|
|
boolean_field: true
|
|
}.stringify_keys
|
|
|
|
assert_equal expected_attributes, data.attributes
|
|
end
|
|
|
|
test "reading attribute names" do
|
|
names = [
|
|
"integer_field",
|
|
"string_field",
|
|
"decimal_field",
|
|
"string_with_default",
|
|
"date_field",
|
|
"boolean_field"
|
|
]
|
|
|
|
assert_equal names, ModelForAttributesTest.attribute_names
|
|
assert_equal names, ModelForAttributesTest.new.attribute_names
|
|
end
|
|
|
|
test "nonexistent attribute" do
|
|
assert_raise ActiveModel::UnknownAttributeError do
|
|
ModelForAttributesTest.new(nonexistent: "nonexistent")
|
|
end
|
|
end
|
|
|
|
test "children inherit attributes" do
|
|
data = ChildModelForAttributesTest.new(integer_field: "4.4")
|
|
|
|
assert_equal 4, data.integer_field
|
|
end
|
|
|
|
test "children can override parents" do
|
|
data = GrandchildModelForAttributesTest.new(integer_field: "4.4")
|
|
|
|
assert_equal "4.4", data.integer_field
|
|
end
|
|
|
|
test "attributes with proc defaults can be marshalled" do
|
|
data = ModelForAttributesTest.new
|
|
attributes = data.instance_variable_get(:@attributes)
|
|
round_tripped = Marshal.load(Marshal.dump(data))
|
|
new_attributes = round_tripped.instance_variable_get(:@attributes)
|
|
|
|
assert_equal attributes, new_attributes
|
|
end
|
|
end
|
|
end
|