From f458536ac6bb09cf93f4f4b53b9fce3741b6c0c8 Mon Sep 17 00:00:00 2001
From: Peter Goldstein <peter.m.goldstein@gmail.com>
Date: Tue, 2 Jan 2024 09:49:09 -0500
Subject: [PATCH] Update for trilogy (#707)

---
 lib/database_cleaner/spec/database_helper.rb | 4 ++--
 spec/support/sample.config.yml               | 9 +++++++++
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/lib/database_cleaner/spec/database_helper.rb b/lib/database_cleaner/spec/database_helper.rb
index b1ae281e..e486a2f1 100644
--- a/lib/database_cleaner/spec/database_helper.rb
+++ b/lib/database_cleaner/spec/database_helper.rb
@@ -4,7 +4,7 @@ module DatabaseCleaner
   module Spec
     class DatabaseHelper < Struct.new(:db)
       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
@@ -43,7 +43,7 @@ def load_schema
         id_column = case db
           when :sqlite3
             "id INTEGER PRIMARY KEY AUTOINCREMENT"
-          when :mysql2
+          when :mysql2, :trilogy
             "id INTEGER PRIMARY KEY AUTO_INCREMENT"
           when :postgres
             "id SERIAL PRIMARY KEY"
diff --git a/spec/support/sample.config.yml b/spec/support/sample.config.yml
index bcbeb1de..06127152 100644
--- a/spec/support/sample.config.yml
+++ b/spec/support/sample.config.yml
@@ -7,6 +7,15 @@ mysql2:
   port: 3306
   encoding: utf8
 
+trilogy:
+  adapter: trilogy
+  database: database_cleaner_test
+  username: root
+  password:
+  host: 127.0.0.1
+  port: 3306
+  encoding: utf8
+
 postgres:
   adapter: postgresql
   database: database_cleaner_test