enum_test.go

package enum_test

import (
	"testing"

	"github.com/stretchr/testify/require"
	"github.com/yylego/enum"
)

// TestNewEnum tests the creation and Code method of Enum instance
// Checks Code method returns expected value
//
// 验证 Enum 实例的创建和 Code 方法
// 测试 Code 方法返回预期值
func TestNewEnum(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
		failure Status = "failure"
	)
	one := enum.NewEnum(success)
	t.Log(one.Code())
	require.Equal(t, success, one.Code())
	require.Equal(t, unknown, enum.NewEnum(unknown).Code())
	require.Equal(t, failure, enum.NewEnum(failure).Code())
}

// TestNewEnumWithDesc tests Enum creation with description
// Checks both Code and Meta().Desc() return expected values
//
// 验证带描述的 Enum 创建
// 测试 Code 和 Meta().Desc() 返回预期值
func TestNewEnumWithDesc(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
		failure Status = "failure"
	)
	one := enum.NewEnumWithDesc(success, "Success")
	t.Log(one.Code())
	t.Log(one.Meta())
	require.Equal(t, success, one.Code())
	require.Equal(t, "Success", one.Meta().Desc())
	require.Equal(t, "Unknown", enum.NewEnumWithDesc(unknown, "Unknown").Meta().Desc())
	require.Equal(t, "Failure", enum.NewEnumWithDesc(failure, "Failure").Meta().Desc())
}

// TestNewEnumWithMeta tests Enum creation with custom metadata
// Checks Code and custom Meta fields return expected values
//
// 验证带自定义元数据的 Enum 创建
// 测试 Code 和自定义 Meta 字段返回预期值
func TestNewEnumWithMeta(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
		failure Status = "failure"
	)

	type MetaType struct {
		Flag int
		Mark string
	}

	one := enum.NewEnumWithMeta(success, &MetaType{Flag: 200, Mark: "Success"})
	t.Log(one.Code())
	t.Log(one.Meta())
	require.Equal(t, success, one.Code())
	require.Equal(t, 200, one.Meta().Flag)
	require.Equal(t, "Success", one.Meta().Mark)
	require.Equal(t, "Unknown", enum.NewEnumWithMeta(unknown, &MetaType{Flag: 0, Mark: "Unknown"}).Meta().Mark)
	require.Equal(t, 400, enum.NewEnumWithMeta(failure, &MetaType{Flag: 400, Mark: "Failure"}).Meta().Flag)
}