go-libs/pkg/testing/deferred/ginkgo/helpers.go

package ginkgo

import (
	"context"

	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"

	"github.com/hanzo-fi/go-libs/v5/pkg/testing/deferred"
)

func DeferMap[From, To any](d *deferred.Deferred[From], mapper func(From) To) *deferred.Deferred[To] {
	ret := deferred.New[To]()
	BeforeEach(func(specContext SpecContext) {
		ret.Reset()
		ret.LoadAsync(func() (To, error) {
			return deferred.WaitAndMap(d, mapper)
		})
	})
	return ret
}

func Wait[V any](ctx context.Context, d *deferred.Deferred[V]) V {
	v, err := d.Wait(ctx)
	Expect(err).To(BeNil())

	return v
}