Use DatabaseCleaner properly with MongoMapper

Jaymes Waters and Matt Gauger · mathias · commit ad89eb02e181 · 2011-09-23T16:33:10.000-05:00
- For both Cucumber and RSpec
diff --git a/features/step_definitions/answer_steps.rb b/features/step_definitions/answer_steps.rb
@@ -1,5 +1,5 @@
 Given /^that someone has provided "([^"]*)" as an answer for my question$/ do |solution|
-  other_user =  Fabricate(:user)
+  other_user = Fabricate(:user)
   other_user.save!
   @question.answers.build(:description => solution, :user => other_user).save!
 end
diff --git a/features/support/env.rb b/features/support/env.rb
@@ -31,14 +31,11 @@
 
 # Remove/comment out the lines below if your app doesn't have a database.
 # For some databases (like MongoDB and CouchDB) you may need to use :truncation instead.
-DatabaseCleaner.strategy = :truncation
+DatabaseCleaner.orm = "mongo_mapper"
+DatabaseCleaner[:mongo_mapper].strategy = :truncation
 
 Before do
-  MongoMapper.database.collections.each do |collection|
-    unless collection.name.match(/^system\./)
-      collection.remove
-    end
-  end
+  DatabaseCleaner[:mongo_mapper].clean
 end
 
 # You may also want to configure DatabaseCleaner to use different strategies for certain features and scenarios.
diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
@@ -0,0 +1,26 @@
+ENV["RAILS_ENV"] ||= 'test'
+require File.expand_path("../../config/environment", __FILE__)
+require 'rspec/rails'
+require 'database_cleaner'
+
+# Requires supporting ruby files with custom matchers and macros, etc,
+# in spec/support/ and its subdirectories.
+Dir[Rails.root.join("spec/support/**/*.rb")].each {|f| require f}
+
+RSpec.configure do |config|
+  # == Mock Framework
+  #
+  # If you prefer to use mocha, flexmock or RR, uncomment the appropriate line:
+  #
+  # config.mock_with :mocha
+  # config.mock_with :flexmock
+  # config.mock_with :rr
+  config.mock_with :rspec
+
+  DatabaseCleaner.orm = "mongo_mapper"
+  DatabaseCleaner[:mongo_mapper].strategy = :truncation
+
+  config.before(:each) do
+    DatabaseCleaner[:mongo_mapper].clean
+  end
+end
