orchard/pkg/resource/v1/restart_policy_test.go

27 lines
950 B
Go

package v1_test
import (
v1 "github.com/cirruslabs/orchard/pkg/resource/v1"
"github.com/stretchr/testify/assert"
"testing"
)
func TestNewRestartPolicyFromString(t *testing.T) {
_, err := v1.NewRestartPolicyFromString("")
assert.Error(t, err, "empty restart policy should yield an error")
_, err = v1.NewRestartPolicyFromString("non-existent")
assert.Error(t, err, "non-existent restart policy should yield an error")
_, err = v1.NewRestartPolicyFromString("never")
assert.Error(t, err, "improperly capitalized but existent policy should yield an error")
restartPolicy, err := v1.NewRestartPolicyFromString("Never")
assert.NoError(t, err, "Never policy should be parsed correctly")
assert.Equal(t, v1.RestartPolicyNever, restartPolicy)
restartPolicy, err = v1.NewRestartPolicyFromString("OnFailure")
assert.NoError(t, err, "OnFailure policy should be parsed correctly")
assert.Equal(t, v1.RestartPolicyOnFailure, restartPolicy)
}