Module: UseDb::TestFixtures

Defined in:
lib/use_db/override_test_case.rb

Overview

setup and teardown fixtures in all databases

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.included(base) ⇒ Object 



6
7
8
9
 
# File 'lib/use_db/override_test_case.rb', line 6

def self.included(base)
  base.alias_method_chain :setup_fixtures, :use_db
  base.alias_method_chain :teardown_fixtures, :use_db
end

Instance Method Details

#setup_fixtures_with_use_db ⇒ Object 



11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
 
# File 'lib/use_db/override_test_case.rb', line 11

def setup_fixtures_with_use_db
  UseDbPlugin.all_use_dbs.collect do |klass|
    return unless defined?(ActiveRecord) && !ActiveRecord::Base.configurations.blank?

    if pre_loaded_fixtures && !use_transactional_fixtures
      raise RuntimeError, 'pre_loaded_fixtures requires use_transactional_fixtures'
    end 

    @fixture_cache = {}
    @@already_loaded_fixtures ||= {}

    # Load fixtures once and begin transaction.
    if run_in_transaction?
      if @@already_loaded_fixtures[self.class]
        @loaded_fixtures = @@already_loaded_fixtures[self.class]
      else
        load_fixtures
        @@already_loaded_fixtures[self.class] = @loaded_fixtures
      end 
      klass.connection.increment_open_transactions
      klass.connection.transaction_joinable = false
      klass.connection.begin_db_transaction
    # Load fixtures for every test.
    else
      Fixtures.reset_cache
      @@already_loaded_fixtures[self.class] = nil 
      load_fixtures
    end 

    # Instantiate fixtures for every test if requested.
    instantiate_fixtures if use_instantiated_fixtures
  end
end

#teardown_fixtures_with_use_db ⇒ Object 



45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
 
# File 'lib/use_db/override_test_case.rb', line 45

def teardown_fixtures_with_use_db
  UseDbPlugin.all_use_dbs.collect do |klass|
    return unless defined?(ActiveRecord) && !ActiveRecord::Base.configurations.blank?

    unless run_in_transaction?
      Fixtures.reset_cache
    end

    # Rollback changes if a transaction is active.
    if run_in_transaction? && klass.connection.open_transactions != 0
      klass.connection.rollback_db_transaction
      klass.connection.decrement_open_transactions
    end
    klass.clear_active_connections!
  end
end