diff options
Diffstat (limited to 'internal/pkg/net_test.go')
| -rw-r--r-- | internal/pkg/net_test.go | 16 |
1 files changed, 13 insertions, 3 deletions
diff --git a/internal/pkg/net_test.go b/internal/pkg/net_test.go index 9bf58d92..c7c49061 100644 --- a/internal/pkg/net_test.go +++ b/internal/pkg/net_test.go @@ -2,6 +2,7 @@ package pkg_test import ( "crypto/sha512" + "io" "net/http" "reflect" "testing" @@ -31,13 +32,17 @@ func TestHTTPGet(t *testing.T) { checkWithCache(t, []cacheTestCase{ {"direct", nil, func(t *testing.T, base *check.Absolute, c *pkg.Cache) { + f := pkg.NewHTTPGet( &client, "file:///testdata", testdataChecksum.Value(), ) - if got, err := f.Cure(t.Context()); err != nil { + var got []byte + if r, err := f.Cure(t.Context()); err != nil { t.Fatalf("Cure: error = %v", err) + } else if got, err = io.ReadAll(r); err != nil { + t.Fatalf("ReadAll: error = %v", err) } else if string(got) != testdata { t.Fatalf("Cure: %x, want %x", got, testdata) } @@ -85,8 +90,11 @@ func TestHTTPGet(t *testing.T) { t.Fatalf("Cure: %x, want %x", checksum.Value(), testdataChecksum.Value()) } - if got, err := f.Cure(t.Context()); err != nil { + var got []byte + if r, err := f.Cure(t.Context()); err != nil { t.Fatalf("Cure: error = %v", err) + } else if got, err = io.ReadAll(r); err != nil { + t.Fatalf("ReadAll: error = %v", err) } else if string(got) != testdata { t.Fatalf("Cure: %x, want %x", got, testdata) } @@ -97,8 +105,10 @@ func TestHTTPGet(t *testing.T) { "file:///testdata", testdataChecksum.Value(), ) - if got, err := f.Cure(t.Context()); err != nil { + if r, err := f.Cure(t.Context()); err != nil { t.Fatalf("Cure: error = %v", err) + } else if got, err = io.ReadAll(r); err != nil { + t.Fatalf("ReadAll: error = %v", err) } else if string(got) != testdata { t.Fatalf("Cure: %x, want %x", got, testdata) } |