69 lines
1.6 KiB
Go
69 lines
1.6 KiB
Go
package migrations
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestJob_StatusReport(t *testing.T) {
|
|
jobWithStatus := func(status *MigratoryStatus) *Job {
|
|
return &Job{
|
|
migratories: map[string]*FetchMigratory{
|
|
"test/test": {
|
|
Migratory: Migratory{
|
|
Status: status,
|
|
},
|
|
},
|
|
},
|
|
Repositories: []string{
|
|
"test/test",
|
|
},
|
|
}
|
|
}
|
|
// Pending
|
|
pendingJob := &Job{
|
|
Repositories: []string{
|
|
"test/test",
|
|
},
|
|
}
|
|
report := pendingJob.StatusReport()
|
|
assert.Len(t, report.Pending, 1)
|
|
assert.Equal(t, report.Pending[0], "test/test")
|
|
assert.Len(t, report.Failed, 0)
|
|
assert.Len(t, report.Running, 0)
|
|
assert.Len(t, report.Finished, 0)
|
|
|
|
// Finished
|
|
report = jobWithStatus(&MigratoryStatus{
|
|
Stage: Finished,
|
|
}).StatusReport()
|
|
assert.Len(t, report.Pending, 0)
|
|
assert.Len(t, report.Failed, 0)
|
|
assert.Len(t, report.Running, 0)
|
|
assert.Len(t, report.Finished, 1)
|
|
assert.Equal(t, Finished, report.Finished["test/test"].Stage)
|
|
|
|
// Failed
|
|
report = jobWithStatus(&MigratoryStatus{
|
|
Stage: Failed,
|
|
FatalError: fmt.Errorf("test"),
|
|
}).StatusReport()
|
|
assert.Len(t, report.Failed, 1)
|
|
assert.Equal(t, "test", report.Failed["test/test"])
|
|
assert.Len(t, report.Pending, 0)
|
|
assert.Len(t, report.Running, 0)
|
|
assert.Len(t, report.Finished, 0)
|
|
|
|
// Running
|
|
report = jobWithStatus(&MigratoryStatus{
|
|
Stage: Migrating,
|
|
}).StatusReport()
|
|
assert.Len(t, report.Running, 1)
|
|
assert.Equal(t, Migrating, report.Running["test/test"].Stage)
|
|
assert.Len(t, report.Pending, 0)
|
|
assert.Len(t, report.Failed, 0)
|
|
assert.Len(t, report.Finished, 0)
|
|
}
|