database_cleaner 0.5.2 → 0.6.0.rc.1

data/examples/config/database.yml

@@ -0,0 +1,7 @@
+--- 
+two: 
+  adapter: sqlite3
+  database: /Users/bmabey/Programming/ruby/database_cleaner/examples/features/support/../../db/activerecord_two.db
+one: 
+  adapter: sqlite3
+  database: /Users/bmabey/Programming/ruby/database_cleaner/examples/features/support/../../db/activerecord_one.db

data/examples/config/database.yml.example

@@ -0,0 +1,8 @@
+#This is an example of what database.yml *should* look like (when I wrote it)
+#The real database.yml is generated automatically by the active record model lib (so it can be correct)
+two: 
+  adapter: sqlite3
+  database: /path/to/examples/features/support/../../db/activerecord_two.db
+one: 
+  adapter: sqlite3
+  database: /path/to/examples/features/support/../../db/activerecord_one.db

data/examples/features/example_multiple_db.feature

@@ -0,0 +1,23 @@
+Feature: example
+ In order to test DataBase Cleaner
+ Here are some scenarios that rely on the DB being clean!
+
+  # Background:
+  #   Given I have setup DatabaseCleaner to clean multiple databases
+  #
+  Scenario: dirty the db
+    When I create a widget in one db
+     And I create a widget in another db
+    Then I should see 1 widget in one db
+     And I should see 1 widget in another db
+
+  Scenario: assume a clean db
+    When I create a widget in one db
+    Then I should see 1 widget in one db
+     And I should see 0 widget in another db
+
+  Scenario: assume a clean db
+    When I create a widget in another db
+    Then I should see 0 widget in one db
+     And I should see 1 widget in another db
+

data/examples/features/example_multiple_orm.feature

@@ -0,0 +1,22 @@
+Feature: example
+ In order to test DataBase Cleaner
+ Here are some scenarios that rely on the DB being clean!
+
+ # Background:
+ #   Given I have setup DatabaseCleaner to clean multiple orms
+
+  Scenario: dirty the db
+    When I create a widget in one orm
+     And I create a widget in another orm
+    Then I should see 1 widget in one orm
+     And I should see 1 widget in another orm
+
+  Scenario: assume a clean db
+    When I create a widget in one orm
+    Then I should see 1 widget in one orm
+     And I should see 0 widget in another orm
+
+  Scenario: assume a clean db
+    When I create a widget in another orm
+    Then I should see 0 widget in one orm
+     And I should see 1 widget in another orm

data/examples/features/step_definitions/activerecord_steps.rb

@@ -0,0 +1,31 @@
+Given /^I have setup database cleaner to clean multiple databases using activerecord$/ do
+  #DatabaseCleaner
+  # require "#{File.dirname(__FILE__)}/../../../lib/datamapper_models"
+  #
+  # DatabaseCleaner[:datamapper, {:connection => :one} ].strategy = :truncation
+  # DatabaseCleaner[:datamapper, {:connection => :two} ].strategy = :truncation
+end
+
+When /^I create a widget using activerecord$/ do
+  ActiveRecordWidget.create!
+end
+
+Then /^I should see ([\d]+) widget using activerecord$/ do |widget_count|
+  ActiveRecordWidget.count.should == widget_count.to_i
+end
+
+When /^I create a widget in one db using activerecord$/ do
+  ActiveRecordWidgetUsingDatabaseOne.create!
+end
+
+When /^I create a widget in another db using activerecord$/ do
+  ActiveRecordWidgetUsingDatabaseTwo.create!
+end
+
+Then /^I should see ([\d]+) widget in one db using activerecord$/ do |widget_count|
+  ActiveRecordWidgetUsingDatabaseOne.count.should == widget_count.to_i
+end
+
+Then /^I should see ([\d]+) widget in another db using activerecord$/ do |widget_count|
+  ActiveRecordWidgetUsingDatabaseTwo.count.should == widget_count.to_i
+end

data/examples/features/step_definitions/couchpotato_steps.rb

@@ -0,0 +1,31 @@
+Given /^I have setup database cleaner to clean multiple databases using couchpotato$/ do
+  #DatabaseCleaner
+  # require "#{File.dirname(__FILE__)}/../../../lib/couchpotato_models"
+  #
+  # DatabaseCleaner[:couchpotato, {:connection => :one} ].strategy = :truncation
+  # DatabaseCleaner[:couchpotato, {:connection => :two} ].strategy = :truncation
+end
+
+When /^I create a widget using couchpotato$/ do
+  CouchPotatoWidget.create!
+end
+
+Then /^I should see ([\d]+) widget using couchpotato$/ do |widget_count|
+  CouchPotatoWidget.count.should == widget_count.to_i
+end
+
+When /^I create a widget in one db using couchpotato$/ do
+  CouchPotatoWidgetUsingDatabaseOne.create!
+end
+
+When /^I create a widget in another db using couchpotato$/ do
+  CouchPotatoWidgetUsingDatabaseTwo.create!
+end
+
+Then /^I should see ([\d]+) widget in one db using couchpotato$/ do |widget_count|
+  CouchPotatoWidgetUsingDatabaseOne.count.should == widget_count.to_i
+end
+
+Then /^I should see ([\d]+) widget in another db using couchpotato$/ do |widget_count|
+  CouchPotatoWidgetUsingDatabaseTwo.count.should == widget_count.to_i
+end

data/examples/features/step_definitions/datamapper_steps.rb

@@ -0,0 +1,37 @@
+Given /^I have setup database cleaner to clean multiple databases using datamapper$/ do
+  #DatabaseCleaner
+  # require "#{File.dirname(__FILE__)}/../../../lib/datamapper_models"
+  #
+  # DatabaseCleaner[:datamapper, {:connection => :one} ].strategy = :truncation
+  # DatabaseCleaner[:datamapper, {:connection => :two} ].strategy = :truncation
+end
+
+When /^I create a widget using datamapper$/ do
+  DataMapperWidget.create!
+end
+
+Then /^I should see ([\d]+) widget using datamapper$/ do |widget_count|
+  DataMapperWidget.count.should == widget_count.to_i
+end
+
+When /^I create a widget in one db using datamapper$/ do
+  begin
+    DataMapperWidgetUsingDatabaseOne.create!
+  rescue StandardError => e
+    BREAK = e.backtrace
+    debugger
+    DataMapperWidgetUsingDatabaseOne.create!
+  end
+end
+
+When /^I create a widget in another db using datamapper$/ do
+  DataMapperWidgetUsingDatabaseTwo.create!
+end
+
+Then /^I should see ([\d]+) widget in one db using datamapper$/ do |widget_count|
+  DataMapperWidgetUsingDatabaseOne.count.should == widget_count.to_i
+end
+
+Then /^I should see ([\d]+) widget in another db using datamapper$/ do |widget_count|
+  DataMapperWidgetUsingDatabaseTwo.count.should == widget_count.to_i
+end

data/examples/features/step_definitions/mongoid_steps.rb

@@ -0,0 +1,23 @@
+When /^I create a widget using mongoid$/ do
+  MongoidWidget.create!( :id => rand(1000)+1000)
+end
+
+Then /^I should see ([\d]+) widget using mongoid$/ do |widget_count|
+  MongoidWidget.count.should == widget_count.to_i
+end
+
+When /^I create a widget in one db using mongoid$/ do
+  MongoidWidgetUsingDatabaseOne.create!
+end
+
+When /^I create a widget in another db using mongoid$/ do
+  MongoidWidgetUsingDatabaseTwo.create!
+end
+
+Then /^I should see ([\d]+) widget in one db using mongoid$/ do |widget_count|
+  MongoidWidgetUsingDatabaseOne.count.should == widget_count.to_i
+end
+
+Then /^I should see ([\d]+) widget in another db using mongoid$/ do |widget_count|
+  MongoidWidgetUsingDatabaseTwo.count.should == widget_count.to_i
+end

data/examples/features/step_definitions/mongomapper_steps.rb

@@ -0,0 +1,31 @@
+Given /^I have setup database cleaner to clean multiple databases using mongomapper$/ do
+  #DatabaseCleaner
+  # require "#{File.dirname(__FILE__)}/../../../lib/datamapper_models"
+  #
+  # DatabaseCleaner[:datamapper, {:connection => :one} ].strategy = :truncation
+  # DatabaseCleaner[:datamapper, {:connection => :two} ].strategy = :truncation
+end
+
+When /^I create a widget using mongomapper$/ do
+  MongoMapperWidget.create!
+end
+
+Then /^I should see ([\d]+) widget using mongomapper$/ do |widget_count|
+  MongoMapperWidget.count.should == widget_count.to_i
+end
+
+When /^I create a widget in one db using mongomapper$/ do
+  MongoMapperWidgetUsingDatabaseOne.create!
+end
+
+When /^I create a widget in another db using mongomapper$/ do
+  MongoMapperWidgetUsingDatabaseTwo.create!
+end
+
+Then /^I should see ([\d]+) widget in one db using mongomapper$/ do |widget_count|
+  MongoMapperWidgetUsingDatabaseOne.count.should == widget_count.to_i
+end
+
+Then /^I should see ([\d]+) widget in another db using mongomapper$/ do |widget_count|
+  MongoMapperWidgetUsingDatabaseTwo.count.should == widget_count.to_i
+end

data/examples/features/step_definitions/translation_steps.rb

@@ -0,0 +1,55 @@
+When /^I create a widget$/ do
+  When "I create a widget using #{ENV['ORM'].downcase}"
+end
+
+Then /^I should see 1 widget$/ do
+  Then "I should see 1 widget using #{ENV['ORM'].downcase}"
+end
+
+When /^I create a widget in one orm$/ do
+  When "I create a widget using #{ENV['ORM'].downcase}"
+end
+
+When /^I create a widget in another orm$/ do
+  When "I create a widget using #{ENV['ANOTHER_ORM'].downcase}"
+end
+
+Then /^I should see 1 widget in one orm$/ do
+  When "I should see 1 widget using #{ENV['ORM'].downcase}"
+end
+
+Then /^I should see 1 widget in another orm$/ do
+  When "I should see 1 widget using #{ENV['ANOTHER_ORM'].downcase}"
+end
+
+Then /^I should see 0 widget in another orm$/ do
+  When "I should see 0 widget using #{ENV['ANOTHER_ORM'].downcase}"
+end
+
+Then /^I should see 0 widget in one orm$/ do
+  When "I should see 0 widget using #{ENV['ORM'].downcase}"
+end
+
+When /^I create a widget in one db$/ do
+  When "I create a widget in one db using #{ENV['ORM'].downcase}"
+end
+
+When /^I create a widget in another db$/ do
+  When "I create a widget in another db using #{ENV['ORM'].downcase}"
+end
+
+Then /^I should see 1 widget in one db$/ do
+  When "I should see 1 widget in one db using #{ENV['ORM'].downcase}"
+end
+
+Then /^I should see 1 widget in another db$/ do
+  When "I should see 1 widget in another db using #{ENV['ORM'].downcase}"
+end
+
+Then /^I should see 0 widget in another db$/ do
+  When "I should see 0 widget in another db using #{ENV['ORM'].downcase}"
+end
+
+Then /^I should see 0 widget in one db$/ do
+  When "I should see 0 widget in one db using #{ENV['ORM'].downcase}"
+end

data/examples/features/support/env.rb

@@ -1,23 +1,62 @@
+#Hilarious as it seems, this is necessary so bundle exec cucumber works for mongoid cukeage (I'm assuming mongomapper is automatically present because its a git repo)
+Object.send(:remove_const, 'MongoMapper')  if defined?(::MongoMapper)
+
 require 'rubygems'
+require 'bundler'
+
+Bundler.setup
 require 'spec/expectations'
+require 'ruby-debug'
+
+DB_DIR = "#{File.dirname(__FILE__)}/../../db"
+
+orm         = ENV['ORM']
+another_orm = ENV['ANOTHER_ORM']
+strategy    = ENV['STRATEGY']
+multiple_db = ENV['MULTIPLE_DBS']
 
-orm      = ENV['ORM']
-strategy = ENV['STRATEGY']
 
 if orm && strategy
+  $:.unshift(File.dirname(__FILE__) + '/../../../lib')
+  require 'database_cleaner'
+  require 'database_cleaner/cucumber'
 
   begin
-    require "#{File.dirname(__FILE__)}/../../lib/#{orm}_models"
-  rescue LoadError
+    require "#{File.dirname(__FILE__)}/../../lib/#{orm.downcase}_models"
+  rescue LoadError => e
     raise "You don't have the #{orm} ORM installed"
   end
 
-  $:.unshift(File.dirname(__FILE__) + '/../../../lib')
-  require 'database_cleaner'
-  require 'database_cleaner/cucumber'
+  if another_orm
+     begin
+      require "#{File.dirname(__FILE__)}/../../lib/#{another_orm.downcase}_models"
+    rescue LoadError => e
+      raise "You don't have the #{another_orm} ORM installed"
+    end
+  end
+
+
+
+
+  if multiple_db 
+    DatabaseCleaner.app_root = "#{File.dirname(__FILE__)}/../.."
+    orm_sym = orm.gsub(/(.)([A-Z]+)/,'\1_\2').downcase.to_sym
+
+    if orm_sym == :mongo_mapper
+      DatabaseCleaner[ orm_sym, {:connection => 'database_cleaner_test_one'} ].strategy = strategy.to_sym
+      DatabaseCleaner[ orm_sym, {:connection => 'database_cleaner_test_two'} ].strategy = strategy.to_sym
+    else
+      DatabaseCleaner[ orm_sym, {:connection => :one} ].strategy = strategy.to_sym
+      DatabaseCleaner[ orm_sym, {:connection => :two} ].strategy = strategy.to_sym
+    end
+
+  elsif another_orm
+    DatabaseCleaner[         orm.gsub(/(.)([A-Z]+)/,'\1_\2').downcase.to_sym ].strategy = strategy.to_sym
+    DatabaseCleaner[ another_orm.gsub(/(.)([A-Z]+)/,'\1_\2').downcase.to_sym ].strategy = strategy.to_sym
+  else
+    DatabaseCleaner.strategy = strategy.to_sym
+  end
 
-  DatabaseCleaner.strategy = strategy.to_sym
-  
 else
-  raise "Run 'ORM=activerecord|datamapper|mongomapper|couchpotato STRATEGY=transaction|truncation cucumber examples/features'"
+  raise "Run 'ORM=ActiveRecord|DataMapper|MongoMapper|CouchPotato [ANOTHER_ORM=...] [MULTIPLE_DBS=true] STRATEGY=transaction|truncation cucumber examples/features'"
 end

data/examples/lib/activerecord_models.rb

@@ -1,12 +1,41 @@
 require 'active_record'
+databases_config = {
+  "one" => {"adapter" => "#{"jdbc" if defined?(JRUBY_VERSION)}sqlite3", "database" => "#{DB_DIR}/activerecord_one.db"},
+  "two" => {"adapter" => "#{"jdbc" if defined?(JRUBY_VERSION)}sqlite3", "database" => "#{DB_DIR}/activerecord_two.db"}
+}
 
-ActiveRecord::Base.establish_connection(:adapter => "#{"jdbc" if defined?(JRUBY_VERSION)}sqlite3", :database => ":memory:")
+File.open("#{File.dirname(__FILE__)}/../config/database.yml", 'w') do |file|
+  file.write(YAML.dump(databases_config))
+end
+
+["two","one"].each do |db|
+  ActiveRecord::Base.establish_connection(databases_config[db])
+  ActiveRecord::Base.connection.execute('DROP TABLE IF EXISTS "active_record_widgets"')
+  ActiveRecord::Base.connection.execute('DROP TABLE IF EXISTS "active_record_widget_using_database_ones"')
+  ActiveRecord::Base.connection.execute('DROP TABLE IF EXISTS "active_record_widget_using_database_twos"')
+
+  ActiveRecord::Schema.define(:version => 1) do
+    create_table :active_record_widgets do |t|
+      t.string :name
+    end
+
+    create_table :active_record_widget_using_database_ones do |t|
+      t.string :name
+    end
 
-ActiveRecord::Schema.define(:version => 1) do
-  create_table :widgets do |t|
-    t.string :name
+    create_table :active_record_widget_using_database_twos do |t|
+      t.string :name
+    end
   end
 end
 
-class Widget < ActiveRecord::Base
+class ActiveRecordWidget < ActiveRecord::Base
+end
+
+class ActiveRecordWidgetUsingDatabaseOne < ActiveRecord::Base
+  establish_connection(:adapter => "#{"jdbc" if defined?(JRUBY_VERSION)}sqlite3", :database => "#{DB_DIR}/activerecord_one.db")
+end
+
+class ActiveRecordWidgetUsingDatabaseTwo < ActiveRecord::Base
+  establish_connection(:adapter => "#{"jdbc" if defined?(JRUBY_VERSION)}sqlite3", :database => "#{DB_DIR}/activerecord_two.db")
 end

data/examples/lib/couchpotato_models.rb

@@ -1,21 +1,61 @@
 require 'couch_potato'
-
+require 'json/pure' unless defined? ::JSON
 ::CouchPotato::Config.database_name = 'couch_potato_test'
 
-class Widget
+class CouchPotatoWidget
+  include CouchPotato::Persistence
+
+  property :name
+  view :by_name, :key => :name
+
+
+  # mimic the AR interface used in example_steps
+
+  def self.create!(attrs = {})
+    CouchPotato.database.save(self.new)
+  end
+
+  def self.count
+    CouchPotato.database.view(self.by_name).size
+  end
+end
+
+class CouchPotatoWidgetUsingDatabaseOne
+  include CouchPotato::Persistence
+
+  database_name = 'couch_potato_test_one'
+
+  property :name
+  view :by_name, :key => :name
+
+
+  # mimic the AR interface used in example_steps
+
+  def self.create!(attrs = {})
+    CouchPotato.database.save(self.new)
+  end
+
+  def self.count
+    CouchPotato.database.view(self.by_name).size
+  end
+end
+
+class CouchPotatoWidgetUsingDatabaseTwo
   include CouchPotato::Persistence
-  
+
+  database_name = 'couch_potato_test_two'
+
   property :name
   view :by_name, :key => :name
-  
+
 
   # mimic the AR interface used in example_steps
 
   def self.create!(attrs = {})
     CouchPotato.database.save(self.new)
   end
-  
+
   def self.count
-    CouchPotato.database.view(::Widget.by_name).size
+    CouchPotato.database.view(self.by_name).size
   end
 end