blob: ceef244bc6e4f79ac48134f6648b80ce6a3de0be [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
// Code generated by mockery v2.42.2. DO NOT EDIT.
package tracer
import mock "github.com/stretchr/testify/mock"
// MockTracer is an autogenerated mock type for the Tracer type
type MockTracer struct {
mock.Mock
}
type MockTracer_Expecter struct {
mock *mock.Mock
}
func (_m *MockTracer) EXPECT() *MockTracer_Expecter {
return &MockTracer_Expecter{mock: &_m.Mock}
}
// AddTrace provides a mock function with given fields: operation, message
func (_m *MockTracer) AddTrace(operation string, message string) {
_m.Called(operation, message)
}
// MockTracer_AddTrace_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddTrace'
type MockTracer_AddTrace_Call struct {
*mock.Call
}
// AddTrace is a helper method to define mock.On call
// - operation string
// - message string
func (_e *MockTracer_Expecter) AddTrace(operation interface{}, message interface{}) *MockTracer_AddTrace_Call {
return &MockTracer_AddTrace_Call{Call: _e.mock.On("AddTrace", operation, message)}
}
func (_c *MockTracer_AddTrace_Call) Run(run func(operation string, message string)) *MockTracer_AddTrace_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(string), args[1].(string))
})
return _c
}
func (_c *MockTracer_AddTrace_Call) Return() *MockTracer_AddTrace_Call {
_c.Call.Return()
return _c
}
func (_c *MockTracer_AddTrace_Call) RunAndReturn(run func(string, string)) *MockTracer_AddTrace_Call {
_c.Call.Return(run)
return _c
}
// AddTransactionalStartTrace provides a mock function with given fields: operation, message
func (_m *MockTracer) AddTransactionalStartTrace(operation string, message string) string {
ret := _m.Called(operation, message)
if len(ret) == 0 {
panic("no return value specified for AddTransactionalStartTrace")
}
var r0 string
if rf, ok := ret.Get(0).(func(string, string) string); ok {
r0 = rf(operation, message)
} else {
r0 = ret.Get(0).(string)
}
return r0
}
// MockTracer_AddTransactionalStartTrace_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddTransactionalStartTrace'
type MockTracer_AddTransactionalStartTrace_Call struct {
*mock.Call
}
// AddTransactionalStartTrace is a helper method to define mock.On call
// - operation string
// - message string
func (_e *MockTracer_Expecter) AddTransactionalStartTrace(operation interface{}, message interface{}) *MockTracer_AddTransactionalStartTrace_Call {
return &MockTracer_AddTransactionalStartTrace_Call{Call: _e.mock.On("AddTransactionalStartTrace", operation, message)}
}
func (_c *MockTracer_AddTransactionalStartTrace_Call) Run(run func(operation string, message string)) *MockTracer_AddTransactionalStartTrace_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(string), args[1].(string))
})
return _c
}
func (_c *MockTracer_AddTransactionalStartTrace_Call) Return(_a0 string) *MockTracer_AddTransactionalStartTrace_Call {
_c.Call.Return(_a0)
return _c
}
func (_c *MockTracer_AddTransactionalStartTrace_Call) RunAndReturn(run func(string, string) string) *MockTracer_AddTransactionalStartTrace_Call {
_c.Call.Return(run)
return _c
}
// AddTransactionalTrace provides a mock function with given fields: transactionId, operation, message
func (_m *MockTracer) AddTransactionalTrace(transactionId string, operation string, message string) {
_m.Called(transactionId, operation, message)
}
// MockTracer_AddTransactionalTrace_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddTransactionalTrace'
type MockTracer_AddTransactionalTrace_Call struct {
*mock.Call
}
// AddTransactionalTrace is a helper method to define mock.On call
// - transactionId string
// - operation string
// - message string
func (_e *MockTracer_Expecter) AddTransactionalTrace(transactionId interface{}, operation interface{}, message interface{}) *MockTracer_AddTransactionalTrace_Call {
return &MockTracer_AddTransactionalTrace_Call{Call: _e.mock.On("AddTransactionalTrace", transactionId, operation, message)}
}
func (_c *MockTracer_AddTransactionalTrace_Call) Run(run func(transactionId string, operation string, message string)) *MockTracer_AddTransactionalTrace_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(string), args[1].(string), args[2].(string))
})
return _c
}
func (_c *MockTracer_AddTransactionalTrace_Call) Return() *MockTracer_AddTransactionalTrace_Call {
_c.Call.Return()
return _c
}
func (_c *MockTracer_AddTransactionalTrace_Call) RunAndReturn(run func(string, string, string)) *MockTracer_AddTransactionalTrace_Call {
_c.Call.Return(run)
return _c
}
// FilterTraces provides a mock function with given fields: traces, connectionIdFilter, transactionIdFilter, operationFilter, messageFilter
func (_m *MockTracer) FilterTraces(traces []TraceEntry, connectionIdFilter string, transactionIdFilter string, operationFilter string, messageFilter string) []TraceEntry {
ret := _m.Called(traces, connectionIdFilter, transactionIdFilter, operationFilter, messageFilter)
if len(ret) == 0 {
panic("no return value specified for FilterTraces")
}
var r0 []TraceEntry
if rf, ok := ret.Get(0).(func([]TraceEntry, string, string, string, string) []TraceEntry); ok {
r0 = rf(traces, connectionIdFilter, transactionIdFilter, operationFilter, messageFilter)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]TraceEntry)
}
}
return r0
}
// MockTracer_FilterTraces_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'FilterTraces'
type MockTracer_FilterTraces_Call struct {
*mock.Call
}
// FilterTraces is a helper method to define mock.On call
// - traces []TraceEntry
// - connectionIdFilter string
// - transactionIdFilter string
// - operationFilter string
// - messageFilter string
func (_e *MockTracer_Expecter) FilterTraces(traces interface{}, connectionIdFilter interface{}, transactionIdFilter interface{}, operationFilter interface{}, messageFilter interface{}) *MockTracer_FilterTraces_Call {
return &MockTracer_FilterTraces_Call{Call: _e.mock.On("FilterTraces", traces, connectionIdFilter, transactionIdFilter, operationFilter, messageFilter)}
}
func (_c *MockTracer_FilterTraces_Call) Run(run func(traces []TraceEntry, connectionIdFilter string, transactionIdFilter string, operationFilter string, messageFilter string)) *MockTracer_FilterTraces_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].([]TraceEntry), args[1].(string), args[2].(string), args[3].(string), args[4].(string))
})
return _c
}
func (_c *MockTracer_FilterTraces_Call) Return(_a0 []TraceEntry) *MockTracer_FilterTraces_Call {
_c.Call.Return(_a0)
return _c
}
func (_c *MockTracer_FilterTraces_Call) RunAndReturn(run func([]TraceEntry, string, string, string, string) []TraceEntry) *MockTracer_FilterTraces_Call {
_c.Call.Return(run)
return _c
}
// GetConnectionId provides a mock function with given fields:
func (_m *MockTracer) GetConnectionId() string {
ret := _m.Called()
if len(ret) == 0 {
panic("no return value specified for GetConnectionId")
}
var r0 string
if rf, ok := ret.Get(0).(func() string); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(string)
}
return r0
}
// MockTracer_GetConnectionId_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetConnectionId'
type MockTracer_GetConnectionId_Call struct {
*mock.Call
}
// GetConnectionId is a helper method to define mock.On call
func (_e *MockTracer_Expecter) GetConnectionId() *MockTracer_GetConnectionId_Call {
return &MockTracer_GetConnectionId_Call{Call: _e.mock.On("GetConnectionId")}
}
func (_c *MockTracer_GetConnectionId_Call) Run(run func()) *MockTracer_GetConnectionId_Call {
_c.Call.Run(func(args mock.Arguments) {
run()
})
return _c
}
func (_c *MockTracer_GetConnectionId_Call) Return(_a0 string) *MockTracer_GetConnectionId_Call {
_c.Call.Return(_a0)
return _c
}
func (_c *MockTracer_GetConnectionId_Call) RunAndReturn(run func() string) *MockTracer_GetConnectionId_Call {
_c.Call.Return(run)
return _c
}
// GetTraces provides a mock function with given fields:
func (_m *MockTracer) GetTraces() []TraceEntry {
ret := _m.Called()
if len(ret) == 0 {
panic("no return value specified for GetTraces")
}
var r0 []TraceEntry
if rf, ok := ret.Get(0).(func() []TraceEntry); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]TraceEntry)
}
}
return r0
}
// MockTracer_GetTraces_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetTraces'
type MockTracer_GetTraces_Call struct {
*mock.Call
}
// GetTraces is a helper method to define mock.On call
func (_e *MockTracer_Expecter) GetTraces() *MockTracer_GetTraces_Call {
return &MockTracer_GetTraces_Call{Call: _e.mock.On("GetTraces")}
}
func (_c *MockTracer_GetTraces_Call) Run(run func()) *MockTracer_GetTraces_Call {
_c.Call.Run(func(args mock.Arguments) {
run()
})
return _c
}
func (_c *MockTracer_GetTraces_Call) Return(_a0 []TraceEntry) *MockTracer_GetTraces_Call {
_c.Call.Return(_a0)
return _c
}
func (_c *MockTracer_GetTraces_Call) RunAndReturn(run func() []TraceEntry) *MockTracer_GetTraces_Call {
_c.Call.Return(run)
return _c
}
// ResetTraces provides a mock function with given fields:
func (_m *MockTracer) ResetTraces() {
_m.Called()
}
// MockTracer_ResetTraces_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ResetTraces'
type MockTracer_ResetTraces_Call struct {
*mock.Call
}
// ResetTraces is a helper method to define mock.On call
func (_e *MockTracer_Expecter) ResetTraces() *MockTracer_ResetTraces_Call {
return &MockTracer_ResetTraces_Call{Call: _e.mock.On("ResetTraces")}
}
func (_c *MockTracer_ResetTraces_Call) Run(run func()) *MockTracer_ResetTraces_Call {
_c.Call.Run(func(args mock.Arguments) {
run()
})
return _c
}
func (_c *MockTracer_ResetTraces_Call) Return() *MockTracer_ResetTraces_Call {
_c.Call.Return()
return _c
}
func (_c *MockTracer_ResetTraces_Call) RunAndReturn(run func()) *MockTracer_ResetTraces_Call {
_c.Call.Return(run)
return _c
}
// SetConnectionId provides a mock function with given fields: connectionId
func (_m *MockTracer) SetConnectionId(connectionId string) {
_m.Called(connectionId)
}
// MockTracer_SetConnectionId_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetConnectionId'
type MockTracer_SetConnectionId_Call struct {
*mock.Call
}
// SetConnectionId is a helper method to define mock.On call
// - connectionId string
func (_e *MockTracer_Expecter) SetConnectionId(connectionId interface{}) *MockTracer_SetConnectionId_Call {
return &MockTracer_SetConnectionId_Call{Call: _e.mock.On("SetConnectionId", connectionId)}
}
func (_c *MockTracer_SetConnectionId_Call) Run(run func(connectionId string)) *MockTracer_SetConnectionId_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(string))
})
return _c
}
func (_c *MockTracer_SetConnectionId_Call) Return() *MockTracer_SetConnectionId_Call {
_c.Call.Return()
return _c
}
func (_c *MockTracer_SetConnectionId_Call) RunAndReturn(run func(string)) *MockTracer_SetConnectionId_Call {
_c.Call.Return(run)
return _c
}
// NewMockTracer creates a new instance of MockTracer. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
// The first argument is typically a *testing.T value.
func NewMockTracer(t interface {
mock.TestingT
Cleanup(func())
}) *MockTracer {
mock := &MockTracer{}
mock.Mock.Test(t)
t.Cleanup(func() { mock.AssertExpectations(t) })
return mock
}