with_default_test.go

package enum_test

import (
	"testing"

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

// TestEnums_DefaultValue verifies default value features
// Tests that enums return default value when lookup fails
//
// 验证默认值功能
// 测试查找失败时枚举返回默认值
func TestEnums_DefaultValue(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
		failure Status = "failure"
	)

	enums := enum.NewEnums(
		enum.NewEnum(unknown),
		enum.NewEnum(success),
		enum.NewEnum(failure),
	)

	// Test that first item becomes default
	defaultEnum := enums.GetDefault()
	require.NotNil(t, defaultEnum)
	require.Equal(t, unknown, defaultEnum.Code())

	// Test lookup returns default when not found
	notFound := enums.Get(Status("not_exists"))
	require.NotNil(t, notFound)
	require.Equal(t, defaultEnum, notFound)
}

// TestEnums_GetDefaultCode verifies GetDefaultCode returns the code value
// Tests that GetDefaultCode returns the correct code type
//
// 验证 GetDefaultCode 返回 code 值
// 测试 GetDefaultCode 返回正确的 code 类型
func TestEnums_GetDefaultCode(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
		failure Status = "failure"
	)

	enums := enum.NewEnums(
		enum.NewEnum(unknown),
		enum.NewEnum(success),
		enum.NewEnum(failure),
	)

	// Test GetDefaultCode returns the code value
	defaultCode := enums.GetDefaultCode()
	t.Log(defaultCode)
	require.Equal(t, unknown, defaultCode)

	// Test panic when no default is set
	enums.UnsetDefault()
	require.Panics(t, func() {
		enums.GetDefaultCode()
	})
}

// TestEnums_SetDefault verifies default value setting once unset
// Tests that SetDefault works once the existing default is unset
//
// 验证丢弃后设置默认值
// 测试 SetDefault 在丢弃现有默认值后可以工作
func TestEnums_SetDefault(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
		failure Status = "failure"
	)

	// Create a new enum collection with auto default
	enums := enum.NewEnums(
		enum.NewEnum(unknown),
		enum.NewEnum(success),
		enum.NewEnum(failure),
	)

	// Must unset default first before setting new one
	enums.UnsetDefault()

	// Now set default to SUCCESS
	successEnum := enums.MustGet(success)
	enums.SetDefault(successEnum)

	// Check new default
	newDefault := enums.GetDefault()
	require.NotNil(t, newDefault)
	require.Equal(t, success, newDefault.Code())

	// Test lookup returns new default when not found
	notFound := enums.Get(Status("not_exists"))
	require.NotNil(t, notFound)
	require.Equal(t, success, notFound.Code())
}

// TestEnums_SetDefaultPanicsOnDuplicate verifies SetDefault panics when default exists
// Tests that SetDefault panics when default value is present
//
// 验证 SetDefault 在已有默认值时 panic
// 测试当默认值已存在时 SetDefault 会 panic
func TestEnums_SetDefaultPanicsOnDuplicate(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
	)

	enums := enum.NewEnums(
		enum.NewEnum(unknown),
		enum.NewEnum(success),
	)

	// SetDefault should panic because default is present (first item)
	require.Panics(t, func() {
		successEnum := enums.MustGet(success)
		enums.SetDefault(successEnum)
	})
}

// TestEnums_SetDefaultNilPanics verifies that SetDefault panics with nil input
// Tests that passing nil to SetDefault causes panic
//
// 验证 SetDefault 在 nil 参数时 panic
// 测试传递 nil 给 SetDefault 会导致 panic
func TestEnums_SetDefaultNilPanics(t *testing.T) {
	type Status string

	// Create blank collection without default
	enums := enum.NewEnums[Status, *enum.MetaNone]()

	// SetDefault with nil should panic (must.Full check)
	require.Panics(t, func() {
		enums.SetDefault(nil)
	})
}

// TestEnums_UnsetDefault verifies unsetting the default value
// Tests that UnsetDefault removes the default value and lookups panic
//
// 验证取消设置默认值
// 测试 UnsetDefault 移除默认值后查找会 panic
func TestEnums_UnsetDefault(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
		failure Status = "failure"
	)

	// Create a new enum collection with default
	enums := enum.NewEnums(
		enum.NewEnum(unknown),
		enum.NewEnum(success),
		enum.NewEnum(failure),
	)

	// Check default exists
	require.NotNil(t, enums.GetDefault())
	require.Equal(t, unknown, enums.GetDefault().Code())

	// Unset the default
	enums.UnsetDefault()

	// Check GetDefault panics once unset
	require.Panics(t, func() {
		enums.GetDefault()
	})

	// Test Get also panics when not found (because no default)
	require.Panics(t, func() {
		enums.Get(Status("not_exists"))
	})
}

// TestEnums_WithUnsetDefault verifies chain-style default unset
// Tests that WithUnsetDefault removes default value and returns the instance
//
// 验证链式取消设置默认值
// 测试 WithUnsetDefault 移除默认值并返回实例
func TestEnums_WithUnsetDefault(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
		failure Status = "failure"
	)

	// Create enum collection and unset default in chain
	enums := enum.NewEnums(
		enum.NewEnum(unknown),
		enum.NewEnum(success),
		enum.NewEnum(failure),
	).WithUnsetDefault()

	// Check GetDefault panics once chain unset
	require.Panics(t, func() {
		enums.GetDefault()
	})

	// Test Get also panics when not found (because no default)
	require.Panics(t, func() {
		enums.Get(Status("not_exists"))
	})
}

// TestEnums_ChainMethods verifies chain-style configuration methods
// Tests WithDefault and WithDefaultCode with fluent API
// Checks panic actions when invalid code is specified
//
// 验证链式配置方法
// 测试 WithDefault 和 WithDefaultCode 的流式 API
// 同时验证指定无效代码时的 panic 行为
func TestEnums_ChainMethods(t *testing.T) {
	// Test WithDefault chain method - add enum and set as default in one chain
	t.Run("with-default-enum", func(t *testing.T) {
		type Status string
		const (
			success Status = "success"
		)

		// Create blank collection, then add enum and set as default using chain method
		enums := enum.NewEnums[Status, *enum.MetaNone]().
			WithDefault(enum.NewEnum(success))

		require.NotNil(t, enums.GetDefault())
		require.Equal(t, success, enums.GetDefault().Code())
	})

	// Test that WithDefaultXxx panics when default is present
	t.Run("with-default-panics-on-existing", func(t *testing.T) {
		type Status string
		const (
			unknown Status = "unknown"
			success Status = "success"
		)

		require.Panics(t, func() {
			// NewEnums sets first item as default, then WithDefaultCode should panic
			enum.NewEnums(
				enum.NewEnum(unknown),
				enum.NewEnum(success),
			).WithDefaultCode(success)
		})
	})

	// Test unset then set pattern
	t.Run("unset-then-set-default", func(t *testing.T) {
		type Status string
		const (
			unknown Status = "unknown"
			success Status = "success"
			failure Status = "failure"
		)

		enums := enum.NewEnums(
			enum.NewEnum(unknown),
			enum.NewEnum(success),
			enum.NewEnum(failure),
		)
		// Must unset first, then set new default
		enums.UnsetDefault()
		successEnum := enums.MustGet(success)
		enums.SetDefault(successEnum)

		require.NotNil(t, enums.GetDefault())
		require.Equal(t, success, enums.GetDefault().Code())
	})

	// Test chain with non-existent code (should panic)
	t.Run("with-invalid-code-panics", func(t *testing.T) {
		type Status string

		require.Panics(t, func() {
			enum.NewEnums[Status, *enum.MetaNone]().WithDefaultCode(Status("not_exists"))
		})
	})
}

// TestEnums_SetDefaultCode verifies SetDefaultCode sets default using code
// Tests that SetDefaultCode works with valid code value
//
// 验证 SetDefaultCode 使用 code 设置默认值
// 测试 SetDefaultCode 对有效 code 值的工作情况
func TestEnums_SetDefaultCode(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
		failure Status = "failure"
	)

	enums := enum.NewEnums(
		enum.NewEnum(unknown),
		enum.NewEnum(success),
		enum.NewEnum(failure),
	)

	// Must unset default first
	enums.UnsetDefault()

	// Set default using code
	enums.SetDefaultCode(success)

	// Check new default
	require.Equal(t, success, enums.GetDefaultCode())
}

// TestEnums_SetDefaultValid verifies SetDefaultValid marks default as active
// Tests that SetDefaultValid affects Valid() result
//
// 验证 SetDefaultValid 标记默认值为有效
// 测试 SetDefaultValid 影响 Valid() 结果
func TestEnums_SetDefaultValid(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
		failure Status = "failure"
	)

	enums := enum.NewEnums(
		enum.NewEnum(unknown),
		enum.NewEnum(success),
		enum.NewEnum(failure),
	)

	// Before SetDefaultValid, ListValid() excludes default
	validCodes := enums.ListValid()
	require.Len(t, validCodes, 2)
	require.NotContains(t, validCodes, unknown)
	require.Contains(t, validCodes, success)
	require.Contains(t, validCodes, failure)

	// Set default as valid
	enums.SetDefaultValid(true)

	// After SetDefaultValid(true), ListValid() includes default
	validCodesWithDefault := enums.ListValid()
	require.Len(t, validCodesWithDefault, 3)
	require.Contains(t, validCodesWithDefault, unknown)
}

// TestEnums_WithDefaultValid verifies chain-style SetDefaultValid
// Tests that WithDefaultValid sets valid state and returns instance
//
// 验证链式 SetDefaultValid
// 测试 WithDefaultValid 设置有效状态并返回实例
func TestEnums_WithDefaultValid(t *testing.T) {
	type Status string
	const (
		unknown Status = "unknown"
		success Status = "success"
		failure Status = "failure"
	)

	// Create enum collection and set default valid in chain
	enums := enum.NewEnums(
		enum.NewEnum(unknown),
		enum.NewEnum(success),
		enum.NewEnum(failure),
	).WithDefaultValid(true)

	// ListValid() should include default
	validCodes := enums.ListValid()
	require.Len(t, validCodes, 3)
	require.Contains(t, validCodes, unknown)
}