Overhaul database package

go/database/testutil/testutil.go

Index: go/database/testutil/testutil.go
diff --git a/go/database/testutil/testutil.go b/go/database/testutil/testutil.go
index cebea84683c5363dc4297580d7141aad933f23af..7c96a36c60879ea16b729a0bdae7957b581e6570 100644
--- a/go/database/testutil/testutil.go
+++ b/go/database/testutil/testutil.go
@@ -1,8 +1,6 @@
 package testutil
 
 import (
-	"fmt"
-	"os"
 	"strings"
 	"testing"
 )
@@ -20,37 +18,17 @@ const (
 	// parameters are: username, password, database.
 	MYSQL_DB_OPEN = "%s:%s@tcp(localhost:3306)/%s?parseTime=true"
 
-	// File path to the local SQLite testing databse.
-	SQLITE_DB_PATH = "./testing.db"
-
 	// Name of the MySQL lock
 	SQL_LOCK = "mysql_testlock"
 )
 
-// Creates an SQLite test database and runs migration tests against it using the
-// given migration steps.
-func SQLiteVersioningTests(t *testing.T, migrationSteps []database.MigrationStep) {
-	// Initialize without argument to test against SQLite3
-	conf := &database.DatabaseConfig{
-		SQLiteFilePath: SQLITE_DB_PATH,
-		MigrationSteps: migrationSteps,
-	}
-
-	vdb := database.NewVersionedDB(conf)
-	assert.False(t, vdb.IsMySQL)
-	testDBVersioning(t, vdb)
-}
-
 // Creates an MySQL test database and runs migration tests against it using the
 // given migration steps. See Get for required credentials.
 // The test assumes that the database is empty and that the readwrite user is
 // not allowed to create/drop/alter tables.
 func MySQLVersioningTests(t *testing.T, dbName string, migrationSteps []database.MigrationStep) {
 	// OpenDB as root user and remove all tables.
-	rootConf := &database.DatabaseConfig{
-		MySQLString:    GetTestMySQLConnStr(t, "root", dbName),
-		MigrationSteps: migrationSteps,
-	}
+	rootConf := database.LocalTestRootDatabaseConfig(migrationSteps)
 	lockVdb := GetMySQlLock(t, rootConf)
 	defer func() {
 		ReleaseMySQLLock(t, lockVdb)
@@ -58,15 +36,11 @@ func MySQLVersioningTests(t *testing.T, dbName string, migrationSteps []database
 	}()
 
 	rootVdb := database.NewVersionedDB(rootConf)
-	assert.True(t, rootVdb.IsMySQL)
 	ClearMySQLTables(t, rootVdb)
 	assert.Nil(t, rootVdb.Close())
 
 	// Configuration for the readwrite user without DDL privileges.
-	readWriteConf := &database.DatabaseConfig{
-		MySQLString:    GetTestMySQLConnStr(t, "readwrite", dbName),
-		MigrationSteps: migrationSteps,
-	}
+	readWriteConf := database.LocalTestDatabaseConfig(migrationSteps)
 
 	// Open DB as readwrite user and make sure it fails because of a missing
 	// version table.
@@ -84,28 +58,6 @@ func MySQLVersioningTests(t *testing.T, dbName string, migrationSteps []database
 	})
 }
 
-// Returns a connection string to the local MySQL server and the given database.
-// The test will be skipped if these environement variables are not set:
-//      MYSQL_TESTING_RWPW   (password of readwrite user)
-//      MYSQL_TESTING_ROOTPW (password of the db root user)
-func GetTestMySQLConnStr(t *testing.T, user string, dbName string) string {
-	rwUserPw, rootPw := os.Getenv("MYSQL_TESTING_RWPW"), os.Getenv("MYSQL_TESTING_ROOTPW")
-	if testing.Short() {
-		t.Skip("Skipping test against MySQL in short mode.")
-	}
-
-	// Skip this test unless there are environment variables with the rwuser and
-	// root password for the local MySQL instance.
-	if (rwUserPw == "") || (rootPw == "") {
-		t.Skip("Skipping test against MySQL. Set 'MYSQL_TESTING_ROOTPW' and 'MYSQL_TESTING_RWPW' to enable tests.")
-	}
-	pw := rwUserPw
-	if user == "root" {
-		pw = rootPw
-	}
-	return fmt.Sprintf(MYSQL_DB_OPEN, user, pw, dbName)
-}
-
 // Get a lock from MySQL to serialize DB tests.
 func GetMySQlLock(t *testing.T, conf *database.DatabaseConfig) *database.VersionedDB {
 	vdb := database.NewVersionedDB(conf)
@@ -141,6 +93,44 @@ func ClearMySQLTables(t *testing.T, vdb *database.VersionedDB) {
 	}
 }
 
+type MySQLTestDatabase struct {
+	rootVdb *database.VersionedDB
+	t       *testing.T
+}
+
+// SetupMySQLTestDatabase prepares the test database for use. Returns a
+// VersionedDB which must be closed after use.
+// Example usage:
+//
+// db := SetupMySQLTestDatabase(t, migrationSteps)
+// defer db.Close()
+// ... Tests here ...
+func SetupMySQLTestDatabase(t *testing.T, migrationSteps []database.MigrationStep) *MySQLTestDatabase {
+	conf := database.LocalTestRootDatabaseConfig(migrationSteps)
+	lockVdb := GetMySQlLock(t, conf)
+	rootVdb := database.NewVersionedDB(conf)
+	ClearMySQLTables(t, rootVdb)
+	if err := rootVdb.Close(); err != nil {
+		t.Fatal(err)
+	}
+	rootVdb = database.NewVersionedDB(conf)
+	if err := rootVdb.Migrate(rootVdb.MaxDBVersion()); err != nil {
+		t.Fatal(err)
+	}
+	if err := rootVdb.Close(); err != nil {
+		t.Fatal(err)
+	}
+	return &MySQLTestDatabase{lockVdb, t}
+}
+
+func (d *MySQLTestDatabase) Close() {
+	if err := d.rootVdb.Migrate(0); err != nil {
+		d.t.Fatal(err)
+	}
+	ReleaseMySQLLock(d.t, d.rootVdb)
+	d.rootVdb.Close()
+}
+
 // Test wether the migration steps execute correctly.
 func testDBVersioning(t *testing.T, vdb *database.VersionedDB) {
 	// get the DB version