primal/athens
1
0
Fork 0
mirror of https://github.com/gomods/athens synced 2026-02-03 11:00:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

download: expand redirect tests (#1287)

Browse Source
This commit is contained in:
Marwan Sulaiman
2019-06-18 14:22:03 -04:00
committed by Aaron Schlesinger
parent d138a42599
commit dea04b175f
1 changed files with 27 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/download/handler_test.go
+27 -10
View File
@@ -2,6 +2,7 @@ package download
import (
"context"
"io"
"net/http"
"net/http/httptest"
"testing"
@@ -22,16 +23,22 @@ func TestRedirect(t *testing.T) {
DownloadURL: "https://gomods.io",
},
})
req := httptest.NewRequest("GET", "/github.com/gomods/athens/@v/v0.4.0.info", nil)
w := httptest.NewRecorder()
r.ServeHTTP(w, req)
if w.Code != http.StatusMovedPermanently {
t.Fatalf("expected a redirect status (301) but got %v", w.Code)
}
expectedRedirect := "https://gomods.io/github.com/gomods/athens/@v/v0.4.0.info"
givenRedirect := w.HeaderMap.Get("location")
if expectedRedirect != givenRedirect {
t.Fatalf("expected the handler to redirect to %q but got %q", expectedRedirect, givenRedirect)
for _, path := range [...]string{
"/github.com/gomods/athens/@v/v0.4.0.info",
"/github.com/gomods/athens/@v/v0.4.0.mod",
"/github.com/gomods/athens/@v/v0.4.0.zip",
} {
req := httptest.NewRequest("GET", path, nil)
w := httptest.NewRecorder()
r.ServeHTTP(w, req)
if w.Code != http.StatusMovedPermanently {
t.Fatalf("expected a redirect status (301) but got %v", w.Code)
}
expectedRedirect := "https://gomods.io" + path
givenRedirect := w.HeaderMap.Get("location")
if expectedRedirect != givenRedirect {
t.Fatalf("expected the handler to redirect to %q but got %q", expectedRedirect, givenRedirect)
}
}
}
@@ -43,3 +50,13 @@ func (mp *mockProtocol) Info(ctx context.Context, mod, ver string) ([]byte, erro
const op errors.Op = "mockProtocol.Info"
return nil, errors.E(op, "not found", errors.KindRedirect)
}
func (mp *mockProtocol) GoMod(ctx context.Context, mod, ver string) ([]byte, error) {
const op errors.Op = "mockProtocol.GoMod"
return nil, errors.E(op, "not found", errors.KindRedirect)
}
func (mp *mockProtocol) Zip(ctx context.Context, mod, ver string) (io.ReadCloser, error) {
const op errors.Op = "mockProtocol.Zip"
return nil, errors.E(op, "not found", errors.KindRedirect)
}