invoke_test.go

package imbue_test

import (
	"context"
	"errors"

	"github.com/dogmatiq/imbue"
	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
)

var _ = Describe("func InvokeX()", func() {
	var container *imbue.Container

	BeforeEach(func() {
		container = imbue.New()
	})

	AfterEach(func() {
		container.Close()
	})

	It("can obtain a single dependency from the container", func() {
		imbue.With0(
			container,
			func(ctx imbue.Context) (Concrete1, error) {
				return "<concrete-1>", nil
			},
		)

		err := imbue.Invoke1(
			context.Background(),
			container,
			func(
				ctx context.Context,
				dep Concrete1,
			) error {
				Expect(dep).To(Equal(Concrete1("<concrete-1>")))
				return nil
			},
		)
		Expect(err).ShouldNot(HaveOccurred())
	})

	It("can obtain multiple dependencies from the container", func() {
		imbue.With0(
			container,
			func(ctx imbue.Context) (Concrete1, error) {
				return "<concrete-1>", nil
			},
		)

		imbue.With0(
			container,
			func(ctx imbue.Context) (Concrete2, error) {
				return "<concrete-2>", nil
			},
		)

		err := imbue.Invoke2(
			context.Background(),
			container,
			func(
				ctx context.Context,
				dep1 Concrete1,
				dep2 Concrete2,
			) error {
				Expect(dep1).To(Equal(Concrete1("<concrete-1>")))
				Expect(dep2).To(Equal(Concrete2("<concrete-2>")))
				return nil
			},
		)
		Expect(err).ShouldNot(HaveOccurred())
	})

	It("returns the error returned by the invoked function", func() {
		imbue.With0(
			container,
			func(ctx imbue.Context) (Concrete1, error) {
				return "<concrete>", nil
			},
		)

		err := imbue.Invoke1(
			context.Background(),
			container,
			func(
				ctx context.Context,
				dep Concrete1,
			) error {
				return errors.New("<error>")
			},
		)
		Expect(err).To(MatchError("<error>"))
	})

	It("returns an error when a constructor returns an error", func() {
		imbue.With0(
			container,
			func(ctx imbue.Context) (Concrete1, error) {
				return "", errors.New("<error>")
			},
		)

		err := imbue.Invoke1(
			context.Background(),
			container,
			func(
				ctx context.Context,
				dep Concrete1,
			) error {
				panic("unexpected call")
			},
		)
		Expect(err).Should(HaveOccurred())
		Expect(err.Error()).To(MatchRegexp(
			`imbue_test\.Concrete1 constructor \(invoke_test\.go:\d+\) failed: <error>`,
		))
	})

	It("returns an error when an decorator returns an error", func() {
		imbue.With0(
			container,
			func(ctx imbue.Context) (Concrete1, error) {
				return "<concrete-1>", nil
			},
		)

		imbue.With0(
			container,
			func(ctx imbue.Context) (Concrete2, error) {
				return "<concrete-2>", nil
			},
		)

		imbue.Decorate1(
			container,
			func(
				ctx imbue.Context,
				v Concrete1,
				dep Concrete2,
			) (Concrete1, error) {
				return "", errors.New("<error>")
			},
		)

		err := imbue.Invoke1(
			context.Background(),
			container,
			func(
				ctx context.Context,
				dep Concrete1,
			) error {
				panic("unexpected call")
			},
		)
		Expect(err).Should(HaveOccurred())
		Expect(err.Error()).To(MatchRegexp(
			`imbue_test\.Concrete1 decorator \(invoke_test\.go:\d+\) failed: <error>`,
		))
	})

	It("panics when a requested dependency is not registered", func() {
		Expect(func() {
			imbue.Invoke1(
				context.Background(),
				container,
				func(
					ctx context.Context,
					dep Concrete1,
				) error {
					panic("unexpected call")
				},
			)
		}).To(
			PanicWith(
				`no constructor is declared for imbue_test.Concrete1`,
			),
		)
	})

	It("panics when an upstream dependency is not registered", func() {
		imbue.With1(
			container,
			func(
				ctx imbue.Context,
				dep Concrete2,
			) (Concrete1, error) {
				return "<concrete-1>", nil
			},
		)

		Expect(func() {
			imbue.Invoke1(
				context.Background(),
				container,
				func(
					ctx context.Context,
					dep Concrete1,
				) error {
					panic("unexpected call")
				},
			)
		}).To(
			PanicWith(
				`no constructor is declared for imbue_test.Concrete2`,
			),
		)
	})
})

var _ = Describe("func Invoke0()", func() {
	var container *imbue.Container

	BeforeEach(func() {
		container = imbue.New()
	})

	AfterEach(func() {
		container.Close()
	})

	It("calls the function", func() {
		err := imbue.Invoke0(
			context.Background(),
			container,
			func(
				ctx context.Context,
			) error {
				return errors.New("<error>")
			},
		)
		Expect(err).To(MatchError("<error>"))
	})
})