49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
|
package triggers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/Silo-Server/silo-server/internal/taskmanager"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestStartupTriggerClearsNextRunAfterFire(t *testing.T) {
|
||
|
|
trigger := NewStartupTrigger(taskmanager.TriggerConfig{Type: taskmanager.TriggerTypeStartup})
|
||
|
|
trigger.delay = time.Millisecond
|
||
|
|
|
||
|
|
trigger.Start(nil)
|
||
|
|
if trigger.NextRunTime().IsZero() {
|
||
|
|
t.Fatal("expected startup trigger to report next run before firing")
|
||
|
|
}
|
||
|
|
|
||
|
|
select {
|
||
|
|
case <-trigger.C():
|
||
|
|
case <-time.After(time.Second):
|
||
|
|
t.Fatal("startup trigger did not fire")
|
||
|
|
}
|
||
|
|
|
||
|
|
if next := trigger.NextRunTime(); !next.IsZero() {
|
||
|
|
t.Fatalf("next run after fire = %s, want zero", next.Format(time.RFC3339Nano))
|
||
|
|
}
|
||
|
|
|
||
|
|
trigger.Start(nil)
|
||
|
|
if next := trigger.NextRunTime(); !next.IsZero() {
|
||
|
|
t.Fatalf("next run after rearm = %s, want zero", next.Format(time.RFC3339Nano))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStartupTriggerClearsNextRunWhenStopped(t *testing.T) {
|
||
|
|
trigger := NewStartupTrigger(taskmanager.TriggerConfig{Type: taskmanager.TriggerTypeStartup})
|
||
|
|
trigger.delay = time.Second
|
||
|
|
|
||
|
|
trigger.Start(nil)
|
||
|
|
if trigger.NextRunTime().IsZero() {
|
||
|
|
t.Fatal("expected startup trigger to report next run before stop")
|
||
|
|
}
|
||
|
|
|
||
|
|
trigger.Stop()
|
||
|
|
if next := trigger.NextRunTime(); !next.IsZero() {
|
||
|
|
t.Fatalf("next run after stop = %s, want zero", next.Format(time.RFC3339Nano))
|
||
|
|
}
|
||
|
|
}
|