client: add context arg
Add a context.Context argument to methods that might involve network
traffic.
Change-Id: Ib743a7b1a8c80d09d16c4529f5d8822726e8b122
Reviewed-on: https://go-review.googlesource.com/c/vuln/+/365054
Trust: Jonathan Amsterdam <jba@google.com>
Run-TryBot: Jonathan Amsterdam <jba@google.com>
TryBot-Result: Go Bot <gobot@golang.org>
Reviewed-by: Julie Qiu <julie@golang.org>
diff --git a/client/client_test.go b/client/client_test.go
index b959764..9b2ca26 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -5,6 +5,7 @@
package client
import (
+ "context"
"encoding/json"
"fmt"
"io"
@@ -157,7 +158,7 @@
if runtime.GOOS == "js" {
t.Skip("skipping test: no network on js")
}
-
+ ctx := context.Background()
// Create a local http database.
srv := newTestServer()
defer srv.Close()
@@ -198,7 +199,7 @@
t.Fatal(err)
}
- vulns, err := client.GetByModule("golang.org/example/one")
+ vulns, err := client.GetByModule(ctx, "golang.org/example/one")
if err != nil {
t.Fatal(err)
}
@@ -231,7 +232,7 @@
hs := &httpSource{url: ts.URL, c: new(http.Client)}
for _, module := range []string{"a", "b", "c"} {
- if _, err := hs.GetByModule(module); err != nil {
+ if _, err := hs.GetByModule(context.Background(), module); err != nil {
t.Fatalf("unexpected error: %s", err)
}
}
@@ -269,7 +270,7 @@
if err != nil {
t.Fatal(err)
}
- vulns, err := client.GetByModule("a")
+ vulns, err := client.GetByModule(context.Background(), "a")
if err != nil {
t.Fatal(err)
}
@@ -309,7 +310,7 @@
if err != nil {
t.Fatal(err)
}
- got, err := client.GetByID("ID")
+ got, err := client.GetByID(context.Background(), "ID")
if err != nil {
t.Fatal(err)
}
@@ -348,7 +349,7 @@
if err != nil {
t.Fatal(err)
}
- got, err := client.ListIDs()
+ got, err := client.ListIDs(context.Background())
if err != nil {
t.Fatal(err)
}
