mirror of
https://github.com/photoprism/photoprism.git
synced 2025-10-31 20:22:55 +08:00
106 lines
2.8 KiB
Go
106 lines
2.8 KiB
Go
package photoprism
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/stretchr/testify/assert"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
const testDataPath = "testdata"
|
|
const testDataUrl = "https://www.dropbox.com/s/na9p9wwt98l7m5b/import.zip?dl=1"
|
|
const testDataHash = "ed3bdb2fe86ea662bc863b63e219b47b8d9a74024757007f7979887d"
|
|
|
|
var darktableCli = "/usr/bin/darktable-cli"
|
|
var testDataZip = GetExpandedFilename(testDataPath + "/import.zip")
|
|
var originalsPath = GetExpandedFilename(testDataPath + "/originals")
|
|
var thumbnailsPath = GetExpandedFilename(testDataPath + "/thumbnails")
|
|
var importPath = GetExpandedFilename(testDataPath + "/import")
|
|
var exportPath = GetExpandedFilename(testDataPath + "/export")
|
|
var databaseDriver = "mysql"
|
|
var databaseDsn = "photoprism:photoprism@tcp(database:3306)/photoprism?parseTime=true"
|
|
|
|
func (c *Config) RemoveTestData(t *testing.T) {
|
|
os.RemoveAll(c.ImportPath)
|
|
os.RemoveAll(c.ExportPath)
|
|
os.RemoveAll(c.OriginalsPath)
|
|
os.RemoveAll(c.ThumbnailsPath)
|
|
}
|
|
|
|
func (c *Config) DownloadTestData(t *testing.T) {
|
|
if fileExists(testDataZip) {
|
|
hash := fileHash(testDataZip)
|
|
|
|
if hash != testDataHash {
|
|
os.Remove(testDataZip)
|
|
t.Logf("Removed outdated test data zip file (fingerprint %s)\n", hash)
|
|
}
|
|
}
|
|
|
|
if !fileExists(testDataZip) {
|
|
fmt.Printf("Downloading latest test data zip file from %s\n", testDataUrl)
|
|
|
|
if err := downloadFile(testDataZip, testDataUrl); err != nil {
|
|
fmt.Printf("Download failed: %s\n", err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
func (c *Config) UnzipTestData(t *testing.T) {
|
|
if _, err := unzip(testDataZip, testDataPath); err != nil {
|
|
t.Logf("Could not unzip test data: %s\n", err.Error())
|
|
}
|
|
}
|
|
|
|
func (c *Config) InitializeTestData(t *testing.T) {
|
|
t.Log("Initializing test data")
|
|
|
|
c.RemoveTestData(t)
|
|
|
|
c.DownloadTestData(t)
|
|
|
|
c.UnzipTestData(t)
|
|
}
|
|
|
|
func NewTestConfig() *Config {
|
|
return &Config{
|
|
DarktableCli: darktableCli,
|
|
OriginalsPath: originalsPath,
|
|
ThumbnailsPath: thumbnailsPath,
|
|
ImportPath: importPath,
|
|
ExportPath: exportPath,
|
|
DatabaseDriver: databaseDriver,
|
|
DatabaseDsn: databaseDsn,
|
|
}
|
|
}
|
|
|
|
func TestNewConfig(t *testing.T) {
|
|
c := NewConfig()
|
|
|
|
assert.IsType(t, &Config{}, c)
|
|
}
|
|
|
|
func TestConfig_SetValuesFromFile(t *testing.T) {
|
|
c := NewConfig()
|
|
|
|
c.SetValuesFromFile(GetExpandedFilename("config.example.yml"))
|
|
|
|
assert.Equal(t, GetExpandedFilename("photos/originals"), c.OriginalsPath)
|
|
assert.Equal(t, GetExpandedFilename("photos/thumbnails"), c.ThumbnailsPath)
|
|
assert.Equal(t, GetExpandedFilename("photos/import"), c.ImportPath)
|
|
assert.Equal(t, GetExpandedFilename("photos/export"), c.ExportPath)
|
|
assert.Equal(t, databaseDriver, c.DatabaseDriver)
|
|
assert.Equal(t, databaseDsn, c.DatabaseDsn)
|
|
}
|
|
|
|
func TestConfig_ConnectToDatabase(t *testing.T) {
|
|
c := NewTestConfig()
|
|
|
|
c.ConnectToDatabase()
|
|
|
|
db := c.GetDb()
|
|
|
|
assert.IsType(t, &gorm.DB{}, db)
|
|
}
|