examples/lib/datamapper.rb lib/database_cleaner/data_mapper/truncation.rb @@ -14,3 +14,5 @@ Feature: database cleaning | ORM | Strategy | | ActiveRecord | transaction | | ActiveRecord | truncation | + | DataMapper | transaction | + | DataMapper | truncation | \ No newline at end of file features/cleaning.feature @@ -12,7 +12,7 @@ module DatabaseCleaner module DataMapper def self.available_strategies - %w[] + %w[truncation transaction] end end @@ -69,14 +69,14 @@ module DatabaseCleaner def orm @orm ||=begin - if defined? ::ActiveRecord - 'active_record' - elsif defined? ::DataMapper - 'data_mapper' - else - raise NoORMDetected, "No known ORM was detected! Is ActiveRecord or DataMapper loaded?" - end - end + if defined? ::ActiveRecord + 'active_record' + elsif defined? ::DataMapper + 'data_mapper' + else + raise NoORMDetected, "No known ORM was detected! Is ActiveRecord or DataMapper loaded?" + end + end end lib/database_cleaner/configuration.rb @@ -1,6 +1,23 @@ module DatabaseCleaner::DataMapper class Transaction - end + def start(repo = :default) + repository(repo) do |r| + transaction = DataMapper::Transaction.new(r) + transaction.begin + r.adapter.push_transaction(transaction) + end + end + + def clean(repo = :default) + repository(repo) do |r| + adapter = r.adapter + while adapter.current_transaction + adapter.current_transaction.rollback + adapter.pop_transaction + end + end + end + end end lib/database_cleaner/data_mapper/transaction.rb af6237d95f15e78e1bbd7c5cf94ff9ba9870e247 snusnu gamsnjaga@gmail.com http://github.com/bmabey/database_cleaner/commit/63147b21cefca764075af50f54834dc33059682d 63147b21cefca764075af50f54834dc33059682d 2009-05-05T08:06:02-07:00 2009-05-05T08:06:02-07:00 basic support for transaction/truncation with datamapper 6fb1860890f606d869be350a9f43c26a9e8d640b snusnu gamsnjaga@gmail.com