chore: improve test case

Author: sysulq

kod_test.go

@@ -3,8 +3,11 @@ package kod
 import (
 	"context"
 	"testing"
+	"time"
 
+	"github.com/knadh/koanf/v2"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 	"go.uber.org/goleak"
 )
 
@@ -51,3 +54,100 @@ func TestConfigEnv(t *testing.T) {
 	assert.Equal(t, k.config.Version, "1.0.0")
 	assert.Equal(t, k.config.Env, "dev")
 }
+
+type testComponent struct {
+	Implements[testInterface]
+	WithConfig[testConfig]
+	initialized bool
+	initErr     error
+	shutdown    bool
+	shutdownErr error
+}
+
+type testConfig struct {
+	Value string `default:"default"`
+}
+
+type testInterface interface {
+	IsInitialized() bool
+}
+
+func (c *testComponent) Init(context.Context) error {
+	c.initialized = true
+	return c.initErr
+}
+
+func (c *testComponent) Shutdown(context.Context) error {
+	c.shutdown = true
+	return c.shutdownErr
+}
+
+func (c *testComponent) IsInitialized() bool {
+	return c.initialized
+}
+
+func (c *testComponent) implements(testInterface) {}
+
+func TestConfigurationLoading(t *testing.T) {
+	tests := []struct {
+		name     string
+		koanf    *koanf.Koanf
+		filename string
+		wantErr  bool
+	}{
+		{
+			name:  "custom koanf",
+			koanf: koanf.New("."), // 使用 koanf.New() 替代空实例
+		},
+		{
+			name:     "invalid file extension",
+			filename: "config.invalid",
+			wantErr:  true,
+		},
+		{
+			name:     "missing file",
+			filename: "notexist.yaml",
+			wantErr:  true, // Should use defaults
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			opts := []func(*options){
+				WithConfigFile(tt.filename),
+			}
+			if tt.koanf != nil {
+				opts = append(opts, WithKoanf(tt.koanf))
+			}
+
+			k, err := newKod(context.Background(), opts...)
+			if tt.wantErr {
+				assert.Error(t, err)
+				return
+			}
+			require.NoError(t, err)
+
+			cfg := k.Config()
+			assert.NotEmpty(t, cfg.Name)
+			assert.NotEmpty(t, cfg.Env)
+			assert.Equal(t, 5*time.Second, cfg.ShutdownTimeout)
+		})
+	}
+}
+
+func TestDeferHooks(t *testing.T) {
+	k, err := newKod(context.Background())
+	require.NoError(t, err)
+
+	executed := false
+	k.Defer("test", func(context.Context) error {
+		executed = true
+		return nil
+	})
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	k.hooker.Do(ctx)
+	assert.True(t, executed)
+}

registry_test.go

@@ -1,12 +1,14 @@
 package kod
 
 import (
+	"context"
 	"io"
 	"reflect"
 	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 
 	"github.com/go-kod/kod/internal/registry"
 )
@@ -104,3 +106,19 @@ func TestCycleRegistrations(t *testing.T) {
 		t.Fatalf("checkCircularDependency: got %q, want %q", err, want)
 	}
 }
+
+func TestGetImpl(t *testing.T) {
+	k, err := newKod(context.Background())
+	require.NoError(t, err)
+
+	_, err = k.getImpl(context.Background(), reflect.TypeOf(struct{}{}))
+	assert.Error(t, err) // Should fail for unregistered type
+}
+
+func TestGetIntf(t *testing.T) {
+	k, err := newKod(context.Background())
+	require.NoError(t, err)
+
+	_, err = k.getIntf(context.Background(), reflect.TypeOf((*interface{})(nil)).Elem())
+	assert.Error(t, err) // Should fail for unregistered interface
+}

testing.go

@@ -61,7 +61,7 @@ func runTest(tb testing.TB, testBody any, opts ...func(*options)) {
 
 	err := runner{options: opts}.sub(tb, testBody)
 	if err != nil {
-		tb.Logf("runTest failed: %v", err)
+		tb.Errorf("runTest failed: %v", err)
 		tb.FailNow()
 	}
 }

testing_test.go

@@ -10,12 +10,6 @@ import (
 	"github.com/go-kod/kod/internal/mock"
 )
 
-type testComponent struct {
-	Implements[testInterface]
-}
-
-type testInterface interface{}
-
 func Test_testRunner_sub(t *testing.T) {
 	t.Run("failure", func(t *testing.T) {
 		mock.ExpectFailure(t, func(tb testing.TB) {