Skip to content
Browse files

make enums distinct per class

  • Loading branch information...
1 parent c45939e commit bbe7fe41692e5a2c869f812616d0c99a2bfcb39c @evanwhalen evanwhalen committed Apr 7, 2014
Showing with 28 additions and 4 deletions.
  1. +7 −0 activerecord/CHANGELOG.md
  2. +6 −4 activerecord/lib/active_record/enum.rb
  3. +15 −0 activerecord/test/cases/enum_test.rb
View
7 activerecord/CHANGELOG.md
@@ -1,3 +1,10 @@
+* Fixed a problem where an enum would overwrite values of another enum
+ with the same name in an unrelated class.
+
+ Fixes #14607.
+
+ *Evan Whalen*
+
* PostgreSQL and SQLite string columns no longer have a default limit of 255.
Fixes #13435, #9153.
View
10 activerecord/lib/active_record/enum.rb
@@ -65,10 +65,13 @@ module ActiveRecord
#
# Where conditions on an enum attribute must use the ordinal value of an enum.
module Enum
- DEFINED_ENUMS = {} # :nodoc:
+ def self.extended(base)
+ base.class_attribute(:defined_enums)
+ base.defined_enums = {}
+ end
def enum_mapping_for(attr_name) # :nodoc:
- DEFINED_ENUMS[attr_name.to_s]
+ defined_enums[attr_name.to_s]
end
def enum(definitions)
@@ -122,9 +125,8 @@ def enum(definitions)
klass.send(:detect_enum_conflict!, name, value, true)
klass.scope value, -> { klass.where name => i }
end
-
- DEFINED_ENUMS[name.to_s] = enum_values
end
+ defined_enums[name.to_s] = enum_values
end
end
View
15 activerecord/test/cases/enum_test.rb
@@ -250,4 +250,19 @@ def self.name; 'Book'; end
valid_book = klass.new(status: "written")
assert valid_book.valid?
end
+
+ test "enums are distinct per class" do
+ Plane = Class.new(ActiveRecord::Base) do
+ enum status: [:grounded, :operational]
+ end
+ assert_equal({ "proposed" => 0, "written" => 1, "published" => 2 }, Book.statuses)
+ assert_equal({ "grounded" => 0, "operational" => 1 }, Plane.statuses)
+ end
+
+ test "enums are inheritable" do
+ Encyclopedia = Class.new(Book) do
+ enum status: [:published, :reprinted]
+ end
+ assert_equal({ "published" => 0, "reprinted" => 1 }, Encyclopedia.statuses)
+ end
end

0 comments on commit bbe7fe4

Please sign in to comment.
Something went wrong with that request. Please try again.