diff options
Diffstat (limited to 'internal/task/task_test.go')
| -rw-r--r-- | internal/task/task_test.go | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/internal/task/task_test.go b/internal/task/task_test.go index 96f5f6f..a8e0a84 100644 --- a/internal/task/task_test.go +++ b/internal/task/task_test.go @@ -21,6 +21,7 @@ func TestValidTransition_AllowedTransitions(t *testing.T) { {"running to cancelled", StateRunning, StateCancelled}, {"running to budget exceeded", StateRunning, StateBudgetExceeded}, {"failed to queued (retry)", StateFailed, StateQueued}, + {"timed out to queued (retry)", StateTimedOut, StateQueued}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { @@ -43,6 +44,7 @@ func TestValidTransition_DisallowedTransitions(t *testing.T) { {"completed to running", StateCompleted, StateRunning}, {"completed to queued", StateCompleted, StateQueued}, {"failed to completed", StateFailed, StateCompleted}, + {"timed out to completed", StateTimedOut, StateCompleted}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { |