Skip to content

Commit 3496960

Browse files
committed
Fix tests for updated APM implementation
Change-Id: Iacd5ffa839ddd9e1da1bbf8904e3cfd8eb4d3f0f
1 parent d77ce02 commit 3496960

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

mongo/retryable_writes_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
package mongo
88

99
import (
10+
"context"
1011
"encoding/json"
1112
"io/ioutil"
1213
"os"
@@ -70,7 +71,7 @@ var retryMonitoredTopologyOnce sync.Once
7071
var retryStartedChan = make(chan *event.CommandStartedEvent, 100)
7172

7273
var retryMonitor = &event.CommandMonitor{
73-
Started: func(cse *event.CommandStartedEvent) {
74+
Started: func(ctx context.Context, cse *event.CommandStartedEvent) {
7475
retryStartedChan <- cse
7576
},
7677
}

mongo/transactions_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ type transError struct {
104104
var transStartedChan = make(chan *event.CommandStartedEvent, 100)
105105

106106
var transMonitor = &event.CommandMonitor{
107-
Started: func(cse *event.CommandStartedEvent) {
107+
Started: func(ctx context.Context, cse *event.CommandStartedEvent) {
108108
//fmt.Printf("STARTED: %v\n", cse)
109109
transStartedChan <- cse
110110
},

0 commit comments

Comments
 (0)