package flagutil

import (
	"testing"

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

var (
	ntp         = "ntp"
	kubevip     = "kube-vip"
	dns         = "dns"
	invalidType = "invalid-service-type"
)

func TestValidServiceType_NTP(t *testing.T) {
	assert.True(t, IsValidServiceType(ntp))
}

func TestValidServiceType_KubeVip(t *testing.T) {
	assert.True(t, IsValidServiceType(kubevip))
}

func TestValidServiceType_DNS(t *testing.T) {
	assert.True(t, IsValidServiceType(dns))
}

func TestValidServiceType_invalid(t *testing.T) {
	assert.False(t, IsValidServiceType(invalidType))
}