From 1582731bb664b4b200e85528d1e1cdc5aff91ac9 Mon Sep 17 00:00:00 2001 From: Kamal Fariz Mahyuddin Date: Tue, 27 Jul 2010 12:18:30 +0900 Subject: [PATCH] Add Mysql2Adapter#truncate_table --- lib/database_cleaner/active_record/truncation.rb | 6 ++++++ spec/database_cleaner/active_record/truncation_spec.rb | 4 ++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/lib/database_cleaner/active_record/truncation.rb b/lib/database_cleaner/active_record/truncation.rb index 860e577..e6f9a84 100755 --- a/lib/database_cleaner/active_record/truncation.rb +++ b/lib/database_cleaner/active_record/truncation.rb @@ -18,6 +18,12 @@ module ActiveRecord end end + class Mysql2Adapter < AbstractAdapter + def truncate_table(table_name) + execute("TRUNCATE TABLE #{quote_table_name(table_name)};") + end + end + class SQLite3Adapter < SQLiteAdapter def truncate_table(table_name) execute("DELETE FROM #{quote_table_name(table_name)};") diff --git a/spec/database_cleaner/active_record/truncation_spec.rb b/spec/database_cleaner/active_record/truncation_spec.rb index 4e04905..defa621 100644 --- a/spec/database_cleaner/active_record/truncation_spec.rb +++ b/spec/database_cleaner/active_record/truncation_spec.rb @@ -5,8 +5,8 @@ require 'database_cleaner/active_record/truncation' module ActiveRecord module ConnectionAdapters - [MysqlAdapter, SQLite3Adapter, JdbcAdapter, PostgreSQLAdapter].each do |adapter| - describe(adapter, "#truncate_table") do + [MysqlAdapter, Mysql2Adapter, SQLite3Adapter, JdbcAdapter, PostgreSQLAdapter].each do |adapter| + describe adapter, "#truncate_table" do it "should truncate the table" end end