test: Adjust driver tests to accept custom migration content

Most DBs can use a set of default queries. But those queries aren't
valid for cassandra. Allow a driver to supply DB preparation queries.

Add test input validation

Make a copy of test inputs rather than mutate it in place (the invalid
sql test did this).

drivers/cassandra/cassandra_test.go

@@ -8,5 +8,20 @@ import (
 )
 
 func Test(t *testing.T) {
- test.RunDriverTests(t, cassandra.NewDriver())
+ queries := test.Queries{
+ CreateFoos: test.MigrationContent{
+ Forward: "CREATE TABLE foos (id int PRIMARY KEY);",
+ Reverse: "DROP TABLE foos;",
+ },
+ CreateBars: test.MigrationContent{
+ Forward: "CREATE TABLE bars (id int PRIMARY KEY);",
+ Reverse: "DROP TABLE bars;",
+ },
+ AlterFoos: test.MigrationContent{
+ Forward: "ALTER TABLE foos ADD a varchar;",
+ Reverse: "ALTER TABLE foos DROP a;",
+ },
+ }
+
+ test.RunDriverTests(t, cassandra.NewDriver(), queries)
 }

drivers/mysql/mysql_test.go

@@ -8,5 +8,5 @@ import (
 )
 
 func Test(t *testing.T) {
- test.RunDriverTests(t, mysql.NewDriver())
+ test.RunDriverTests(t, mysql.NewDriver(), test.DefaultQueries)
 }

drivers/postgres/postgres_test.go

@@ -8,5 +8,5 @@ import (
 )
 
 func Test(t *testing.T) {
- test.RunDriverTests(t, postgres.NewDriver())
+ test.RunDriverTests(t, postgres.NewDriver(), test.DefaultQueries)
 }

drivers/sqlite3/sqlite3_test.go

@@ -8,5 +8,5 @@ import (
 )
 
 func Test(t *testing.T) {
- test.RunDriverTests(t, sqlite3.NewDriver())
+ test.RunDriverTests(t, sqlite3.NewDriver(), test.DefaultQueries)
 }

internal/stub/driver_test.go

@@ -8,5 +8,5 @@ import (
 )
 
 func Test(t *testing.T) {
- test.RunDriverTests(t, &stub.Driver{})
+ test.RunDriverTests(t, &stub.Driver{}, test.DefaultQueries)
 }