From a527cfdc4edf7e27d3d7e90ac4d6bc80c8800ba4 Mon Sep 17 00:00:00 2001
From: Lachlan Dowding <lachlan@permafrost.space>
Date: Tue, 31 Oct 2023 23:10:25 +1000
Subject: [PATCH] Fix `DB::DriverSpecs#with_db` `connection_string` query param
 support (#192)

---
 src/spec.cr | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/spec.cr b/src/spec.cr
index 2423f80..ca9e8d0 100644
--- a/src/spec.cr
+++ b/src/spec.cr
@@ -396,7 +396,16 @@ module DB
     # :nodoc:
     def with_db(options = nil)
       @before.call
-      DB.open("#{connection_string}#{"?#{options}" if options}") do |db|
+
+      if options
+        uri = URI.parse connection_string
+        uri.query_params.merge! URI::Params.parse(options)
+        connection_string_with_options = uri.to_s
+      else
+        connection_string_with_options = connection_string
+      end
+
+      DB.open(connection_string_with_options) do |db|
         db.exec(sql_drop_table("table1"))
         db.exec(sql_drop_table("table2"))
         db.exec(sql_drop_table("person"))