github.com/kiali/kiali@v1.84.0/tracing/tracingtest/mock.go (about)

     1  package tracingtest
     2  
     3  import (
     4  	"time"
     5  
     6  	"github.com/stretchr/testify/mock"
     7  
     8  	"github.com/kiali/kiali/models"
     9  	"github.com/kiali/kiali/tracing/jaeger/model"
    10  )
    11  
    12  type TracingClientMock struct {
    13  	mock.Mock
    14  }
    15  
    16  func (j *TracingClientMock) GetAppTraces(ns, app string, query models.TracingQuery) (traces *model.TracingResponse, err error) {
    17  	args := j.Called(ns, app, query)
    18  	return args.Get(0).(*model.TracingResponse), args.Error(1)
    19  }
    20  
    21  func (j *TracingClientMock) GetTraceDetail(traceId string) (trace *model.TracingSingleTrace, err error) {
    22  	args := j.Called(traceId)
    23  	return args.Get(0).(*model.TracingSingleTrace), args.Error(1)
    24  }
    25  
    26  func (j *TracingClientMock) GetErrorTraces(ns string, app string, duration time.Duration) (errorTraces int, err error) {
    27  	args := j.Called(ns, app, duration)
    28  	return args.Get(0).(int), args.Error(1)
    29  }
    30  
    31  func (j *TracingClientMock) GetServiceStatus() (available bool, err error) {
    32  	args := j.Called()
    33  	return args.Get(0).(bool), args.Error(1)
    34  }