quintodrome/tests/test_helpers.go

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())
	}
}