whitespace

This commit is contained in:
Ben Mabey 2009-05-08 18:03:29 -06:00
parent 9037010594
commit 612f33c738
3 changed files with 37 additions and 37 deletions

View file

@ -1,7 +1,7 @@
module ActiveRecord
module ConnectionAdapters
class MysqlAdapter
def truncate_table(table_name)
execute("TRUNCATE TABLE #{quote_table_name(table_name)};")
@ -50,7 +50,7 @@ module DatabaseCleaner::ActiveRecord
def connection
::ActiveRecord::Base.connection
end
# overwritten
def migration_storage_name
'schema_migrations'

View file

@ -1,5 +1,5 @@
module DatabaseCleaner
class NoStrategySetError < StandardError; end
class NoORMDetected < StandardError; end
class UnknownStrategySpecified < ArgumentError; end
@ -90,18 +90,18 @@ module DatabaseCleaner
end
end
# common base class for truncation strategies
class TruncationBase
def initialize(options = {})
if !options.empty? && !(options.keys - [:only, :except]).empty?
raise ArgumentError, "The only valid options are :only and :except. You specified #{options.keys.join(',')}."
end
if options.has_key?(:only) && options.has_key?(:except)
raise ArgumentError, "You may only specify either :only or :either. Doing both doesn't really make sense does it?"
raise ArgumentError, "You may only specify either :only or :either. Doing both doesn't really make sense does it?"
end
@only = options[:only]
@ -118,20 +118,20 @@ module DatabaseCleaner
def clean
raise NotImplementedError
end
private
def tables_to_truncate
raise NotImplementedError
end
# overwrite in subclasses
# default implementation given because migration storage need not be present
def migration_storage_name
nil
end
end
end

View file

@ -1,25 +1,25 @@
module DataMapper
module Adapters
class DataObjectsAdapter
def storage_names(repository = :default)
raise NotImplementedError
end
end
class MysqlAdapter < DataObjectsAdapter
# taken from http://github.com/godfat/dm-mapping/tree/master
def storage_names(repository = :default)
query 'SHOW TABLES'
end
def truncate_table(table_name)
execute("TRUNCATE TABLE #{quote_table_name(table_name)};")
end
# copied from activerecord
def disable_referential_integrity
old = query("SELECT @@FOREIGN_KEY_CHECKS;")
@ -30,11 +30,11 @@ module DataMapper
execute("SET FOREIGN_KEY_CHECKS = #{old};")
end
end
end
class Sqlite3Adapter < DataObjectsAdapter
# taken from http://github.com/godfat/dm-mapping/tree/master
def storage_names(repository = :default)
# activerecord-2.1.0/lib/active_record/connection_adapters/sqlite_adapter.rb: 177
@ -46,18 +46,18 @@ module DataMapper
# activerecord-2.1.0/lib/active_record/connection_adapters/sqlite_adapter.rb: 181
query sql
end
def truncate_table(table_name)
execute("DELETE FROM #{quote_table_name(table_name)};")
end
# this is a no-op copied from activerecord
# i didn't find out if/how this is possible
# activerecord also doesn't do more here
def disable_referential_integrity
yield
end
end
@ -67,7 +67,7 @@ module DataMapper
# i don't have postgres available, so i won't be the one to write this.
# maybe codes below gets some postgres/datamapper user going, though.
class PostgresAdapter < DataObjectsAdapter
# taken from http://github.com/godfat/dm-mapping/tree/master
def storage_names(repository = :default)
sql = <<-SQL.compress_lines
@ -76,11 +76,11 @@ module DataMapper
SQL
query(sql)
end
def truncate_table(table_name)
execute("TRUNCATE TABLE #{quote_table_name(table_name)};")
end
# FIXME
# copied from activerecord
def supports_disable_referential_integrity?
@ -89,24 +89,24 @@ module DataMapper
rescue
return false
end
# FIXME
# copied unchanged from activerecord
def disable_referential_integrity(repository = :default)
if supports_disable_referential_integrity? then
execute(storage_names(repository).collect do |name|
"ALTER TABLE #{quote_table_name(name)} DISABLE TRIGGER ALL"
execute(storage_names(repository).collect do |name|
"ALTER TABLE #{quote_table_name(name)} DISABLE TRIGGER ALL"
end.join(";"))
end
yield
ensure
if supports_disable_referential_integrity? then
execute(storage_names(repository).collect do |name|
"ALTER TABLE #{quote_table_name(name)} ENABLE TRIGGER ALL"
execute(storage_names(repository).collect do |name|
"ALTER TABLE #{quote_table_name(name)} ENABLE TRIGGER ALL"
end.join(";"))
end
end
end
end
@ -115,7 +115,7 @@ end
module DatabaseCleaner::DataMapper
class Truncation < ::DatabaseCleaner::TruncationBase
def clean(repository = :default)
adapter = DataMapper.repository(repository).adapter
adapter.disable_referential_integrity do
@ -124,17 +124,17 @@ module DatabaseCleaner::DataMapper
end
end
end
private
def tables_to_truncate(repository = :default)
(@only || DataMapper.repository(repository).adapter.storage_names(repository)) - @tables_to_exclude
end
# overwritten
def migration_storage_name
'migration_info'
end
end
end