github.com/yankunsam/loki/v2@v2.6.3-0.20220817130409-389df5235c27/pkg/util/mock.go (about)

     1  package util
     2  
     3  import (
     4  	"github.com/stretchr/testify/mock"
     5  )
     6  
     7  type ExtendedMock struct {
     8  	mock.Mock
     9  }
    10  
    11  func (m *ExtendedMock) GetMockedCallsByMethod(method string) []mock.Call {
    12  	calls := make([]mock.Call, 0)
    13  
    14  	for _, call := range m.Calls {
    15  		if call.Method == method {
    16  			calls = append(calls, call)
    17  		}
    18  	}
    19  
    20  	return calls
    21  }