Use apid.DB interface
diff --git a/api_test.go b/api_test.go
index 485ee29..0a544f7 100644
--- a/api_test.go
+++ b/api_test.go
@@ -20,7 +20,7 @@
var _ = Describe("api", func() {
var tmpDir string
- var db *sql.DB
+ var db apid.DB
var server *httptest.Server
BeforeSuite(func() {
diff --git a/init.go b/init.go
index 3733719..b0a45e8 100644
--- a/init.go
+++ b/init.go
@@ -1,7 +1,6 @@
package apidVerifyApiKey
import (
- "database/sql"
"github.com/30x/apid"
)
@@ -48,7 +47,7 @@
return nil
}
-func createTables(db *sql.DB) {
+func createTables(db apid.DB) {
_, err := db.Exec(`
CREATE TABLE api_product (
id text,