gocon_test.go

package gocon_test

import (
	"context"
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/s6n-labs/gocon"
)

type Greeter struct {
	message  string
	disposed bool
}

func (g *Greeter) Greet() string {
	return g.message
}

func (g *Greeter) Dispose() {
	g.disposed = true
}

type AnotherGreeter struct {
	name string
}

func (g *AnotherGreeter) Greet() string {
	return fmt.Sprintf("Hello, %s!", g.name)
}

type GreeterLike interface {
	Greet() string
}

func TestKeyOf(t *testing.T) {
	t.Parallel()

	assert.Equal(t, "github.com/s6n-labs/gocon_test.Greeter", gocon.KeyOf[Greeter]())
	assert.Equal(t, "*github.com/s6n-labs/gocon_test.Greeter", gocon.KeyOf[*Greeter]())
}

func Test_SimpleGetSet(t *testing.T) {
	container := gocon.NewContainer(nil)
	ctx := gocon.WithContainer(context.Background(), container)

	greeter := &Greeter{message: "Hello, world!"}
	err := gocon.Set(ctx, gocon.Value(greeter))
	require.NoError(t, err)

	def, err := gocon.FromContext(ctx).Get(gocon.KeyOf[*Greeter]())
	require.NoError(t, err)
	assert.Len(t, def.Tags, 0)

	actual, err := gocon.Get[*Greeter](ctx)
	require.NoError(t, err)
	assert.Equal(t, "Hello, world!", actual.message)
}

func Test_TaggedGetSet(t *testing.T) {
	container := gocon.NewContainer(nil)
	ctx := gocon.WithContainer(context.Background(), container)

	greeter1 := &Greeter{message: "Hello, world!"}
	greeter2 := &AnotherGreeter{name: "John"}

	require.NoError(t, gocon.Set(ctx, gocon.Value(greeter1).WithTags("greeter")))
	require.NoError(t, gocon.Set(ctx, gocon.Value(greeter2).WithTags("greeter")))

	defs, err := gocon.FromContext(ctx).GetTagged("greeter")
	require.NoError(t, err)
	assert.Len(t, defs, 2)

	greeters, err := gocon.GetTagged[GreeterLike](ctx, "greeter")
	require.NoError(t, err)
	assert.Len(t, greeters, 2)

	for _, g := range greeters {
		message := g.Greet()
		assert.True(t, message == "Hello, world!" || message == "Hello, John!")
	}
}