photoprism/internal/entity/entity_test.go

32 lines
564 B
Go

package entity
import (
"os"
"testing"
"github.com/stretchr/testify/assert"
"github.com/sirupsen/logrus"
)
func TestMain(m *testing.M) {
log = logrus.StandardLogger()
log.SetLevel(logrus.TraceLevel)
if err := os.Remove(".test.db"); err == nil {
log.Debugln("removed .test.db")
}
db := InitTestDb(os.Getenv("PHOTOPRISM_TEST_DRIVER"), os.Getenv("PHOTOPRISM_TEST_DSN"))
defer db.Close()
code := m.Run()
os.Exit(code)
}
func TestTypeString(t *testing.T) {
assert.Equal(t, "unknown", TypeString(""))
assert.Equal(t, "foo", TypeString("foo"))
}