Fix truncation for PostgreSQL

Fix suggested by bjeanes on GH #15
DatabaseCleaner · Oct 4, 2010 · 7159f90 · 7159f90
1 parent 57807b0
commit 7159f90
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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