1 files changed, 127 insertions, 0 deletions
diff --git a/pkg/v1/remote/error_roundtrip_test.go b/pkg/v1/remote/error_roundtrip_test.go
new file mode 100644
index 0000000..5b81ee5
--- /dev/null
+++ b/pkg/v1/remote/error_roundtrip_test.go
@@ -0,0 +1,127 @@
+// Copyright 2019 Google LLC All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//    http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package remote_test
+
+import (
+	"errors"
+	"fmt"
+	"log"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"github.com/google/go-containerregistry/pkg/name"
+	"github.com/google/go-containerregistry/pkg/registry"
+	"github.com/google/go-containerregistry/pkg/v1/random"
+	"github.com/google/go-containerregistry/pkg/v1/remote"
+	"github.com/google/go-containerregistry/pkg/v1/remote/transport"
+)
+
+func TestStatusCodeReturned(t *testing.T) {
+	tcs := []struct {
+		Description string
+		Handler     http.Handler
+	}{{
+		Description: "Only returns teapot status",
+		Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			w.WriteHeader(http.StatusTeapot)
+		}),
+	}, {
+		Description: "Handle v2, returns teapot status else",
+		Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			log.Print(r.URL.Path)
+			if r.URL.Path == "/v2/" {
+				return
+			}
+			w.WriteHeader(http.StatusTeapot)
+		}),
+	}}
+
+	for _, tc := range tcs {
+		t.Run(tc.Description, func(t *testing.T) {
+			o := httptest.NewServer(tc.Handler)
+			defer o.Close()
+
+			ref, err := name.NewDigest(strings.TrimPrefix(o.URL+"/foo:@sha256:53b27244ffa2f585799adbfaf79fba5a5af104597751b289c8b235e7b8f7ebf5", "http://"))
+
+			if err != nil {
+				t.Fatalf("Unable to parse digest: %v", err)
+			}
+
+			_, err = remote.Image(ref)
+			var terr *transport.Error
+			if !errors.As(err, &terr) {
+				t.Fatalf("Unable to cast error to transport error: %v", err)
+			}
+			if terr.StatusCode != http.StatusTeapot {
+				t.Errorf("Incorrect status code received, got %v, wanted %v", terr.StatusCode, http.StatusTeapot)
+			}
+		})
+	}
+}
+
+func TestBlobStatusCodeReturned(t *testing.T) {
+	reg := registry.New()
+	rh := httptest.NewServer(reg)
+	defer rh.Close()
+	i, _ := random.Image(1024, 16)
+	tag := strings.TrimPrefix(fmt.Sprintf("%s/foo:bar", rh.URL), "http://")
+	d, _ := name.NewTag(tag)
+	if err := remote.Write(d, i); err != nil {
+		t.Fatalf("Unable to write empty image: %v", err)
+	}
+
+	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		log.Print(r.URL.Path)
+		if strings.Contains(r.URL.Path, "blob") {
+			w.WriteHeader(http.StatusTeapot)
+			return
+		}
+		reg.ServeHTTP(w, r)
+	})
+
+	o := httptest.NewServer(handler)
+	defer o.Close()
+
+	ref, err := name.NewTag(strings.TrimPrefix(fmt.Sprintf("%s/foo:bar", o.URL), "http://"))
+	if err != nil {
+		t.Fatalf("Unable to parse digest: %v", err)
+	}
+
+	ri, err := remote.Image(ref)
+	if err != nil {
+		t.Fatalf("Unable to fetch manifest: %v", err)
+	}
+	l, err := ri.Layers()
+	if err != nil {
+		t.Fatalf("Unable to fetch layers: %v", err)
+	}
+	_, err = l[0].Compressed()
+	var terr *transport.Error
+	if !errors.As(err, &terr) {
+		t.Fatalf("Unable to cast error to transport error: %v", err)
+	}
+	if terr.StatusCode != http.StatusTeapot {
+		t.Errorf("Incorrect status code received, got %v, wanted %v", terr.StatusCode, http.StatusTeapot)
+	}
+	_, err = l[0].Uncompressed()
+	if !errors.As(err, &terr) {
+		t.Fatalf("Unable to cast error to transport error: %v", err)
+	}
+	if terr.StatusCode != http.StatusTeapot {
+		t.Errorf("Incorrect status code received, got %v, wanted %v", terr.StatusCode, http.StatusTeapot)
+	}
+}