From 15dc3115dd446b270bfc7982aa12bd619fac20c9 Mon Sep 17 00:00:00 2001 From: stellard Date: Thu, 14 Jun 2012 14:41:25 +0100 Subject: [PATCH] add multiple truncate --- lib/adapter_extensions/adapters/abstract_adapter.rb | 4 ++-- test/abstract_adapter_test.rb | 7 +++++++ 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/lib/adapter_extensions/adapters/abstract_adapter.rb b/lib/adapter_extensions/adapters/abstract_adapter.rb index 4131508..6bad213 100644 --- a/lib/adapter_extensions/adapters/abstract_adapter.rb +++ b/lib/adapter_extensions/adapters/abstract_adapter.rb @@ -5,7 +5,7 @@ module AdapterExtensions::AbstractAdapter def truncate(table_name, options=nil) statement = [ 'TRUNCATE TABLE', - table_name, + [table_name].flatten.join(','), options ].compact.join(' ') execute(statement) @@ -41,4 +41,4 @@ def do_bulk_load(file, table_name, options={}) raise NotImplementedError, "do_bulk_load is an abstract method" end -end \ No newline at end of file +end diff --git a/test/abstract_adapter_test.rb b/test/abstract_adapter_test.rb index 7276717..09a0374 100644 --- a/test/abstract_adapter_test.rb +++ b/test/abstract_adapter_test.rb @@ -32,6 +32,13 @@ def test_truncate assert_equal "TRUNCATE TABLE #{table_name}", adapter.query end + def test_multi_truncate + table_names = ['foo', 'bar'] + adapter.truncate(table_names) + assert_equal "TRUNCATE TABLE #{table_names.join(',')}", adapter.query + end + + def test_support_select_into_table_should_return_false assert_equal false, adapter.support_select_into_table? end