mirror of
https://github.com/stashapp/stash.git
synced 2025-12-06 16:34:02 +01:00
5495d72849
* Restructure data layer part 2 (#2599) * Refactor and separate image model * Refactor image query builder * Handle relationships in image query builder * Remove relationship management methods * Refactor gallery model/query builder * Add scenes to gallery model * Convert scene model * Refactor scene models * Remove unused methods * Add unit tests for gallery * Add image tests * Add scene tests * Convert unnecessary scene value pointers to values * Convert unnecessary pointer values to values * Refactor scene partial * Add scene partial tests * Refactor ImagePartial * Add image partial tests * Refactor gallery partial update * Add partial gallery update tests * Use zero/null package for null values * Add files and scan system * Add sqlite implementation for files/folders * Add unit tests for files/folders * Image refactors * Update image data layer * Refactor gallery model and creation * Refactor scene model * Refactor scenes * Don't set title from filename * Allow galleries to freely add/remove images * Add multiple scene file support to graphql and UI * Add multiple file support for images in graphql/UI * Add multiple file for galleries in graphql/UI * Remove use of some deprecated fields * Remove scene path usage * Remove gallery path usage * Remove path from image * Move funscript to video file * Refactor caption detection * Migrate existing data * Add post commit/rollback hook system * Lint. Comment out import/export tests * Add WithDatabase read only wrapper * Prepend tasks to list * Add 32 pre-migration * Add warnings in release and migration notes
150 lines
2.4 KiB
Go
150 lines
2.4 KiB
Go
package job
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func createProgress(m *Manager, j *Job) Progress {
|
|
return Progress{
|
|
updater: &updater{
|
|
m: m,
|
|
job: j,
|
|
},
|
|
total: 100,
|
|
defined: true,
|
|
processed: 10,
|
|
percent: 10,
|
|
}
|
|
}
|
|
|
|
func TestProgressIndefinite(t *testing.T) {
|
|
m := NewManager()
|
|
j := &Job{}
|
|
|
|
p := createProgress(m, j)
|
|
|
|
p.Indefinite()
|
|
|
|
assert := assert.New(t)
|
|
|
|
// ensure job progress was updated
|
|
assert.Equal(ProgressIndefinite, j.Progress)
|
|
}
|
|
|
|
func TestProgressSetTotal(t *testing.T) {
|
|
m := NewManager()
|
|
j := &Job{}
|
|
|
|
p := createProgress(m, j)
|
|
|
|
p.SetTotal(50)
|
|
|
|
assert := assert.New(t)
|
|
|
|
// ensure job progress was updated
|
|
assert.Equal(0.2, j.Progress)
|
|
|
|
p.SetTotal(0)
|
|
assert.Equal(ProgressIndefinite, j.Progress)
|
|
|
|
p.SetTotal(-10)
|
|
assert.Equal(ProgressIndefinite, j.Progress)
|
|
|
|
p.SetTotal(9)
|
|
assert.Equal(float64(1), j.Progress)
|
|
}
|
|
|
|
func TestProgressSetProcessed(t *testing.T) {
|
|
m := NewManager()
|
|
j := &Job{}
|
|
|
|
p := createProgress(m, j)
|
|
|
|
p.SetProcessed(30)
|
|
|
|
assert := assert.New(t)
|
|
|
|
// ensure job progress was updated
|
|
assert.Equal(0.3, j.Progress)
|
|
|
|
p.SetProcessed(-10)
|
|
assert.Equal(float64(0), j.Progress)
|
|
|
|
p.SetProcessed(200)
|
|
assert.Equal(float64(1), j.Progress)
|
|
}
|
|
|
|
func TestProgressSetPercent(t *testing.T) {
|
|
m := NewManager()
|
|
j := &Job{}
|
|
|
|
p := createProgress(m, j)
|
|
|
|
p.SetPercent(0.3)
|
|
|
|
assert := assert.New(t)
|
|
|
|
// ensure job progress was updated
|
|
assert.Equal(0.3, j.Progress)
|
|
|
|
p.SetPercent(-10)
|
|
assert.Equal(float64(0), j.Progress)
|
|
|
|
p.SetPercent(200)
|
|
assert.Equal(float64(1), j.Progress)
|
|
}
|
|
|
|
func TestProgressIncrement(t *testing.T) {
|
|
m := NewManager()
|
|
j := &Job{}
|
|
|
|
p := createProgress(m, j)
|
|
|
|
p.SetProcessed(49)
|
|
p.Increment()
|
|
|
|
assert := assert.New(t)
|
|
|
|
// ensure job progress was updated
|
|
assert.Equal(0.5, j.Progress)
|
|
|
|
p.SetProcessed(100)
|
|
p.Increment()
|
|
assert.Equal(float64(1), j.Progress)
|
|
}
|
|
|
|
func TestExecuteTask(t *testing.T) {
|
|
m := NewManager()
|
|
j := &Job{}
|
|
|
|
p := createProgress(m, j)
|
|
|
|
c := make(chan struct{}, 1)
|
|
const taskDesciption = "taskDescription"
|
|
|
|
go p.ExecuteTask(taskDesciption, func() {
|
|
<-c
|
|
})
|
|
|
|
time.Sleep(sleepTime)
|
|
|
|
assert := assert.New(t)
|
|
|
|
m.mutex.Lock()
|
|
// ensure task is added to the job details
|
|
assert.Equal(taskDesciption, j.Details[0])
|
|
m.mutex.Unlock()
|
|
|
|
// allow task to finish
|
|
close(c)
|
|
|
|
time.Sleep(sleepTime)
|
|
|
|
m.mutex.Lock()
|
|
// ensure task is removed from the job details
|
|
assert.Len(j.Details, 0)
|
|
m.mutex.Unlock()
|
|
}
|