Test against all supported databases.

Author: Daniel Perez

.env

@@ -1,3 +1,4 @@
 /dbpopulate
 
 *.db
+.env

.gitignore

@@ -1,5 +1,20 @@
 language: go
 
+services:
+  - postgresql
+  - mysql
+
+before_script:
+  - sh -c "if [ '$DB' = 'postgres' ]; then psql -c 'create database dbpopulate_test;' -U postgres; fi"
+  - sh -c "if [ '$DB' = 'mysql' ]; then mysql -e 'create database dbpopulate_test;'; fi"
+  - go get -t -v ./...
+
 go:
   - 1.4
-  - tip
+  - 1.5
+  - 1.6
+
+env:
+  - DB=sqlite3 DATABASE_URL="sqlite3://:memory:"
+  - DB=postgres DATABASE_URL="postgres://postgres@localhost/dbpopulate_test?sslmode=disable"
+  - DB=mysql DATABASE_URL="mysql://root@tcp(localhost:3306)/dbpopulate_test"

.travis.yml

@@ -1,8 +1,10 @@
+DROP TABLE IF EXISTS countries;
 CREATE TABLE countries(
   id integer primary key auto_increment,
   name varchar(255)
 );
 
+DROP TABLE IF EXISTS regions;
 CREATE TABLE regions(
   id integer primary key auto_increment,
   name varchar(255),
@@ -11,6 +13,7 @@ CREATE TABLE regions(
 
 CREATE INDEX regions_country_id ON regions (country_id);
 
+DROP TABLE IF EXISTS prefectures;
 CREATE TABLE prefectures(
   id integer primary key auto_increment,
   name varchar(255),

db/tables.mysql.sql

@@ -1,8 +1,10 @@
+DROP TABLE IF EXISTS countries;
 CREATE TABLE countries(
   id serial primary key,
   name varchar(255)
 );
 
+DROP TABLE IF EXISTS regions;
 CREATE TABLE regions(
   id serial primary key,
   name varchar(255),
@@ -11,6 +13,7 @@ CREATE TABLE regions(
 
 CREATE INDEX regions_country_id ON regions (country_id);
 
+DROP TABLE IF EXISTS prefectures;
 CREATE TABLE prefectures(
   id serial primary key,
   name varchar(255),

db/tables.pg.sql renamed to db/tables.postgres.sql

@@ -1,8 +1,10 @@
+DROP TABLE IF EXISTS countries;
 CREATE TABLE countries(
   id integer primary key autoincrement,
   name varchar(255)
 );
 
+DROP TABLE IF EXISTS regions;
 CREATE TABLE regions(
   id integer primary key autoincrement,
   name varchar(255),
@@ -11,6 +13,7 @@ CREATE TABLE regions(
 
 CREATE INDEX regions_country_id ON regions (country_id);
 
+DROP TABLE IF EXISTS prefectures;
 CREATE TABLE prefectures(
   id integer primary key autoincrement,
   name varchar(255),

db/tables.sqlite3.sql

@@ -3,16 +3,17 @@ package main
 import (
 	"fmt"
 	"io/ioutil"
+	"net/url"
 	"os"
+	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 )
 
 const (
-	dbFile     = "db/data.sqlite.db"
-	dbUrl      = "sqlite3://" + dbFile
-	schemaFile = "./db/tables.sqlite3.sql"
+	dbURL              = "sqlite3://:memory:"
+	schemaFileTemplate = "./db/tables.%s.sql"
 )
 
 var populator *Populator
@@ -34,18 +35,38 @@ func showErrAndExitTest(err error) {
 }
 
 func TestMain(m *testing.M) {
-	os.Remove(dbFile)
 	var err error
-	if populator, err = NewPopulator(dbUrl); err != nil {
+
+	connectionURL := dbURL
+	if envURL := os.Getenv("DATABASE_URL"); envURL != "" {
+		connectionURL = envURL
+	}
+
+	if populator, err = NewPopulator(connectionURL); err != nil {
 		showErrAndExitTest(err)
 	}
-	createTablesStmt, err := ioutil.ReadFile(schemaFile)
+
+	uri, err := url.Parse(connectionURL)
 	if err != nil {
 		showErrAndExitTest(err)
 	}
-	if _, err = populator.DB.Exec(string(createTablesStmt)); err != nil {
+
+	schemaFile := fmt.Sprintf(schemaFileTemplate, uri.Scheme)
+	createTablesStmt, err := ioutil.ReadFile(schemaFile)
+	if err != nil {
 		showErrAndExitTest(err)
 	}
+
+	statements := strings.Split(string(createTablesStmt), ";")
+	for _, statement := range statements {
+		if strings.TrimSpace(statement) == "" {
+			continue
+		}
+		if _, err = populator.DB.Exec(statement); err != nil {
+			showErrAndExitTest(err)
+		}
+	}
+
 	os.Exit(m.Run())
 }