connect* (clj)

;;;; Copyright © 2011 Paul Stadig
;;;;
;;;; Licensed under the Apache License, Version 2.0 (the "License"); you may not
;;;; use this file except in compliance with the License.  You may obtain a copy
;;;; of the License at
;;;;
;;;;   http://www.apache.org/licenses/LICENSE-2.0
;;;;
;;;; Unless required by applicable law or agreed to in writing, software
;;;; distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
;;;; WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
;;;; License for the specific language governing permissions and limitations
;;;; under the License.
(ns migratus.test.database
  (:require [clojure.java.io :as io]
            [next.jdbc :as jdbc]
            [next.jdbc.result-set :as rs]
            [next.jdbc.quoted :as q]
            [next.jdbc.sql :as sql]
            [migratus.protocols :as proto]
            [migratus.core :as core]
            [clojure.test :refer :all]
            [migratus.database :refer :all :as db]
            [clojure.tools.logging :as log]
            [migratus.test.migration.edn :as test-edn]
            [migratus.test.migration.sql :as test-sql]
            [migratus.utils :as utils])
  (:import java.io.File
           java.sql.Connection
           java.util.jar.JarFile
           (java.util.concurrent CancellationException)))

(defn verify-data [config table-name]
  (let [db     (connect* (:db config))
        conn   (:connection db)
        result (sql/query conn
                          [(str "SELECT * from " table-name)]
                          {:builder-fn rs/as-unqualified-lower-maps})]
    (.close conn)
    result))

(deftest test-connect*-returns-a-connection
  (testing "connect* works with a ^java.sql.Connection"
    (let [ds (jdbc/get-datasource db-mem)]
      (with-open [connection (jdbc/get-connection ds)]
        (let [res (db/connect* {:connection connection})]
          (is (map? res) "connect* response is a map")
          (is (contains? res :connection) "connect* response contains :connection")
          (is (instance? Connection (:connection res))
              "connect* response has a ^java.sql.Connection")
          (is (= connection (:connection res))
              "connect* response contains the same connection we passed")))))

  (testing "connect* works with a ^javax.sql.DataSource"
    (let [ds (jdbc/get-datasource db-mem)
          res (db/connect* {:datasource ds})]
      (is (map? res) "connect* response is a map")
      (is (contains? res :connection) "connect* response contains :connection")
      (is (instance? Connection (:connection res))
          "connect* response has a ^java.sql.Connection")))

  (testing "connect* works with a db spec"
    (let [res (db/connect* db-mem)]
      (is (map? res) "connect* response is a map")
      (is (contains? res :connection) "connect* response contains :connection")
      (is (instance? Connection (:connection res))
          "connect* response has a ^java.sql.Connection"))))