make transaction management configurable

Author: Matt Remsik and Ethan Gunderson

Rakefile

@@ -5,6 +5,10 @@ $:.unshift File.expand_path(File.join(File.dirname(__FILE__),'lib'))
 
 require 'rake/rdoctask'
 require 'rspec/rails/version'
+require 'rspec'
+require 'rspec/core/rake_task'
+
+Rspec::Core::RakeTask.new(:spec)
 
 begin
   require 'jeweler'
@@ -87,5 +91,5 @@ task :clobber_app do
   rm_rf "tmp/example_app"
 end
 
-task :default => ["generate:app", "generate:stuff", :run_specs]
+task :default => [:spec, "generate:app", "generate:stuff", :run_specs]
 

lib/generators/rspec/install/templates/spec/spec_helper.rb

@@ -17,4 +17,8 @@
   # config.mock_with :flexmock
   # config.mock_with :rr
   config.mock_with :rspec
+
+  # If you'd prefer not to run each of your examples within a transaction,
+  # uncomment the following line.
+  # config.use_transactional_examples false
 end

lib/rspec/rails/transactional_database_support.rb

@@ -8,15 +8,21 @@ def active_record_configured?
         defined?(::ActiveRecord) && !::ActiveRecord::Base.configurations.blank?
       end
 
+      def use_transactional_examples?
+        Rspec.configuration.use_transactional_examples?
+      end
+
       def transactional_protection_start
         return unless active_record_configured?
+        return unless use_transactional_examples?
 
         ::ActiveRecord::Base.connection.increment_open_transactions
         ::ActiveRecord::Base.connection.begin_db_transaction
       end
 
       def transactional_protection_cleanup
         return unless active_record_configured?
+        return unless use_transactional_examples?
 
         if ::ActiveRecord::Base.connection.open_transactions != 0
           ::ActiveRecord::Base.connection.rollback_db_transaction

rspec-rails.gemspec

@@ -9,7 +9,7 @@ Gem::Specification.new do |s|
 
   s.required_rubygems_version = Gem::Requirement.new("> 1.3.1") if s.respond_to? :required_rubygems_version=
   s.authors = ["David Chelimsky", "Chad Humphries"]
-  s.date = %q{2010-03-15}
+  s.date = %q{2010-03-23}
   s.description = %q{Rspec-2 for Rails-3}
   s.email = %q{[email protected];[email protected]}
   s.extra_rdoc_files = [
@@ -63,6 +63,8 @@ Gem::Specification.new do |s|
      "lib/rspec/rails/transactional_database_support.rb",
      "lib/rspec/rails/version.rb",
      "rspec-rails.gemspec",
+     "spec/lib/rspec/rails/transactional_database_support_spec.rb",
+     "spec/spec_helper.rb",
      "templates/generate_stuff.rb",
      "templates/run_specs.rb"
   ]
@@ -85,6 +87,10 @@ Gem::Specification.new do |s|
   s.rubyforge_project = %q{rspec}
   s.rubygems_version = %q{1.3.6}
   s.summary = %q{rspec-rails-2.0.0.beta.4}
+  s.test_files = [
+    "spec/lib/rspec/rails/transactional_database_support_spec.rb",
+     "spec/spec_helper.rb"
+  ]
 
   if s.respond_to? :specification_version then
     current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION

spec/lib/rspec/rails/transactional_database_support_spec.rb

@@ -0,0 +1,62 @@
+require 'spec_helper'
+
+describe Rspec::Rails::TransactionalDatabaseSupport do
+  class TestBed
+    include Rspec::Rails::TransactionalDatabaseSupport
+  end
+
+  let(:test_bed) { TestBed.new }
+  let(:connection) { stub() }
+
+  describe "#transactional_protection_start" do
+    before do
+      test_bed.stub(:active_record_configured?).and_return(true)
+      ::ActiveRecord::Base.stub(:connection).and_return(connection)
+    end
+
+    context 'when running with examples with transactions' do
+      it "opens a new transaction" do
+        test_bed.stub(:use_transactional_examples?).and_return true
+        connection.should_receive(:increment_open_transactions)
+        connection.should_receive(:begin_db_transaction)
+        test_bed.transactional_protection_start
+      end
+    end
+
+    context 'with transactionless examples' do
+
+      it "doesn't open a transaction" do
+        test_bed.stub(:use_transactional_examples?).and_return false
+        connection.should_not_receive(:increment_open_transactions)
+        connection.should_not_receive(:begin_db_transaction)
+        test_bed.transactional_protection_start
+      end
+    end
+  end
+
+  describe "#transactional_protection_cleanup" do
+    before do
+      test_bed.stub(:active_record_configured?).and_return(true)
+      connection.stub(:open_transactions).and_return(1)
+      ::ActiveRecord::Base.stub(:connection).and_return(connection)
+    end
+
+    context 'when running with examples with transactions' do
+      it "opens a new transaction" do
+        test_bed.stub(:use_transactional_examples?).and_return(true)
+        connection.should_receive(:rollback_db_transaction)
+        connection.should_receive(:decrement_open_transactions)
+        test_bed.transactional_protection_cleanup
+      end
+    end
+
+    context 'with transactionless examples' do
+      it "doesn't close an open transaction" do
+        test_bed.stub(:use_transactional_examples?).and_return(false)
+        connection.should_not_receive(:decrement_open_transactions)
+        connection.should_not_receive(:rollback_db_transaction)
+        test_bed.transactional_protection_cleanup
+      end
+    end
+  end
+end

spec/spec_helper.rb

@@ -0,0 +1,14 @@
+%w(actionmailer actionpack activemodel activerecord activeresource activesupport railties).each do |directory|
+  $LOAD_PATH.unshift File.expand_path( File.join(File.dirname(__FILE__), "..", "tmp", "rails", directory, "lib") )
+end
+
+require 'i18n'
+require 'rack'
+require 'rack/mock'
+require 'rack/mime'
+require 'active_support'
+require 'active_support/core_ext'
+require 'action_dispatch'
+require 'action_controller'
+require 'active_record'
+require 'rspec/rails'