28 lines
830 B
Go
28 lines
830 B
Go
package mock
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
"github.com/influxdata/influxdb/query"
|
|
)
|
|
|
|
var _ query.ProxyQueryService = (*ProxyQueryService)(nil)
|
|
|
|
// ProxyQueryService is a mock implementation of a query.ProxyQueryService.
|
|
type ProxyQueryService struct {
|
|
QueryFn func(context.Context, io.Writer, *query.ProxyRequest) (int64, error)
|
|
}
|
|
|
|
// NewProxyQueryService returns a mock of ProxyQueryService where its methods will return zero values.
|
|
func NewProxyQueryService() *ProxyQueryService {
|
|
return &ProxyQueryService{
|
|
QueryFn: func(context.Context, io.Writer, *query.ProxyRequest) (int64, error) { return 0, nil },
|
|
}
|
|
}
|
|
|
|
// Query performs the requested query and encodes the results into w.
|
|
func (s *ProxyQueryService) Query(ctx context.Context, w io.Writer, req *query.ProxyRequest) (int64, error) {
|
|
return s.QueryFn(ctx, w, req)
|
|
}
|