update

tests/test_helpers.go

@@ -0,0 +1,60 @@
+package tests
+
+import (
+	"context"
+	"os"
+	"path/filepath"
+
+	"github.com/navidrome/navidrome/db"
+	"github.com/navidrome/navidrome/log"
+	"github.com/navidrome/navidrome/model/id"
+	"github.com/sirupsen/logrus"
+	"github.com/sirupsen/logrus/hooks/test"
+)
+
+type testingT interface {
+	TempDir() string
+}
+
+func TempFileName(t testingT, prefix, suffix string) string {
+	return filepath.Join(t.TempDir(), prefix+id.NewRandom()+suffix)
+}
+
+func TempFile(t testingT, prefix, suffix string) (*os.File, string, error) {
+	name := TempFileName(t, prefix, suffix)
+	f, err := os.Create(name)
+	return f, name, err
+}
+
+// ClearDB deletes all tables and data from the database
+// https://stackoverflow.com/questions/525512/drop-all-tables-command
+func ClearDB() error {
+	_, err := db.Db().ExecContext(context.Background(), `
+			PRAGMA writable_schema = 1;
+			DELETE FROM sqlite_master;
+			PRAGMA writable_schema = 0;
+			VACUUM;
+			PRAGMA integrity_check;
+		`)
+	return err
+}
+
+// LogHook sets up a logrus test hook and configures the default logger to use it.
+// It returns the hook and a cleanup function to restore the default logger.
+// Example usage:
+//
+//	hook, cleanup := LogHook()
+//	defer cleanup()
+//	// ... perform logging operations ...
+//	Expect(hook.LastEntry()).ToNot(BeNil())
+//	Expect(hook.LastEntry().Level).To(Equal(logrus.WarnLevel))
+//	Expect(hook.LastEntry().Message).To(Equal("log message"))
+func LogHook() (*test.Hook, func()) {
+	l, hook := test.NewNullLogger()
+	log.SetLevel(log.LevelWarn)
+	log.SetDefaultLogger(l)
+	return hook, func() {
+		// Restore default logger after test
+		log.SetDefaultLogger(logrus.New())
+	}
+}