From 7159f903bfbc0c6a75e2ae81cc98e8bb390f7b4d Mon Sep 17 00:00:00 2001 From: Gabriel Sobrinho Date: Mon, 4 Oct 2010 09:53:58 +0800 Subject: [PATCH] Fix truncation for PostgreSQL Fix suggested by bjeanes on GH #15 --- lib/database_cleaner/active_record/truncation.rb | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/lib/database_cleaner/active_record/truncation.rb b/lib/database_cleaner/active_record/truncation.rb index 7e6af769..d489530b 100755 --- a/lib/database_cleaner/active_record/truncation.rb +++ b/lib/database_cleaner/active_record/truncation.rb @@ -42,19 +42,19 @@ def truncate_table(table_name) class PostgreSQLAdapter < AbstractAdapter - def self.db_version - @db_version ||= connection.select_values( + def db_version + @db_version ||= select_values( "SELECT CHARACTER_VALUE FROM INFORMATION_SCHEMA.SQL_IMPLEMENTATION_INFO WHERE IMPLEMENTATION_INFO_NAME = 'DBMS VERSION' ").join.to_s end - def self.cascade + def cascade @cascade ||= db_version >= "08.02" ? "CASCADE" : "" end def truncate_table(table_name) - execute("TRUNCATE TABLE #{quote_table_name(table_name)} #{self.class.cascade};") + execute("TRUNCATE TABLE #{quote_table_name(table_name)} #{cascade};") end end