Function TestPreadRemoteUpstream

pkg/discovery/content/reader/reader_test.go:26–96

(t *testing.T)

)

func TestPreadRemoteUpstream(t *testing.T) {
	// Setup
	m := map[string][]string{}
	key := "somekey"
	expected := "expected-result"
	peersTried := 0
	svr3 := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		peersTried++
		w.WriteHeader(http.StatusUnauthorized)
	}))
	defer svr3.Close()
	svr2 := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		peersTried++
		w.WriteHeader(http.StatusNotFound)
	}))
	defer svr2.Close()
	svr1 := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		peersTried++
		w.WriteHeader(http.StatusBadGateway)
	}))
	defer svr1.Close()
	val := []string{svr1.URL, svr2.URL, svr3.URL}
	m[key] = val

	svr := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if "?"+r.URL.RawQuery == query {
			w.Header().Set("Content-Type", "application/octet-stream")
			// nolint:errcheck
			w.Write([]byte(expected))
		} else {
			w.WriteHeader(http.StatusNotFound)
		}
	}))
	defer svr.Close()
	p := svr.URL + "/some-path"
	u := "http://127.0.0.1:5000/blobs/" + p + query
	req, err := http.NewRequest("GET", u, nil)
	if err != nil {
		t.Fatal(err)
	}

	router := mocks.NewMockRouter(m)
	c, _ := gin.CreateTestContext(httptest.NewRecorder())
	c.Request = req
	c.Params = []gin.Param{
		{Key: "url", Value: p},
	}

	pc := pcontext.FromContext(c)

	pc.Set(pcontext.BlobUrlCtxKey, pcontext.BlobUrl(pc))
	pc.Set(pcontext.BlobRangeCtxKey, "bytes=0-10")
	pc.Set(pcontext.FileChunkCtxKey, key)

	r := NewReader(pc, router, 3, 500*time.Millisecond, mr).(*reader)
	b := make([]byte, 10)

	// Test