1 files changed, 25 insertions, 6 deletions
diff --git a/activerecord/test/attribute_methods_test.rb b/activerecord/test/attribute_methods_test.rb
index 8dcca2fc4a..0fa13c65ca 100755
--- a/activerecord/test/attribute_methods_test.rb
+++ b/activerecord/test/attribute_methods_test.rb
@@ -2,14 +2,30 @@ require 'abstract_unit'
 
 class AttributeMethodsTest < Test::Unit::TestCase
   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
@@ -19,12 +35,15 @@ class AttributeMethodsTest < Test::Unit::TestCase
     assert !topic.respond_to?('title_hello_world')
     assert_raise(NoMethodError) { topic.title_hello_world }
 
-    @target.class_eval "def attribute_hello_world(*args) args end"
-    @target.attribute_method_suffix '_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
 
-    assert topic.respond_to?('title_hello_world')
-    assert_equal ['title'], topic.title_hello_world
-    assert_equal ['title', 'a'], topic.title_hello_world('a')
-    assert_equal ['title', 1, 2, 3], topic.title_hello_world(1, 2, 3)
+      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
 end