From 80408434da46b3101f574217ef2cfb683056b5a4 Mon Sep 17 00:00:00 2001 From: Alberto Perdomo Date: Wed, 11 Mar 2009 18:46:00 +0000 Subject: [PATCH] Added truncation support for Postgresql. --- lib/database_cleaner/active_record/truncation.rb | 7 +++++++ spec/database_cleaner/active_record/truncation_spec.rb | 2 +- 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/lib/database_cleaner/active_record/truncation.rb b/lib/database_cleaner/active_record/truncation.rb index 95c0fe4..b94771d 100644 --- a/lib/database_cleaner/active_record/truncation.rb +++ b/lib/database_cleaner/active_record/truncation.rb @@ -19,6 +19,13 @@ module ActiveRecord end end + class PostgreSQLAdapter + def truncate_table(table_name) + execute("TRUNCATE TABLE #{table_name};") + end + end + + end end diff --git a/spec/database_cleaner/active_record/truncation_spec.rb b/spec/database_cleaner/active_record/truncation_spec.rb index 287b690..41b5782 100644 --- a/spec/database_cleaner/active_record/truncation_spec.rb +++ b/spec/database_cleaner/active_record/truncation_spec.rb @@ -3,7 +3,7 @@ require 'database_cleaner/active_record/truncation' require 'active_record' module ActiveRecord module ConnectionAdapters - [MysqlAdapter, SQLite3Adapter, JdbcAdapter].each do |adapter| + [MysqlAdapter, SQLite3Adapter, JdbcAdapter, PostgreSQLAdapter].each do |adapter| describe adapter, "#truncate_table" do it "should truncate the table" end