summaryrefslogtreecommitdiffstats
path: root/src/orchestrator/pkg/infra/db/mongo_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'src/orchestrator/pkg/infra/db/mongo_test.go')
-rw-r--r--src/orchestrator/pkg/infra/db/mongo_test.go5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/orchestrator/pkg/infra/db/mongo_test.go b/src/orchestrator/pkg/infra/db/mongo_test.go
index 9f813ca4..d57c19dd 100644
--- a/src/orchestrator/pkg/infra/db/mongo_test.go
+++ b/src/orchestrator/pkg/infra/db/mongo_test.go
@@ -79,6 +79,11 @@ func (c *mockCollection) UpdateOne(ctx context.Context, filter interface{}, upda
return nil, c.Err
}
+func (c *mockCollection) CountDocuments(ctx context.Context, filter interface{},
+ opts ...*options.CountOptions) (int64, error) {
+ return 1, c.Err
+}
+
func TestCreate(t *testing.T) {
testCases := []struct {
label string