examples: Add missing Close calls

sqlc-dev · Feb 7, 2025 · 6608af4 · 6608af4
1 parent 9cf5dbc
commit 6608af4
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 1 deletion.
diff --git a/examples/authors/sqlite/db_test.go b/examples/authors/sqlite/db_test.go
@@ -12,6 +12,7 @@ import (
 
 func TestAuthors(t *testing.T) {
 	sdb, cleanup := sqltest.SQLite(t, []string{"schema.sql"})
+	defer sdb.Close()
 	defer cleanup()
 
 	ctx := context.Background()

diff --git a/examples/booktest/sqlite/db_test.go b/examples/booktest/sqlite/db_test.go
@@ -18,6 +18,7 @@ const (
 
 func TestBooks(t *testing.T) {
 	db, cleanup := sqltest.SQLite(t, []string{"schema.sql"})
+	defer sdb.Close()
 	defer cleanup()
 
 	ctx := context.Background()

diff --git a/examples/ondeck/sqlite/db_test.go b/examples/ondeck/sqlite/db_test.go
@@ -149,6 +149,7 @@ func TestPrepared(t *testing.T) {
 	t.Parallel()
 
 	sdb, cleanup := sqltest.SQLite(t, []string{"schema"})
+	defer sdb.Close()
 	defer cleanup()
 
 	q, err := Prepare(context.Background(), sdb)
@@ -163,6 +164,7 @@ func TestQueries(t *testing.T) {
 	t.Parallel()
 
 	sdb, cleanup := sqltest.SQLite(t, []string{"schema"})
+	defer sdb.Close()
 	defer cleanup()
 
 	runOnDeckQueries(t, New(sdb))

diff --git a/internal/sqltest/sqlite.go b/internal/sqltest/sqlite.go
@@ -16,7 +16,11 @@ func SQLite(t *testing.T, migrations []string) (*sql.DB, func()) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	return CreateSQLiteDatabase(t, source.Name(), migrations)
+	db, cleanup := CreateSQLiteDatabase(t, source.Name(), migrations)
+	return db, func() {
+		source.Close()
+		cleanup()
+	}
 }
 
 func CreateSQLiteDatabase(t *testing.T, path string, migrations []string) (*sql.DB, func()) {