Function TestHandlePeerHasAndGet

cmd/cache-proxy/proxy_test.go:759–795

(t *testing.T)

	if rec.Code != http.StatusNotFound {
		t.Fatalf("status = %d, want 404 forwarded verbatim", rec.Code)
	}
}

// TestHandleProxyForwardsOrigin416Verbatim: Range Not Satisfiable carries
// semantically important metadata for DuckLake; collapsing to 502 made it
// look like a network error.
func TestHandleProxyForwardsOrigin416Verbatim(t *testing.T) {
	proxy := newTestProxy(t)
	_, originURL := newTestServer(t, func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Range", "bytes */1024")
		w.WriteHeader(http.StatusRequestedRangeNotSatisfiable)
	})
	rec := doForwardProxyRequest(proxy, "GET", originURL+"/bucket/short.parquet", http.Header{"Range": []string{"bytes=999999-1000000"}})
	if rec.Code != http.StatusRequestedRangeNotSatisfiable {
		t.Fatalf("status = %d, want 416 forwarded verbatim", rec.Code)
	}
	if cr := rec.Header().Get("Content-Range"); cr != "bytes */1024" {
		t.Errorf("Content-Range = %q, want 'bytes */1024' (DuckDB uses this to learn the actual file size)", cr)
	}
}

// TestHandleProxyDoesNotCacheErrorResponses: a transient origin error
// must not poison the cache. The next request for the same key has to hit
// the origin again — otherwise an ExpiredToken error would persist past
// the credential refresh that fixes it.
func TestHandleProxyDoesNotCacheErrorResponses(t *testing.T) {
	proxy := newTestProxy(t)
	var calls int32
	_, originURL := newTestServer(t, func(w http.ResponseWriter, r *http.Request) {
		n := atomic.AddInt32(&calls, 1)
		if n == 1 {
			w.WriteHeader(http.StatusBadRequest)
			_, _ = w.Write([]byte("first call fails"))
			return
		}
		w.Header().Set("Content-Type", "application/octet-stream")
		w.WriteHeader(http.StatusOK)
		_, _ = w.Write([]byte("ok-now"))
	})