TestDB: add flags

database/test.go

@@ -44,6 +44,16 @@
 	ctx      context.Context
 	tb       testing.TB
 	lockConn *sqlx.Conn
+
+	flags map[string]bool
+}
+
+func (db *TestDB) GetFlag(name string) bool {
+	return db.flags[name]
+}
+
+func (db *TestDB) SetFlag(name string, value bool) {
+	db.flags[name] = value
 }
 
 // SetTB changes the current tb and returns a function to restore the original one.
@@ -131,7 +141,7 @@
 
 	success = true
 
-	return &TestDB{db, dsn, ctx, tb, c}
+	return &TestDB{db, dsn, ctx, tb, c, make(map[string]bool)}
 }
 
 // GetTestDB creates a db and returns it. It must be closed within the test.
@@ -200,7 +210,7 @@
 		}
 	}
 
-	testDB := TestDB{db, dsn, ctx, tb, c}
+	testDB := TestDB{db, dsn, ctx, tb, c, make(map[string]bool)}
 	if len(dbtablenames) == 0 {
 		if err := db.Select(&dbtablenames,
 			"SELECT tablename "+