chore: add state for task

This commit is contained in:
Noah Hsu
2022-06-23 21:09:54 +08:00
parent aedcae840d
commit 6c61f1d261
4 changed files with 37 additions and 30 deletions

View File

@ -61,10 +61,10 @@ func TestDown(t *testing.T) {
for {
tsk := tasks[0]
t.Logf("task: %+v", tsk)
if tsk.Status == task.FINISHED {
if tsk.GetState() == task.FINISHED {
break
}
if tsk.Status == task.ERRORED {
if tsk.GetState() == task.ERRORED {
t.Fatalf("failed to download: %+v", tsk)
}
time.Sleep(time.Second)
@ -75,10 +75,10 @@ func TestDown(t *testing.T) {
}
tsk := transferTaskManager.GetAll()[0]
t.Logf("task: %+v", tsk)
if tsk.Status == task.FINISHED {
if tsk.GetState() == task.FINISHED {
break
}
if tsk.Status == task.ERRORED {
if tsk.GetState() == task.ERRORED {
t.Fatalf("failed to download: %+v", tsk)
}
time.Sleep(time.Second)