cocos/pkg/manager/state_test.go

// Copyright (c) Ultraviolet
// SPDX-License-Identifier: Apache-2.0
package manager

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestManagerState_String(t *testing.T) {
	tests := []struct {
		state    ManagerState
		expected string
	}{
		{VmProvision, "VmProvision"},
		{StopComputationRun, "StopComputationRun"},
		{VmRunning, "VmRunning"},
		{ManagerState(3), "ManagerState(3)"},
		{ManagerState(100), "ManagerState(100)"},
	}

	for _, tt := range tests {
		t.Run(tt.expected, func(t *testing.T) {
			assert.Equal(t, tt.state.String(), tt.expected)
		})
	}
}

func TestManagerStatus_String(t *testing.T) {
	tests := []struct {
		status   ManagerStatus
		expected string
	}{
		{Starting, "Starting"},
		{Stopped, "Stopped"},
		{Warning, "Warning"},
		{Disconnected, "Disconnected"},
		{ManagerStatus(5), "ManagerStatus(5)"},
		{ManagerStatus(100), "ManagerStatus(100)"},
	}

	for _, tt := range tests {
		t.Run(tt.expected, func(t *testing.T) {
			assert.Equal(t, tt.status.String(), tt.expected)
		})
	}
}