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