From 63fdc064387b33d6be363439c7d869cc524a0957 Mon Sep 17 00:00:00 2001
From: "Zed A. Shaw" <zed.shaw@gmail.com>
Date: Sun, 29 Jun 2025 13:33:53 -0400
Subject: [PATCH] Use TestMain to setup the database and shut it down after.

---
 tests/base_test.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tests/base_test.go b/tests/base_test.go
index 59b43c3..4dd5828 100644
--- a/tests/base_test.go
+++ b/tests/base_test.go
@@ -15,7 +15,6 @@ func deleteTestUser(username string) {
 
 func TestLogin(t *testing.T) {
   assert := assert.New(t)
-  data.Setup("sqlite3", "./db.sqlite3")
   deleteTestUser("testerzed")
 
   ctx, cancel := Setup(5); defer cancel()
@@ -69,3 +68,9 @@ func TestStreamPage(t *testing.T) {
 
   assert.Equal(title, "Past Streams")
 }
+
+func TestMain(m *testing.M) {
+  data.Setup("sqlite3", "./db.sqlite3")
+  m.Run()
+  data.Shutdown()
+}