class ActiveRecord::FixtureBuilder::Database

Public Class Methods

new(config) click to toggle source
# File lib/active_record/fixture_builder/database.rb, line 3
def initialize(config)
  @config = config
end

Public Instance Methods

connection() click to toggle source
# File lib/active_record/fixture_builder/database.rb, line 7
def connection
  @config.connection
end
insert(table_name, fixture_name, record) click to toggle source
# 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) click to toggle source
# 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) click to toggle source
# 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() click to toggle source
# File lib/active_record/fixture_builder/database.rb, line 15
def table_names
  connection.tables - @config.excluded_tables
end
truncate(table_name) click to toggle source
# 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!() click to toggle source
# File lib/active_record/fixture_builder/database.rb, line 19
def truncate_all_tables!
  table_names.each{|table_name| truncate table_name }
end