27 lines
483 B
Go
27 lines
483 B
Go
|
package handler
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/golang/mock/gomock"
|
||
|
|
)
|
||
|
|
|
||
|
|
type contextWithValuesMatcher struct {
|
||
|
|
baseCtx context.Context
|
||
|
|
}
|
||
|
|
|
||
|
|
func WrapsContext(ctx context.Context) gomock.Matcher {
|
||
|
|
return contextWithValuesMatcher{ctx}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c contextWithValuesMatcher) Matches(x interface{}) bool {
|
||
|
|
ctx, ok := x.(contextWithValues)
|
||
|
|
if !ok {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
return ctx.Context == c.baseCtx
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c contextWithValuesMatcher) String() string {
|
||
|
|
return "wraps base context "
|
||
|
|
}
|