提交 026d0555 编写于 作者: A Aaron Patterson

ask the fixture set for the sql statements

上级 9f8762f1
......@@ -286,6 +286,10 @@ def reset_sequence!(table, column, sequence = nil)
# Inserts the given fixture into the table. Overridden in adapters that require
# something beyond a simple insert (eg. Oracle).
def insert_fixture(fixture, table_name)
execute fixture_sql(fixture, table_name), 'Fixture Insert'
end
def fixture_sql(fixture, table_name)
columns = schema_cache.columns_hash(table_name)
key_list = []
......@@ -294,7 +298,7 @@ def insert_fixture(fixture, table_name)
quote(value, columns[name])
end
execute "INSERT INTO #{quote_table_name(table_name)} (#{key_list.join(', ')}) VALUES (#{value_list.join(', ')})", 'Fixture Insert'
"INSERT INTO #{quote_table_name(table_name)} (#{key_list.join(', ')}) VALUES (#{value_list.join(', ')})"
end
def empty_insert_statement_value
......
......@@ -504,16 +504,8 @@ def self.create_fixtures(fixtures_directory, fixture_set_names, class_names = {}
connection.transaction(:requires_new => true) do
fixture_sets.each do |fs|
conn = fs.model_class.respond_to?(:connection) ? fs.model_class.connection : connection
table_rows = fs.table_rows
table_rows.keys.each do |table|
conn.delete "DELETE FROM #{conn.quote_table_name(table)}", 'Fixture Delete'
end
table_rows.each do |fixture_set_name, rows|
rows.each do |row|
conn.insert_fixture(row, fixture_set_name)
end
fs.fixture_sql(conn).each do |stmt|
conn.execute stmt
end
end
......@@ -580,6 +572,16 @@ def size
fixtures.size
end
def fixture_sql(conn)
table_rows = self.table_rows
table_rows.keys.map { |table|
"DELETE FROM #{conn.quote_table_name(table)}"
}.concat table_rows.flat_map { |fixture_set_name, rows|
rows.map { |row| conn.fixture_sql(row, fixture_set_name) }
}
end
# Return a hash of rows to be inserted. The key is the table, the value is
# a list of rows to insert to that table.
def table_rows
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册