43 lines
985 B
Go
43 lines
985 B
Go
|
package scheduler
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestPublishQueuedTaskTimeoutFromRunDefaultsToThreeDays(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
|
||
|
|
got := publishQueuedTaskTimeoutFromRun(JobRunContext{})
|
||
|
|
if got != 3*24*time.Hour {
|
||
|
|
t.Fatalf("publishQueuedTaskTimeoutFromRun() = %s, want 72h", got)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPublishQueuedTaskTimeoutFromRunReadsDurationConfig(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
|
||
|
|
got := publishQueuedTaskTimeoutFromRun(JobRunContext{
|
||
|
|
Config: map[string]any{
|
||
|
|
"queued_task_timeout": "72h",
|
||
|
|
},
|
||
|
|
})
|
||
|
|
if got != 72*time.Hour {
|
||
|
|
t.Fatalf("publishQueuedTaskTimeoutFromRun(duration) = %s, want 72h", got)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPublishQueuedTaskTimeoutFromRunReadsSecondsConfig(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
|
||
|
|
got := publishQueuedTaskTimeoutFromRun(JobRunContext{
|
||
|
|
Config: map[string]any{
|
||
|
|
"queued_task_timeout": "72h",
|
||
|
|
"queued_task_timeout_seconds": 3600,
|
||
|
|
},
|
||
|
|
})
|
||
|
|
if got != time.Hour {
|
||
|
|
t.Fatalf("publishQueuedTaskTimeoutFromRun(seconds) = %s, want 1h", got)
|
||
|
|
}
|
||
|
|
}
|