bts-cucumber 0.1.13.2 → 0.1.13.3

data/lib/cucumber/active_record/world.rb

@@ -2,9 +2,14 @@
 # Adapted by Aslak Hellesøy
 # Modified by Brian Schroeder to only use ActiveRecord (without ActionController or ActionMailer)
 
-if defined?(ActiveRecord::Base)
+if !defined?(ActiveRecord::Base)
   throw "ActiveRecord::Base is not defined"
 end
+
+require 'test/unit'
+require 'active_record/fixtures'
+# client user needs to set Test::Unit::TestCase.fixture_path on their own
+
 require 'test/unit/testresult'
 
 # So that Test::Unit doesn't launch at the end - makes it think it has already been run.
@@ -15,7 +20,33 @@ $main = self
 module Cucumber #:nodoc:
   module ActiveRecord
     # All scenarios will execute in the context of a new instance of World.
-    class World
+    class World < Test::Unit::TestCase
+      
+      # From Rails' ActionController::IntegrationTest
+      class << self
+        def use_transactional_fixtures=(flag) #:nodoc:
+          @_use_transactional_fixtures = true
+          @use_transactional_fixtures = flag
+        end
+      
+        def use_instantiated_fixtures=(flag) #:nodoc:
+          @_use_instantiated_fixtures = true
+          @use_instantiated_fixtures = flag
+        end
+      
+        def use_transactional_fixtures #:nodoc:
+          @_use_transactional_fixtures ?
+            @use_transactional_fixtures :
+            superclass.use_transactional_fixtures
+        end
+      
+        def use_instantiated_fixtures #:nodoc:
+          @_use_instantiated_fixtures ?
+            @use_instantiated_fixtures :
+            superclass.use_instantiated_fixtures
+        end
+      end
+      
       self.use_transactional_fixtures = false
 
       def initialize #:nodoc:
@@ -27,20 +58,20 @@ module Cucumber #:nodoc:
       World.use_transactional_fixtures = true
       
       $main.Before do
-        if ActiveRecord::Base.connection.respond_to?(:increment_open_transactions)
-          ActiveRecord::Base.connection.increment_open_transactions
+        if ::ActiveRecord::Base.connection.respond_to?(:increment_open_transactions)
+          ::ActiveRecord::Base.connection.increment_open_transactions
         else
-          ActiveRecord::Base.send :increment_open_transactions
+          ::ActiveRecord::Base.send :increment_open_transactions
         end
-        ActiveRecord::Base.connection.begin_db_transaction
+        ::ActiveRecord::Base.connection.begin_db_transaction
       end
       
       $main.After do
-        ActiveRecord::Base.connection.rollback_db_transaction
-        if ActiveRecord::Base.connection.respond_to?(:decrement_open_transactions)
-          ActiveRecord::Base.connection.decrement_open_transactions
+        ::ActiveRecord::Base.connection.rollback_db_transaction
+        if ::ActiveRecord::Base.connection.respond_to?(:decrement_open_transactions)
+          ::ActiveRecord::Base.connection.decrement_open_transactions
         else
-          ActiveRecord::Base.send :decrement_open_transactions
+          ::ActiveRecord::Base.send :decrement_open_transactions
         end
       end
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: bts-cucumber
 version: !ruby/object:Gem::Version 
-  version: 0.1.13.2
+  version: 0.1.13.3
 platform: ruby
 authors: 
 - "Aslak Helles\xC3\xB8y"