enum_test.go

package checker

import (
	"testing"

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

type enum struct {
	Int    int
	Uint   uint
	Float  float64
	String string
}

func TestEnumTest(t *testing.T) {
	eChecker := NewChecker()

	intRule := InInt("Int", 1, 2, 3)
	eChecker.Add(intRule, "invalid Int")

	uintRule := InUint("Uint", 4, 5, 6)
	eChecker.Add(uintRule, "invalid Uint")

	floatRule := InFloat("Float", 3.14, 6.28, 9.42, 18.0)
	eChecker.Add(floatRule, "invalid Float")

	strRule := InStr("String", "github", "str")
	eChecker.Add(strRule, "invalid String")

	e := enum{
		Int:    3,
		Uint:   5,
		Float:  3.14,
		String: "str",
	}

	isValid, _, _ := eChecker.Check(e)
	assert.Equal(t, true, isValid)

}