Initial project snapshot

agents/rap-node-agent/internal/supervisor/supervisor_test.go

@@ -0,0 +1,35 @@
+package supervisor
+
+import (
+	"context"
+	"testing"
+
+	"github.com/example/remote-access-platform/agents/rap-node-agent/internal/client"
+)
+
+func TestStubSupervisorReportsDegradedForEnabledWorkload(t *testing.T) {
+	statuses, err := (StubSupervisor{Version: "test"}).Apply(context.Background(), []client.DesiredWorkload{
+		{ServiceType: "rdp-worker", DesiredState: "enabled", RuntimeMode: "container"},
+	})
+	if err != nil {
+		t.Fatalf("apply desired workload: %v", err)
+	}
+	if len(statuses) != 1 {
+		t.Fatalf("statuses length = %d", len(statuses))
+	}
+	if statuses[0].ReportedState != "degraded" {
+		t.Fatalf("ReportedState = %q", statuses[0].ReportedState)
+	}
+}
+
+func TestStubSupervisorReportsStoppedForDisabledWorkload(t *testing.T) {
+	statuses, err := (StubSupervisor{Version: "test"}).Apply(context.Background(), []client.DesiredWorkload{
+		{ServiceType: "relay-node", DesiredState: "disabled", RuntimeMode: "container"},
+	})
+	if err != nil {
+		t.Fatalf("apply desired workload: %v", err)
+	}
+	if statuses[0].ReportedState != "stopped" {
+		t.Fatalf("ReportedState = %q", statuses[0].ReportedState)
+	}
+}