82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
package scanner_test
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/photoview/photoview/api/graphql/models"
|
|
"github.com/photoview/photoview/api/scanner/face_detection"
|
|
"github.com/photoview/photoview/api/test_utils"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
os.Exit(test_utils.IntegrationTestRun(m))
|
|
}
|
|
|
|
func TestFullScan(t *testing.T) {
|
|
test_utils.FilesystemTest(t)
|
|
db := test_utils.DatabaseTest(t)
|
|
|
|
pass := "1234"
|
|
user, err := models.RegisterUser(db, "test_user", &pass, true)
|
|
if !assert.NoError(t, err) {
|
|
return
|
|
}
|
|
|
|
root_album := models.Album{
|
|
Title: "root album",
|
|
Path: "./test_data",
|
|
}
|
|
|
|
if !assert.NoError(t, db.Save(&root_album).Error) {
|
|
return
|
|
}
|
|
|
|
err = db.Model(user).Association("Albums").Append(&root_album)
|
|
if !assert.NoError(t, err) {
|
|
return
|
|
}
|
|
|
|
if !assert.NoError(t, face_detection.InitializeFaceDetector(db)) {
|
|
return
|
|
}
|
|
|
|
test_utils.RunScannerOnUser(t, db, user)
|
|
|
|
var all_media []*models.Media
|
|
if !assert.NoError(t, db.Find(&all_media).Error) {
|
|
return
|
|
}
|
|
|
|
assert.Equal(t, 9, len(all_media))
|
|
|
|
var all_media_url []*models.MediaURL
|
|
if !assert.NoError(t, db.Find(&all_media_url).Error) {
|
|
return
|
|
}
|
|
|
|
assert.Equal(t, 18, len(all_media_url))
|
|
|
|
// Verify that faces was recognized
|
|
assert.Eventually(t, func() bool {
|
|
var all_face_groups []*models.FaceGroup
|
|
if !assert.NoError(t, db.Find(&all_face_groups).Error) {
|
|
return false
|
|
}
|
|
|
|
return len(all_face_groups) == 3
|
|
}, time.Second*5, time.Millisecond*500)
|
|
|
|
assert.Eventually(t, func() bool {
|
|
var all_image_faces []*models.ImageFace
|
|
if !assert.NoError(t, db.Find(&all_image_faces).Error) {
|
|
return false
|
|
}
|
|
|
|
return len(all_image_faces) == 6
|
|
}, time.Second*5, time.Millisecond*500)
|
|
|
|
}
|