108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
package scanner
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/viktorstrate/photoview/api/graphql/models"
|
|
)
|
|
|
|
func TestScannerQueue_AddJob(t *testing.T) {
|
|
|
|
scannerJobs := []ScannerJob{
|
|
{scope: JOB_SCAN_ALBUM, model: models.Album{AlbumID: 100, OwnerID: 123}},
|
|
{scope: JOB_SCAN_USER, model: models.User{UserID: 20}},
|
|
}
|
|
|
|
mockScannerQueue := ScannerQueue{
|
|
idle_chan: make(chan bool, 1),
|
|
in_progress: make([]ScannerJob, 0),
|
|
up_next: scannerJobs,
|
|
db: nil,
|
|
}
|
|
|
|
t.Run("add new job to scanner queue", func(t *testing.T) {
|
|
newJob := ScannerJob{
|
|
scope: JOB_SCAN_USER,
|
|
model: models.User{UserID: 253},
|
|
}
|
|
|
|
startingJobs := len(mockScannerQueue.up_next)
|
|
|
|
err := mockScannerQueue.AddJob(&newJob)
|
|
if err != nil {
|
|
t.Errorf(".AddJob() returned an unexpected error: %s", err)
|
|
}
|
|
|
|
if len(mockScannerQueue.up_next) != startingJobs+1 {
|
|
t.Errorf("Expected scanner queue length to be %d but got %d", startingJobs+1, len(mockScannerQueue.up_next))
|
|
} else if mockScannerQueue.up_next[len(mockScannerQueue.up_next)-1] != newJob {
|
|
t.Errorf("Expected scanner queue to contain the job that was added: %+v", newJob)
|
|
}
|
|
|
|
})
|
|
|
|
t.Run("add existing job to scanner queue", func(t *testing.T) {
|
|
startingJobs := len(mockScannerQueue.up_next)
|
|
|
|
err := mockScannerQueue.AddJob(&ScannerJob{
|
|
scope: JOB_SCAN_USER,
|
|
model: models.User{UserID: 20},
|
|
})
|
|
if err != nil {
|
|
t.Errorf(".AddJob() returned an unexpected error: %s", err)
|
|
}
|
|
|
|
if len(mockScannerQueue.up_next) != startingJobs {
|
|
t.Errorf("Expected scanner queue length not to change: start length %d, new length %d", startingJobs, len(mockScannerQueue.up_next))
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
func TestScannerQueue_JobOnQueue(t *testing.T) {
|
|
|
|
scannerJobs := []ScannerJob{
|
|
{scope: JOB_SCAN_ALBUM, model: models.Album{AlbumID: 100, OwnerID: 123}},
|
|
{scope: JOB_SCAN_USER, model: models.User{UserID: 20}},
|
|
}
|
|
|
|
mockScannerQueue := ScannerQueue{
|
|
idle_chan: make(chan bool, 1),
|
|
in_progress: make([]ScannerJob, 0),
|
|
up_next: scannerJobs,
|
|
db: nil,
|
|
}
|
|
|
|
onQueueTests := []struct {
|
|
string
|
|
bool
|
|
ScannerJob
|
|
}{
|
|
{"user that is already on the queue", true, ScannerJob{
|
|
scope: JOB_SCAN_USER,
|
|
model: models.User{UserID: 20},
|
|
}},
|
|
{"album which owner is already on the queue", true, ScannerJob{
|
|
scope: JOB_SCAN_ALBUM,
|
|
model: models.Album{AlbumID: 40, OwnerID: 20},
|
|
}},
|
|
{"album that is not on the queue", false, ScannerJob{
|
|
scope: JOB_SCAN_ALBUM,
|
|
model: models.Album{AlbumID: 321, OwnerID: 11},
|
|
}},
|
|
}
|
|
|
|
for _, test := range onQueueTests {
|
|
t.Run(test.string, func(t *testing.T) {
|
|
onQueue, err := mockScannerQueue.jobOnQueue(&test.ScannerJob)
|
|
if err != nil {
|
|
t.Error("Expected jobOnQueue not to return an error")
|
|
} else if onQueue != test.bool {
|
|
t.Fail()
|
|
}
|
|
})
|
|
}
|
|
|
|
}
|