1 files changed, 60 insertions, 0 deletions
diff --git a/db/db_test.go b/db/db_test.go
new file mode 100644
index 0000000..7bc92ca
--- /dev/null
+++ b/db/db_test.go
@@ -0,0 +1,60 @@
+package db
+
+import (
+	"log"
+	"os"
+	"slices"
+	"testing"
+)
+
+func TestMain(m *testing.M) {
+	err := InitializeDB()
+	if err != nil {
+		log.Fatalln("Failed to initialize database")
+	}
+
+	exitCode := m.Run()
+
+	_ = CloseDB()
+	os.Exit(exitCode)
+}
+
+func TestDatabase(t *testing.T) {
+	var (
+		service = "test"
+		siteA   = "a.com"
+		siteB   = "b.com"
+		siteC   = "c.com"
+	)
+	instances := []string{siteA, siteB, siteC}
+	err := SetInstances(service, instances)
+	if err != nil {
+		t.Fatalf("Failed to set instances: %v\n", err)
+	}
+
+	dbInstances, err := GetAllInstances(service)
+	if err != nil {
+		t.Fatalf("Failed to retrieve instances: %v\n", err)
+	}
+
+	for _, instance := range instances {
+		idx := slices.Index(dbInstances, instance)
+		if idx < 0 {
+			t.Fatalf("Failed to find instance in list")
+		}
+	}
+
+	firstInstance, err := GetInstance(service)
+	if err != nil {
+		t.Fatalf("Failed to fetch single instance: %v\n", err)
+	}
+
+	secondInstance, err := GetInstance(service)
+	if err != nil {
+		t.Fatalf("Failed to fetch single instance (second): %v\n", err)
+	} else if firstInstance == secondInstance {
+		t.Fatalf("Same instance was selected twice")
+	}
+
+	_ = CloseDB()
+}