Merge pull request #85 from zhaitz/main

Add basic support for trilogy

Author: etagwerker

database_cleaner-active_record.gemspec

@@ -28,4 +28,5 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "mysql2"
   spec.add_development_dependency "pg"
   spec.add_development_dependency "sqlite3"
+  spec.add_development_dependency "trilogy"
 end

lib/database_cleaner/active_record/deletion.rb

@@ -64,7 +64,7 @@ def build_table_stats_query(connection)
       end
 
       def information_schema_exists? connection
-        connection.adapter_name == "Mysql2"
+        ["Mysql2", "Trilogy"].include?(connection.adapter_name)
       end
     end
   end

lib/database_cleaner/active_record/truncation.rb

@@ -62,7 +62,7 @@ class ConnectionWrapper < SimpleDelegator
       def initialize(connection)
         extend AbstractAdapter
         case connection.adapter_name
-        when "Mysql2"
+        when "Mysql2", "Trilogy"
           extend AbstractMysqlAdapter
         when "SQLite"
           extend AbstractMysqlAdapter

spec/database_cleaner/active_record/deletion_spec.rb

@@ -91,7 +91,7 @@
           subject(:strategy) { described_class.new(cache_tables: true) }
 
           it 'caches the list of tables to be deleted from' do
-            if helper.db == :mysql2
+            if [:mysql2, :trilogy].include?(helper.db)
               expect(strategy).to receive(:build_table_stats_query).once.and_return("")
             elsif helper.db == :postgres
               expect(strategy.send(:connection)).to receive(:tables_with_schema).once.and_return([])
@@ -108,7 +108,7 @@
           subject(:strategy) { described_class.new(cache_tables: false) }
 
           it 'does not cache the list of tables to be deleted from' do
-            if helper.db == :mysql2
+            if [:mysql2, :trilogy].include?(helper.db)
               expect(strategy).to receive(:build_table_stats_query).twice.and_return("")
             elsif helper.db == :postgres
               expect(strategy.send(:connection)).to receive(:tables_with_schema).twice.and_return([])

spec/support/database_helper.rb

@@ -3,7 +3,7 @@
 
 class DatabaseHelper < DatabaseCleaner::Spec::DatabaseHelper
   def self.with_all_dbs &block
-    %w[mysql2 sqlite3 postgres].map(&:to_sym).each do |db|
+    %w[mysql2 sqlite3 postgres trilogy].map(&:to_sym).each do |db|
       yield new(db)
     end
   end