diff --git a/submission/queue_test.go b/submission/queue_test.go index ed0453d62e40f6a9e6940243f646f26e8c3d860b..9595e257c3f3b8370565ebad0803afe49afb9117 100644 --- a/submission/queue_test.go +++ b/submission/queue_test.go @@ -16,11 +16,9 @@ type fakeClient struct { func (f *fakeClient) Close() {} -func (f *fakeClient) Submit(_ context.Context, ev *ippb.Event, aggr *ippb.Aggregate) error { +func (f *fakeClient) Submit(_ context.Context, evs []*ippb.Event, aggr *ippb.Aggregate) error { f.calls++ - if ev != nil { - f.rcvd++ - } + f.rcvd += len(evs) if aggr != nil { for _, bt := range aggr.ByType { for _, bi := range bt.ByIp { @@ -35,6 +33,10 @@ func (f *fakeClient) GetScore(_ context.Context, _ string) (*ippb.GetScoreRespon return nil, errors.New("not implemented") } +func (f *fakeClient) GetAllScores(_ context.Context, _ float32) (<-chan *ippb.GetScoreResponse, error) { + return nil, errors.New("not implemented") +} + func rateDo(timeout time.Duration, qps float64, f func()) int { end := time.After(timeout) interval := 1 * time.Second