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)
	}
}