allow setting Onyx::SQL::Repository#db with DB::Transaction instances

Author: repomaa

db_spec/pg/repository/db_spec.cr

@@ -0,0 +1,59 @@
+require "../pg_spec"
+require "../../repository_spec"
+
+describe "Repository(Postgres)#db=" do
+  repo = repo(:postgresql)
+
+  context "with DB::Database" do
+    it "sets the db to the given database" do
+      old_db = repo.db
+
+      DB.open(ENV["POSTGRESQL_URL"]) do |db|
+        repo.db = db
+        new_db = repo.db
+
+        new_db.should_not be old_db
+        new_db.should be_a(::DB::Database)
+        new_db.should be db
+      end
+
+      repo.db = old_db
+    end
+  end
+
+  context "with DB::Connection" do
+    it "sets the db to the given connection" do
+      old_db = repo.db
+
+      DB.connect(ENV["POSTGRESQL_URL"]) do |connection|
+        repo.db = connection
+        new_db = repo.db
+
+        new_db.should_not be old_db
+        new_db.should be_a(::DB::Connection)
+        new_db.should be connection
+      end
+
+      repo.db = old_db
+    end
+  end
+
+  context "with DB::Transaction" do
+    it "sets the db to the given transaction's connection" do
+      old_db = repo.db
+
+      DB.open(ENV["POSTGRESQL_URL"]) do |db|
+        db.transaction do |tx|
+          repo.db = tx
+          new_db = repo.db
+
+          new_db.should_not be old_db
+          new_db.should be_a(::DB::Connection)
+          new_db.should be tx.connection
+        end
+      end
+
+      repo.db = old_db
+    end
+  end
+end

db_spec/sqlite3/repository/db_spec.cr

@@ -0,0 +1,59 @@
+require "../sqlite3_spec"
+require "../../repository_spec"
+
+describe "Repository(Sqlite3)#db=" do
+  repo = repo(:sqlite3)
+
+  context "with DB::Database" do
+    it "sets the db to the given database" do
+      old_db = repo.db
+
+      DB.open(ENV["SQLITE3_URL"]) do |db|
+        repo.db = db
+        new_db = repo.db
+
+        new_db.should_not be old_db
+        new_db.should be_a(::DB::Database)
+        new_db.should be db
+      end
+
+      repo.db = old_db
+    end
+  end
+
+  context "with DB::Connection" do
+    it "sets the db to the given connection" do
+      old_db = repo.db
+
+      DB.connect(ENV["SQLITE3_URL"]) do |connection|
+        repo.db = connection
+        new_db = repo.db
+
+        new_db.should_not be old_db
+        new_db.should be_a(::DB::Connection)
+        new_db.should be connection
+      end
+
+      repo.db = old_db
+    end
+  end
+
+  context "with DB::Transaction" do
+    it "sets the db to the given transaction's connection" do
+      old_db = repo.db
+
+      DB.open(ENV["SQLITE3_URL"]) do |db|
+        db.transaction do |tx|
+          repo.db = tx
+          new_db = repo.db
+
+          new_db.should_not be old_db
+          new_db.should be_a(::DB::Connection)
+          new_db.should be tx.connection
+        end
+      end
+
+      repo.db = old_db
+    end
+  end
+end

src/onyx-sql/repository.cr

@@ -32,6 +32,10 @@ module Onyx::SQL
     def initialize(@db : ::DB::Database | ::DB::Connection, @logger : Logger = Logger::Standard.new)
     end
 
+    def db=(transaction : ::DB::Transaction)
+      self.db = transaction.connection
+    end
+
     protected def postgresql?
       {% if Object.all_subclasses.any? { |sc| sc.stringify == "PG::Driver" } %}
         return db.is_a?(PG::Driver)