Make transactional fixtures optional.

david · Oct 25, 2008 · c72ba10 · c72ba10
1 parent 46c2425
commit c72ba10
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 23 deletions.
diff --git a/lib/generators/cucumber/templates/features/env.rb b/lib/generators/cucumber/templates/features/env.rb
@@ -16,5 +16,8 @@
 require "merb_cucumber/helpers/activerecord"
 <% end -%>
 
+# Uncomment if you want transactional fixtures
+# Merb::Test::World::Base.use_transactional_fixtures
+
 Merb.start_environment(:testing => true, :adapter => 'runner', :environment => ENV['MERB_ENV'] || 'test')
 
diff --git a/lib/merb_cucumber/helpers/activerecord.rb b/lib/merb_cucumber/helpers/activerecord.rb
@@ -4,9 +4,8 @@ module Merb
   module Test
     module Helpers
       module ActiveRecord
-
-        def self.included(base)
-
+        module ClassMethods
+          def use_transactional_fixtures
             # Let's set a transaction on the ActiveRecord connection when starting a new scenario
             $main.Before do
               if ActiveRecord::Base.connection.respond_to?(:increment_open_transactions)
@@ -26,15 +25,14 @@ def self.included(base)
                 ActiveRecord::Base.send :decrement_open_transactions
               end
             end
-
+          end
         end
-
       end
     end
 
     module World
       module Base
-        include Helpers::ActiveRecord
+        extend Helpers::ActiveRecord::ClassMethods
       end
     end
   end

diff --git a/lib/merb_cucumber/helpers/datamapper.rb b/lib/merb_cucumber/helpers/datamapper.rb
@@ -4,34 +4,32 @@ module Merb
   module Test
     module Helpers
       module DataMapper
-
-        def self.included(base)
-
-          $main.Before do
-            repository(:default) do
-              transaction = ::DataMapper::Transaction.new(repository)
-              transaction.begin
-              repository.adapter.push_transaction(transaction)
+        module ClassMethods
+          def use_transactional_fixtures
+            $main.Before do
+              repository(:default) do
+                transaction = ::DataMapper::Transaction.new(repository)
+                transaction.begin
+                repository.adapter.push_transaction(transaction)
+              end
             end
-          end
 
-          $main.After do
-            repository(:default) do
-              while repository.adapter.current_transaction
-                repository.adapter.current_transaction.rollback
-                repository.adapter.pop_transaction
+            $main.After do
+              repository(:default) do
+                while repository.adapter.current_transaction
+                  repository.adapter.current_transaction.rollback
+                  repository.adapter.pop_transaction
+                end
               end
             end
           end
-
         end
-
       end
     end
 
     module World
       module Base
-        include Helpers::DataMapper
+        extend Helpers::DataMapper::ClassMethods
       end
     end
   end