From 3806c1cc233b55d354d5f01467c6dd8f84d2e009 Mon Sep 17 00:00:00 2001
From: Nuno Cruces <ncruces@users.noreply.github.com>
Date: Fri, 30 Aug 2024 01:27:22 +0100
Subject: [PATCH] Test tweaks.

---
 ext/blobio/blob_test.go    | 3 ++-
 ext/bloom/bloom_test.go    | 3 ++-
 ext/csv/csv_test.go        | 3 ++-
 ext/pivot/pivot_test.go    | 3 ++-
 ext/statement/stmt_test.go | 3 ++-
 ext/stats/stats_test.go    | 3 ++-
 6 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/ext/blobio/blob_test.go b/ext/blobio/blob_test.go
index 52cdd68..b46daf0 100644
--- a/ext/blobio/blob_test.go
+++ b/ext/blobio/blob_test.go
@@ -58,9 +58,10 @@ func Example() {
 	// Hello BLOB!
 }
 
-func init() {
+func TestMain(m *testing.M) {
 	sqlite3.AutoExtension(blobio.Register)
 	sqlite3.AutoExtension(array.Register)
+	m.Run()
 }
 
 func Test_readblob(t *testing.T) {
diff --git a/ext/bloom/bloom_test.go b/ext/bloom/bloom_test.go
index 0a02942..fb10131 100644
--- a/ext/bloom/bloom_test.go
+++ b/ext/bloom/bloom_test.go
@@ -12,8 +12,9 @@ import (
 	_ "github.com/ncruces/go-sqlite3/internal/testcfg"
 )
 
-func init() {
+func TestMain(m *testing.M) {
 	sqlite3.AutoExtension(bloom.Register)
+	m.Run()
 }
 
 func TestRegister(t *testing.T) {
diff --git a/ext/csv/csv_test.go b/ext/csv/csv_test.go
index 386f855..3732772 100644
--- a/ext/csv/csv_test.go
+++ b/ext/csv/csv_test.go
@@ -54,8 +54,9 @@ func Example() {
 	// On Twosday, 1€ = $1.1342
 }
 
-func init() {
+func TestMain(m *testing.M) {
 	sqlite3.AutoExtension(csv.Register)
+	m.Run()
 }
 
 func TestRegister(t *testing.T) {
diff --git a/ext/pivot/pivot_test.go b/ext/pivot/pivot_test.go
index 41b8389..16c1b6d 100644
--- a/ext/pivot/pivot_test.go
+++ b/ext/pivot/pivot_test.go
@@ -83,8 +83,9 @@ func Example() {
 	// gamma   80      75      78      80
 }
 
-func init() {
+func TestMain(m *testing.M) {
 	sqlite3.AutoExtension(pivot.Register)
+	m.Run()
 }
 
 func TestRegister(t *testing.T) {
diff --git a/ext/statement/stmt_test.go b/ext/statement/stmt_test.go
index 4b407e2..2475fff 100644
--- a/ext/statement/stmt_test.go
+++ b/ext/statement/stmt_test.go
@@ -48,8 +48,9 @@ func Example() {
 	// Twosday was 2022-2-22
 }
 
-func init() {
+func TestMain(m *testing.M) {
 	sqlite3.AutoExtension(statement.Register)
+	m.Run()
 }
 
 func TestRegister(t *testing.T) {
diff --git a/ext/stats/stats_test.go b/ext/stats/stats_test.go
index c3d6b5f..5a5426c 100644
--- a/ext/stats/stats_test.go
+++ b/ext/stats/stats_test.go
@@ -10,8 +10,9 @@ import (
 	_ "github.com/ncruces/go-sqlite3/internal/testcfg"
 )
 
-func init() {
+func TestMain(m *testing.M) {
 	sqlite3.AutoExtension(stats.Register)
+	m.Run()
 }
 
 func TestRegister_variance(t *testing.T) {