mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-19 14:27:32 +00:00
21 lines
417 B
Go
21 lines
417 B
Go
|
package mock
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/golang/mock/gomock"
|
||
|
|
|
||
|
|
"github.com/caos/zitadel/internal/tracing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewSimpleMockTracer(t *testing.T) *MockTracer {
|
||
|
|
return NewMockTracer(gomock.NewController(t))
|
||
|
|
}
|
||
|
|
|
||
|
|
func ExpectServerSpan(ctx context.Context, mock interface{}) {
|
||
|
|
m := mock.(*MockTracer)
|
||
|
|
any := gomock.Any()
|
||
|
|
m.EXPECT().NewServerSpan(any, any).AnyTimes().Return(ctx, &tracing.Span{})
|
||
|
|
}
|