wrote tests

main_test.go

@@ -0,0 +1,83 @@
+package component
+
+import (
+	"context"
+	"github.com/stretchr/testify/assert"
+	"log/slog"
+	"testing"
+)
+
+func TestNew(t *testing.T) {
+	t.Run("creates a new component", func(t *testing.T) {
+		ctx, err := New(context.Background(), "mymodule", "mycomponent")
+		assert.NoError(t, err)
+		c, ok := Get(ctx)
+		assert.True(t, ok)
+		assert.Equal(t, "mymodule", c.Module())
+		assert.Equal(t, []string{"mycomponent"}, c.Path())
+		assert.Equal(t, "mycomponent", c.PathString())
+	})
+
+	t.Run("should return error", func(t *testing.T) {
+		_, err := New(nil, "mymodule", "mycomponent")
+		assert.Error(t, err)
+	})
+
+	t.Run("must variant should panic", func(t *testing.T) {
+		assert.Panics(t, func() {
+			MustNew(nil, "mymodule", "mycomponent")
+		})
+	})
+}
+
+func TestExtend(t *testing.T) {
+	t.Run("extends existing component", func(t *testing.T) {
+		ctx := MustNew(context.Background(), "mymodule", "mycomponent")
+		ctx, err := Extend(ctx, "subcomponent")
+		assert.NoError(t, err)
+		c, ok := Get(ctx)
+		assert.True(t, ok)
+		assert.Equal(t, "mymodule", c.Module())
+		assert.Equal(t, []string{"mycomponent", "subcomponent"}, c.Path())
+		assert.Equal(t, "mycomponent.subcomponent", c.PathString())
+	})
+
+	t.Run("should return error", func(t *testing.T) {
+		_, err := Extend(context.Background(), "subcomponent")
+		assert.Error(t, err)
+	})
+
+	t.Run("must variant should panic", func(t *testing.T) {
+		assert.Panics(t, func() {
+			MustExtend(context.Background(), "subcomponent")
+		})
+	})
+}
+
+func TestGet(t *testing.T) {
+	_, ok := Get(context.Background())
+	assert.False(t, ok)
+
+	ctx := MustNew(context.Background(), "mymodule", "mycomponent")
+	_, ok = Get(ctx)
+	assert.True(t, ok)
+}
+
+func TestGetFields(t *testing.T) {
+	_, ok := GetFields(context.Background())
+	assert.False(t, ok)
+
+	ctx := MustNew(context.Background(), "mymodule", "mycomponent")
+	fields, ok := GetFields(ctx)
+	assert.True(t, ok)
+	assert.Equal(t, "mymodule", fields["module"])
+	assert.Equal(t, "mycomponent", fields["path"])
+}
+
+func TestAttrs(t *testing.T) {
+	ctx := MustNew(context.Background(), "mymodule", "mycomponent")
+	attrs, ok := Attrs(ctx)
+	assert.True(t, ok)
+	assert.Equal(t, slog.String("module", "mymodule"), attrs[0])
+	assert.Equal(t, slog.String("path", "mycomponent"), attrs[1])
+}