From 9bf038d5c287cca98612fb0a76e58147d3694868 Mon Sep 17 00:00:00 2001
From: Tom Meier <tom@venombytes.com>
Date: Tue, 14 May 2013 15:15:28 +1000
Subject: [PATCH] Fix MySqlAdapter superclass bug via class_eval loading of
 superclasses

---
 .../active_record/truncation.rb               | 44 ++++++-------------
 .../active_record/truncation_spec.rb          |  8 +++-
 spec/support/active_record/mysql2_setup.rb    |  2 +
 3 files changed, 21 insertions(+), 33 deletions(-)

diff --git a/lib/database_cleaner/active_record/truncation.rb b/lib/database_cleaner/active_record/truncation.rb
index ee8c7c4..7531012 100755
--- a/lib/database_cleaner/active_record/truncation.rb
+++ b/lib/database_cleaner/active_record/truncation.rb
@@ -176,56 +176,39 @@ module ActiveRecord
   module ConnectionAdapters
     # Activerecord-jdbc-adapter defines class dependencies a bit differently - if it is present, confirm to ArJdbc hierarchy to avoid 'superclass mismatch' errors.
     USE_ARJDBC_WORKAROUND = defined?(ArJdbc)
+    # ActiveRecord 3.1+ support
+    MYSQL_ABSTRACT_ADAPTER = defined?(AbstractMysqlAdapter) ? AbstractMysqlAdapter : AbstractAdapter
 
-    class AbstractAdapter
-      include ::DatabaseCleaner::ActiveRecord::AbstractAdapter
-    end
+    AbstractAdapter.send(:include, ::DatabaseCleaner::ActiveRecord::AbstractAdapter)
 
-    unless USE_ARJDBC_WORKAROUND
-      class SQLiteAdapter < AbstractAdapter
-      end
-    end
-
-    # ActiveRecord 3.1 support
-    if defined?(AbstractMysqlAdapter)
-      MYSQL_ADAPTER_PARENT = USE_ARJDBC_WORKAROUND ? JdbcAdapter : AbstractMysqlAdapter
-      MYSQL2_ADAPTER_PARENT = AbstractMysqlAdapter
+    if USE_ARJDBC_WORKAROUND
+      MYSQL_ADAPTER_PARENT  = JdbcAdapter
     else
-      MYSQL_ADAPTER_PARENT = USE_ARJDBC_WORKAROUND ? JdbcAdapter : AbstractAdapter
-      MYSQL2_ADAPTER_PARENT = AbstractAdapter
+      MYSQL_ADAPTER_PARENT  = MYSQL_ABSTRACT_ADAPTER
+      class SQLiteAdapter < AbstractAdapter; end
     end
+    MYSQL2_ADAPTER_PARENT = MYSQL_ABSTRACT_ADAPTER
 
     if defined?(SQLite3Adapter) && SQLite3Adapter.superclass == ActiveRecord::ConnectionAdapters::AbstractAdapter
       SQLITE_ADAPTER_PARENT = USE_ARJDBC_WORKAROUND ? JdbcAdapter : AbstractAdapter
     else
       SQLITE_ADAPTER_PARENT = USE_ARJDBC_WORKAROUND ? JdbcAdapter : SQLiteAdapter
     end
-    POSTGRE_ADAPTER_PARENT = USE_ARJDBC_WORKAROUND ? JdbcAdapter : AbstractAdapter
+    POSTGRES_ADAPTER_PARENT = USE_ARJDBC_WORKAROUND ? JdbcAdapter : AbstractAdapter
 
-    class MysqlAdapter < MYSQL_ADAPTER_PARENT
-      include ::DatabaseCleaner::ActiveRecord::MysqlAdapter
-    end
-
-    class Mysql2Adapter < MYSQL2_ADAPTER_PARENT
-      include ::DatabaseCleaner::ActiveRecord::MysqlAdapter
-    end
+    MYSQL_ADAPTER_PARENT.class_eval     { include ::DatabaseCleaner::ActiveRecord::MysqlAdapter }
+    MYSQL2_ADAPTER_PARENT.class_eval    { include ::DatabaseCleaner::ActiveRecord::MysqlAdapter }
+    SQLITE_ADAPTER_PARENT.class_eval    { include ::DatabaseCleaner::ActiveRecord::SQLiteAdapter }
+    POSTGRES_ADAPTER_PARENT.class_eval  { include ::DatabaseCleaner::ActiveRecord::PostgreSQLAdapter }
 
     class IBM_DBAdapter < AbstractAdapter
       include ::DatabaseCleaner::ActiveRecord::IBM_DBAdapter
     end
 
-    class SQLite3Adapter < SQLITE_ADAPTER_PARENT
-      include ::DatabaseCleaner::ActiveRecord::SQLiteAdapter
-    end
-
     class JdbcAdapter < AbstractAdapter
       include ::DatabaseCleaner::ActiveRecord::TruncateOrDelete
     end
 
-    class PostgreSQLAdapter < POSTGRE_ADAPTER_PARENT
-      include ::DatabaseCleaner::ActiveRecord::PostgreSQLAdapter
-    end
-
     class SQLServerAdapter < AbstractAdapter
       include ::DatabaseCleaner::ActiveRecord::TruncateOrDelete
     end
@@ -233,7 +216,6 @@ module ActiveRecord
     class OracleEnhancedAdapter < AbstractAdapter
       include ::DatabaseCleaner::ActiveRecord::OracleEnhancedAdapter
     end
-
   end
 end
 
diff --git a/spec/database_cleaner/active_record/truncation_spec.rb b/spec/database_cleaner/active_record/truncation_spec.rb
index d046791..a92a0cd 100644
--- a/spec/database_cleaner/active_record/truncation_spec.rb
+++ b/spec/database_cleaner/active_record/truncation_spec.rb
@@ -1,5 +1,9 @@
 require File.dirname(__FILE__) + '/../../spec_helper'
 require 'active_record'
+require 'active_record/connection_adapters/mysql_adapter'
+require 'active_record/connection_adapters/mysql2_adapter'
+require 'active_record/connection_adapters/sqlite3_adapter'
+require 'active_record/connection_adapters/postgresql_adapter'
 
 require 'database_cleaner/active_record/truncation'
 
@@ -119,14 +123,14 @@ module DatabaseCleaner
           subject.send(:pre_count?).should == false
         end
       end
-      
+
       describe '#reset_ids?' do
         before(:each) do
           connection.stub!(:disable_referential_integrity).and_yield
           connection.stub!(:database_cleaner_view_cache).and_return([])
           ::ActiveRecord::Base.stub!(:connection).and_return(connection)
         end
-        
+
         subject { Truncation.new }
         its(:reset_ids?) { should == true }
 
diff --git a/spec/support/active_record/mysql2_setup.rb b/spec/support/active_record/mysql2_setup.rb
index 86391d6..5b75b6e 100644
--- a/spec/support/active_record/mysql2_setup.rb
+++ b/spec/support/active_record/mysql2_setup.rb
@@ -1,6 +1,7 @@
 require 'support/active_record/database_setup'
 require 'support/active_record/schema_setup'
 
+
 module MySQL2Helper
   puts "Active Record #{ActiveRecord::VERSION::STRING}, mysql2"
 
@@ -36,3 +37,4 @@ end
 RSpec.configure do |c|
   c.include MySQL2Helper
 end
+