From 54e375c62e50c14b178be1fbac79dfef0363ec1b Mon Sep 17 00:00:00 2001
From: Ying Li <ying.li@docker.com>
Date: Fri, 13 Nov 2015 04:18:50 -0800
Subject: [PATCH] Add tests to ensure that the TUF httpstore returns the right
 error on 5XX.

Signed-off-by: Ying Li <ying.li@docker.com>
---
 tuf/store/httpstore_test.go | 43 +++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/tuf/store/httpstore_test.go b/tuf/store/httpstore_test.go
index 512c6b3612..6d1c689acc 100644
--- a/tuf/store/httpstore_test.go
+++ b/tuf/store/httpstore_test.go
@@ -3,6 +3,7 @@ package store
 import (
 	"encoding/base64"
 	"encoding/hex"
+	"fmt"
 	"io"
 	"io/ioutil"
 	"net/http"
@@ -32,6 +33,7 @@ func TestHTTPStoreGetMeta(t *testing.T) {
 		w.Write([]byte(testRoot))
 	}
 	server := httptest.NewServer(http.HandlerFunc(handler))
+	defer server.Close()
 	store, err := NewHTTPStore(
 		server.URL,
 		"metadata",
@@ -152,3 +154,44 @@ func TestPyNaCled25519Compat(t *testing.T) {
 		t.Fatal(err)
 	}
 }
+
+func testErrorCode(t *testing.T, errorCode int, errType error) {
+	handler := func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(errorCode)
+	}
+	server := httptest.NewServer(http.HandlerFunc(handler))
+	defer server.Close()
+
+	store, err := NewHTTPStore(
+		server.URL,
+		"metadata",
+		"txt",
+		"targets",
+		"key",
+		&http.Transport{},
+	)
+	assert.NoError(t, err)
+
+	_, err = store.GetMeta("root", 4801)
+	assert.Error(t, err)
+	assert.IsType(t, errType, err,
+		fmt.Sprintf("%d should translate to %v", errorCode, errType))
+}
+
+func TestErrMetadataNotFound(t *testing.T) {
+	testErrorCode(t, http.StatusNotFound, ErrMetaNotFound{})
+}
+
+func Test500Errors(t *testing.T) {
+	fiveHundreds := []int{
+		http.StatusInternalServerError,
+		http.StatusNotImplemented,
+		http.StatusBadGateway,
+		http.StatusServiceUnavailable,
+		http.StatusGatewayTimeout,
+		http.StatusHTTPVersionNotSupported,
+	}
+	for _, code := range fiveHundreds {
+		testErrorCode(t, code, ErrServerUnavailable{})
+	}
+}