removed testify dependency

This commit is contained in:
Jay
2026-05-09 11:06:03 -04:00
parent 2f85553d1c
commit bd05712982
3 changed files with 82 additions and 42 deletions
+82 -24
View File
@@ -2,29 +2,51 @@ package component
import (
"context"
"github.com/stretchr/testify/assert"
"log/slog"
"slices"
"testing"
)
func assertPanics(t *testing.T, f func()) {
t.Helper()
defer func() {
if recover() == nil {
t.Error("expected panic, but none occurred")
}
}()
f()
}
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)
if err != nil {
t.Fatalf("expected no error, got %v", 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())
if !ok {
t.Fatal("expected component in context")
}
if c.Module() != "mymodule" {
t.Errorf("expected module %q, got %q", "mymodule", c.Module())
}
if !slices.Equal(c.Path(), []string{"mycomponent"}) {
t.Errorf("expected path %v, got %v", []string{"mycomponent"}, c.Path())
}
if c.PathString() != "mycomponent" {
t.Errorf("expected path string %q, got %q", "mycomponent", c.PathString())
}
})
t.Run("should return error", func(t *testing.T) {
_, err := New(nil, "mymodule", "mycomponent")
assert.Error(t, err)
if err == nil {
t.Fatal("expected error, got nil")
}
})
t.Run("must variant should panic", func(t *testing.T) {
assert.Panics(t, func() {
assertPanics(t, func() {
MustNew(nil, "mymodule", "mycomponent")
})
})
@@ -34,21 +56,33 @@ 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)
if err != nil {
t.Fatalf("expected no error, got %v", 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())
if !ok {
t.Fatal("expected component in context")
}
if c.Module() != "mymodule" {
t.Errorf("expected module %q, got %q", "mymodule", c.Module())
}
if !slices.Equal(c.Path(), []string{"mycomponent", "subcomponent"}) {
t.Errorf("expected path %v, got %v", []string{"mycomponent", "subcomponent"}, c.Path())
}
if c.PathString() != "mycomponent.subcomponent" {
t.Errorf("expected path string %q, got %q", "mycomponent.subcomponent", c.PathString())
}
})
t.Run("should return error", func(t *testing.T) {
_, err := Extend(context.Background(), "subcomponent")
assert.Error(t, err)
if err == nil {
t.Fatal("expected error, got nil")
}
})
t.Run("must variant should panic", func(t *testing.T) {
assert.Panics(t, func() {
assertPanics(t, func() {
MustExtend(context.Background(), "subcomponent")
})
})
@@ -56,28 +90,52 @@ func TestExtend(t *testing.T) {
func TestGet(t *testing.T) {
_, ok := Get(context.Background())
assert.False(t, ok)
if ok {
t.Fatal("expected no component on bare context")
}
ctx := MustNew(context.Background(), "mymodule", "mycomponent")
_, ok = Get(ctx)
assert.True(t, ok)
if !ok {
t.Fatal("expected component in context")
}
}
func TestGetFields(t *testing.T) {
_, ok := GetFields(context.Background())
assert.False(t, ok)
if ok {
t.Fatal("expected no fields on bare context")
}
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"])
if !ok {
t.Fatal("expected fields in context")
}
if fields["module"] != "mymodule" {
t.Errorf("expected module %q, got %q", "mymodule", fields["module"])
}
if fields["path"] != "mycomponent" {
t.Errorf("expected path %q, got %q", "mycomponent", fields["path"])
}
}
func assertAttr(t *testing.T, attr slog.Attr, key string, value string) {
t.Helper()
if attr.Key != key {
t.Errorf("expected attr key %q, got %q", key, attr.Key)
}
if attr.Value.String() != value {
t.Errorf("expected attr value %q, got %q", value, attr.Value.String())
}
}
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])
if !ok {
t.Fatal("expected attrs in context")
}
assertAttr(t, attrs[0], "module", "mymodule")
assertAttr(t, attrs[1], "path", "mycomponent")
}