init monlet repo with stage 0-2 (docs, contract, agent MVP)

agent/internal/scheduler/scheduler_test.go

@@ -0,0 +1,52 @@
+package scheduler
+
+import (
+	"context"
+	"sync/atomic"
+	"testing"
+	"time"
+
+	"github.com/monlet/agent/internal/config"
+)
+
+func mkCheck(id, cmd string, interval, timeout time.Duration) config.CheckConfig {
+	return config.CheckConfig{
+		ID:                id,
+		Command:           []string{"sh", "-c", cmd},
+		Interval:          config.Duration{Duration: interval},
+		Timeout:           config.Duration{Duration: timeout},
+		NotificationOwner: "server",
+	}
+}
+
+func TestEmitsEvents(t *testing.T) {
+	ctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond)
+	defer cancel()
+	out := make(chan Event, 10)
+	go Run(ctx, "a1", []config.CheckConfig{mkCheck("c1", "echo hi", 200*time.Millisecond, 100*time.Millisecond)}, out, Hooks{})
+	select {
+	case ev := <-out:
+		if ev.CheckID != "c1" || ev.Status != "ok" {
+			t.Fatalf("unexpected event: %+v", ev)
+		}
+		if ev.IncidentKey != "a1:c1" {
+			t.Fatalf("incident_key: %q", ev.IncidentKey)
+		}
+	case <-ctx.Done():
+		t.Fatal("no event")
+	}
+}
+
+func TestNoOverlapSkips(t *testing.T) {
+	ctx, cancel := context.WithTimeout(context.Background(), 600*time.Millisecond)
+	defer cancel()
+	out := make(chan Event, 10)
+	var skipped int32
+	hooks := Hooks{OnSkipped: func(string) { atomic.AddInt32(&skipped, 1) }}
+	c := mkCheck("slow", "sleep 1", 100*time.Millisecond, 900*time.Millisecond)
+	go Run(ctx, "a1", []config.CheckConfig{c}, out, hooks)
+	<-ctx.Done()
+	if atomic.LoadInt32(&skipped) == 0 {
+		t.Fatal("expected at least one skipped tick")
+	}
+}