Моки и t.Cleanup

[f0rtung]

Добрый день!
Подскажите, как правильно создавать моки и ожидать вызов на них метода AssertExpectations.
Сейчас есть проблема для такого кода:

type BeforeAfterDemoSuite struct {
	suite.Suite

	serive *mocks.Service
}

func (s *BeforeAfterDemoSuite) BeforeEach(t provider.T) {
	t.NewStep("Before suite Step")
	s.serive = mocks.NewService(t)
}

func (s *BeforeAfterDemoSuite) AfterEach(t provider.T) {
	t.NewStep("After suite Step")
}

func (s *BeforeAfterDemoSuite) TestBeforeAfterTest(t provider.T) {
	s.serive.EXPECT().doWork().Return(nil)
	t.Tags("BeforeAfter")
}

func TestBeforesAfters(t *testing.T) {
	suite.RunSuite(t, new(BeforeAfterDemoSuite))
}

в BeforeEach в вызове mocks.NewService(t) регистрируется cleanup-колбек

func NewService(t interface {
	mock.TestingT
	Cleanup(func())
}) *Service {
	mock := &Service{}
	mock.Mock.Test(t)

	t.Cleanup(func() {mock.AssertExpectations(t)})

	return mock
}

котоырй вызывается сразу же после окончания BeforeEach, а не после выполнения теста. Это приводит к тому, что в самом тесте нет проверки на то, что все ожидаемые вызовы на моке были выполнены и тест не проваливается.
Нужно в другом месте создавать моки?