# HG changeset patch
# User Christophe de Vienne <christophe@cdevienne.info>
# Date 1632865130 -7200
#      Tue Sep 28 23:38:50 2021 +0200
# Node ID 5746457e68bcec77f71b69f5eba624437b2d01ff
# Parent  7908dfbcac1f1960c82db6b7c8af2c72d2acab66
database test functions now also accept benchmarks

diff --git a/database/test.go b/database/test.go
--- a/database/test.go
+++ b/database/test.go
@@ -16,7 +16,7 @@
 
 const dbLockID = 15104
 
-func clearDB(t *testing.T, c *sqlx.Conn, dsn string) {
+func clearDB(t testing.TB, c *sqlx.Conn, dsn string) {
 	// Drop all tables, and reinit
 	if _, err := c.ExecContext(context.Background(), "DROP SCHEMA IF EXISTS public CASCADE"); err != nil {
 		t.Fatal(err)
@@ -72,7 +72,7 @@
 
 // GetTestDB creates a db and returns it. It must be closed within the test.
 // If it fails, t.Fatal() is called
-func GetTestDB(t *testing.T, sourceDriver source.Driver) *TestDB {
+func GetTestDB(t testing.TB, sourceDriver source.Driver) *TestDB {
 	var success bool
 
 	dsn := os.Getenv("TEST_DB_DSN")