36 lines
1.1 KiB
Go
36 lines
1.1 KiB
Go
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)
|
|
}
|
|
}
|