From 341fff2004ba71247657a53bd7831965f39d7d0e Mon Sep 17 00:00:00 2001
From: ale <ale@incal.net>
Date: Sun, 18 Apr 2021 23:06:16 +0100
Subject: [PATCH] Add missing methods to test stub
---
submission/queue_test.go | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/submission/queue_test.go b/submission/queue_test.go
index ed0453d..9595e25 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
--
GitLab