mongo mapper tweaks...

lib/database_cleaner/mongo_mapper/base.rb

@@ -7,6 +7,14 @@ module DatabaseCleaner
     
     module Base
       include ::DatabaseCleaner::Generic::Base
+      
+      def db=(desired_db)
+        @db = desired_db
+      end
+      
+      def db
+        @db || :default
+      end
     end
   end
 end

spec/database_cleaner/base_spec.rb

@@ -82,7 +82,7 @@ module DatabaseCleaner
     
     describe "comparison" do
       it "should be equal if orm, connection and strategy are the same" do
-        strategy = :truncation
+        strategy = mock("strategy")
         
         one = DatabaseCleaner::Base.new(:active_record,:connection => :default)
         one.strategy = strategy

spec/database_cleaner/mongo_mapper/base_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+require 'database_cleaner/mongo_mapper/base'
+require 'database_cleaner/shared_strategy_spec'
+
+module DatabaseCleaner
+  describe MongoMapper do
+    it { should respond_to :available_strategies }
+  end                
+  
+  module MongoMapper 
+    class ExampleStrategy
+      include ::DatabaseCleaner::MongoMapper::Base
+    end
+      
+    describe ExampleStrategy do
+            
+      it_should_behave_like "a generic strategy"          
+
+      describe "db" do
+        it { should respond_to :db= }
+        
+        it "should store my desired db" do
+          subject.db = :my_db
+          subject.db.should == :my_db
+        end
+      
+        it "should default to :default" do
+          subject.db.should == :default
+        end
+      end
+    end
+  end
+end