internal/pkg: pass context in request wrapper

This method is for the most common use case, and in actual use there will always be an associated context.

Signed-off-by: Ophestra <cat@gensokyo.uk>

1 files changed, 42 insertions, 7 deletions

diff --git a/internal/pkg/net_test.go b/internal/pkg/net_test.go
index 016d0de8..dbd956a1 100644
--- a/internal/pkg/net_test.go
+++ b/internal/pkg/net_test.go
@@ -34,7 +34,12 @@ func TestHTTP(t *testing.T) {
 	checkWithCache(t, []cacheTestCase{
 		{"direct", nil, func(t *testing.T, base *check.Absolute, c *pkg.Cache) {
 			var got []byte
-			if f, err := c.NewHTTPGet(&client, "file:///testdata", testdataChecksum); err != nil {
+			if f, err := c.NewHTTPGet(
+				t.Context(),
+				&client,
+				"file:///testdata",
+				testdataChecksum,
+			); err != nil {
 				t.Fatalf("NewHTTPGet: error = %v", err)
 			} else if got, err = f.Data(); err != nil {
 				t.Fatalf("Data: error = %v", err)
@@ -48,7 +53,12 @@ func TestHTTP(t *testing.T) {
 			wantErrMismatch := &pkg.ChecksumMismatchError{
 				Got: testdataChecksum,
 			}
-			if f, err := c.NewHTTPGet(&client, "file:///testdata", pkg.Checksum{}); err != nil {
+			if f, err := c.NewHTTPGet(
+				t.Context(),
+				&client,
+				"file:///testdata",
+				pkg.Checksum{},
+			); err != nil {
 				t.Fatalf("NewHTTPGet: error = %v", err)
 			} else if _, err = f.Data(); !reflect.DeepEqual(err, wantErrMismatch) {
 				t.Fatalf("Data: error = %#v, want %#v", err, wantErrMismatch)
@@ -58,7 +68,12 @@ func TestHTTP(t *testing.T) {
 
 			// check direct response error
 			wantErrNotFound := pkg.ResponseStatusError(http.StatusNotFound)
-			if f, err := c.NewHTTPGet(&client, "file:///nonexistent", pkg.Checksum{}); err != nil {
+			if f, err := c.NewHTTPGet(
+				t.Context(),
+				&client,
+				"file:///nonexistent",
+				pkg.Checksum{},
+			); err != nil {
 				t.Fatalf("NewHTTPGet: error = %v", err)
 			} else if _, err = f.Data(); !reflect.DeepEqual(err, wantErrNotFound) {
 				t.Fatalf("Data: error = %#v, want %#v", err, wantErrNotFound)
@@ -68,7 +83,12 @@ func TestHTTP(t *testing.T) {
 		}, pkg.MustDecode("E4vEZKhCcL2gPZ2Tt59FS3lDng-d_2SKa2i5G_RbDfwGn6EemptFaGLPUDiOa94C")},
 
 		{"load or store", nil, func(t *testing.T, base *check.Absolute, c *pkg.Cache) {
-			f, err := c.NewHTTPGet(&client, "file:///testdata", testdataChecksum)
+			f, err := c.NewHTTPGet(
+				t.Context(),
+				&client,
+				"file:///testdata",
+				testdataChecksum,
+			)
 			if err != nil {
 				t.Fatalf("NewHTTPGet: error = %v", err)
 			}
@@ -101,7 +121,12 @@ func TestHTTP(t *testing.T) {
 			}
 
 			// check load from cache
-			if f, err = c.NewHTTPGet(&client, "file:///testdata", testdataChecksum); err != nil {
+			if f, err = c.NewHTTPGet(
+				t.Context(),
+				&client,
+				"file:///testdata",
+				testdataChecksum,
+			); err != nil {
 				t.Fatalf("NewHTTPGet: error = %v", err)
 			} else if got, err = f.Data(); err != nil {
 				t.Fatalf("Data: error = %v", err)
@@ -113,7 +138,12 @@ func TestHTTP(t *testing.T) {
 
 			// check error passthrough
 			wantErrNotFound := pkg.ResponseStatusError(http.StatusNotFound)
-			if f, err = c.NewHTTPGet(&client, "file:///nonexistent", pkg.Checksum{}); err != nil {
+			if f, err = c.NewHTTPGet(
+				t.Context(),
+				&client,
+				"file:///nonexistent",
+				pkg.Checksum{},
+			); err != nil {
 				t.Fatalf("NewHTTPGet: error = %v", err)
 			} else if _, err = f.Pathname(); !reflect.DeepEqual(err, wantErrNotFound) {
 				t.Fatalf("Pathname: error = %#v, want %#v", err, wantErrNotFound)
@@ -132,7 +162,12 @@ func TestHTTP(t *testing.T) {
 				"identifier",
 				testdataChecksumString,
 			)
-			if f, err := c.NewHTTPGet(&client, "file:///testdata", testdataChecksum); err != nil {
+			if f, err := c.NewHTTPGet(
+				t.Context(),
+				&client,
+				"file:///testdata",
+				testdataChecksum,
+			); err != nil {
 				t.Fatalf("NewHTTPGet: error = %v", err)
 			} else if got, err = f.Data(); err != nil {
 				t.Fatalf("Data: error = %v", err)