Class: ActiveRecord::FixtureBuilder::Database

Inherits:
Object
  • Object
show all
Defined in:
lib/active_record/fixture_builder/database.rb

Instance Method Summary collapse

Constructor Details

#initialize(config) ⇒ Database

Returns a new instance of Database.



3
4
5
 
# File 'lib/active_record/fixture_builder/database.rb', line 3

def initialize(config)
  @config = config
end

Instance Method Details

#connection ⇒ Object 



7
8
9
 
# File 'lib/active_record/fixture_builder/database.rb', line 7

def connection
  @config.connection
end

#insert(table_name, fixture_name, record) ⇒ Object 



27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/active_record/fixture_builder/database.rb', line 27

def insert table_name, fixture_name, record
  return if record.blank?

  columns = Hash[connection.columns(table_name).map { |c| [c.name, c] }]
  table_name = connection.quote_table_name(table_name)

  column_names = []
  values = []

  record.each do |column_name, value|
    column_names << connection.quote_column_name(column_name)
    values << connection.quote(value, columns[column_name])
  end

  connection.insert "INSERT INTO #{table_name} (#{column_names.join(', ')}) VALUES (#{values.join(', ')})", "Inserting Fixture #{fixture_name.inspect}"
end

#reset_pk_sequence(table_name) ⇒ Object 



23
24
25
 
# File 'lib/active_record/fixture_builder/database.rb', line 23

def reset_pk_sequence table_name
  connection.reset_pk_sequence! table_name
end

#select_all(table_name) ⇒ Object 



44
45
46
 
# File 'lib/active_record/fixture_builder/database.rb', line 44

def select_all table_name
  connection.select_all("SELECT * FROM #{connection.quote_table_name(table_name)}")
end

#table_names ⇒ Object 



15
16
17
 
# File 'lib/active_record/fixture_builder/database.rb', line 15

def table_names
  connection.tables - @config.excluded_tables
end

#truncate(table_name) ⇒ Object 



11
12
13
 
# File 'lib/active_record/fixture_builder/database.rb', line 11

def truncate table_name
  connection.delete "DELETE FROM #{connection.quote_table_name(table_name)}"
end

#truncate_all_tables! ⇒ Object 



19
20
21
 
# File 'lib/active_record/fixture_builder/database.rb', line 19

def truncate_all_tables!
  table_names.each{|table_name| truncate table_name }
end