mirror of
https://github.com/photoprism/photoprism.git
synced 2025-09-26 21:01:58 +08:00
43 lines
791 B
Go
43 lines
791 B
Go
package query
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/photoprism/photoprism/internal/entity"
|
|
"github.com/photoprism/photoprism/pkg/fs"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
log = logrus.StandardLogger()
|
|
log.SetLevel(logrus.TraceLevel)
|
|
|
|
db := entity.InitTestDb(
|
|
os.Getenv("PHOTOPRISM_TEST_DRIVER"),
|
|
os.Getenv("PHOTOPRISM_TEST_DSN"))
|
|
|
|
defer db.Close()
|
|
|
|
code := m.Run()
|
|
|
|
// Remove temporary SQLite files after running the tests.
|
|
fs.PurgeTestDbFiles(".", false)
|
|
|
|
os.Exit(code)
|
|
}
|
|
|
|
func TestDbDialect(t *testing.T) {
|
|
t.Run("SQLite", func(t *testing.T) {
|
|
assert.Equal(t, "sqlite3", DbDialect())
|
|
})
|
|
}
|
|
|
|
func TestBatchSize(t *testing.T) {
|
|
t.Run("SQLite", func(t *testing.T) {
|
|
assert.Equal(t, 333, BatchSize())
|
|
})
|
|
}
|