Class: Spree::Core::Fixtures

Inherits:
ActiveRecord::Fixtures
  • Object
show all
Defined in:
lib/spree/core/custom_fixtures.rb

Class Method Summary collapse

Class Method Details

.create_fixtures(fixtures_directory, table_names, class_names = {}) ⇒ Object

Replace this method to prevent the table being emptied on each call. Needed when both core & auth have user fixtures, see below for code commented out.



9
10
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
 
# File 'lib/spree/core/custom_fixtures.rb', line 9

def self.create_fixtures(fixtures_directory, table_names, class_names = {})
  table_names = [table_names].flatten.map { |n| n.to_s }
  table_names.each { |n|
    class_names[n.tr('/', '_').to_sym] = n.classify if n.include?('/')
  }

  # FIXME: Apparently JK uses this.
  connection = block_given? ? yield : ActiveRecord::Base.connection

  files_to_read = table_names.reject { |table_name|
    fixture_is_cached?(connection, table_name)
  }

  unless files_to_read.empty?
    connection.disable_referential_integrity do
      fixtures_map = {}

      fixture_files = files_to_read.map do |path|
        table_name = path.tr '/', '_'

        fixtures_map[path] = ActiveRecord::Fixtures.new(
          connection,
          table_name,
          class_names[table_name.to_sym] || table_name.classify,
          ::File.join(fixtures_directory, path))
      end

      all_loaded_fixtures.update(fixtures_map)

      connection.transaction(:requires_new => true) do
        fixture_files.each do |ff|
          conn = ff.model_class.respond_to?(:connection) ? ff.model_class.connection : connection
          table_rows = ff.table_rows

          # REMOVED BY SPREE
          # table_rows.keys.each do |table|
          #   conn.delete "DELETE FROM #{conn.quote_table_name(table)}", 'Fixture Delete'
          # end

          table_rows.each do |table_name,rows|
            rows.each do |row|
              conn.insert_fixture(row, table_name)
            end
          end
        end

        # Cap primary key sequences to max(pk).
        if connection.respond_to?(:reset_pk_sequence!)
          table_names.each do |table_name|
            connection.reset_pk_sequence!(table_name.tr('/', '_'))
          end
        end
      end

      cache_fixtures(connection, fixtures_map)
    end
  end
  cached_fixtures(connection, table_names)
end