Fix test for userstamps plugin

jkaramon · Jul 30, 2011 · 705aab2 · 705aab2
1 parent dcea63a
commit 705aab2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/lib/mongo_mapper/plugins/userstamps.rb b/lib/mongo_mapper/plugins/userstamps.rb
@@ -6,7 +6,7 @@ module Userstamps
 
       module ClassMethods
         def userstamps!(options = {})
-          class_name = options.fetch(:class_name) { (options.fetch :class, 'User').to_s }
+          class_name = options[:class_name] || options[:class] ? options[:class].to_s : 'User'
           key :creator_id, ObjectId
           key :updater_id, ObjectId
           belongs_to :creator, :class_name => class_name

diff --git a/test/functional/test_userstamps.rb b/test/functional/test_userstamps.rb
@@ -1,6 +1,10 @@
 require 'test_helper'
 
 class UserstampsTest < Test::Unit::TestCase
+  class AltUser
+    include MongoMapper::Document
+  end
+
   context "userstamping" do
     setup do
       @document = Doc do