stubs_data.go

package suite_init

import (
	. "github.com/onsi/ginkgo/v2"
	"github.com/prashantv/gostub"
)

type StubsData struct {
	Stubs *gostub.Stubs
}

func NewStubsData() *StubsData {
	data := &StubsData{
		Stubs: gostub.New(),
	}
	SetupStubs(data.Stubs)
	return data
}

func SetupStubs(stubs *gostub.Stubs) bool {
	return AfterEach(func() {
		stubs.Reset()
	})
}