package registry
 

 import (
 	"net/http"
 	"net/http/httptest"
 	"net/url"
 	"testing"
 )

 
 func TestEndpointParse(t *testing.T) {
 	testData := []struct {
 		str      string
 		expected string
 	}{

 		{IndexServer, IndexServer},

 		{"http://0.0.0.0:5000/v1/", "http://0.0.0.0:5000/v1/"},

 		{"http://0.0.0.0:5000", "http://0.0.0.0:5000/v1/"},
 		{"0.0.0.0:5000", "https://0.0.0.0:5000/v1/"},
 		{"http://0.0.0.0:5000/nonversion/", "http://0.0.0.0:5000/nonversion/v1/"},
 		{"http://0.0.0.0:5000/v0/", "http://0.0.0.0:5000/v0/v1/"},

 	}
 	for _, td := range testData {

 		e, err := newV1EndpointFromStr(td.str, nil, "", nil)

 		if err != nil {
 			t.Errorf("%q: %s", td.str, err)
 		}
 		if e == nil {
 			t.Logf("something's fishy, endpoint for %q is nil", td.str)
 			continue
 		}
 		if e.String() != td.expected {
 			t.Errorf("expected %q, got %q", td.expected, e.String())
 		}
 	}
 }

 func TestEndpointParseInvalid(t *testing.T) {
 	testData := []string{
 		"http://0.0.0.0:5000/v2/",
 	}
 	for _, td := range testData {
 		e, err := newV1EndpointFromStr(td, nil, "", nil)
 		if err == nil {
 			t.Errorf("expected error parsing %q: parsed as %q", td, e)
 		}
 	}
 }
 

 // Ensure that a registry endpoint that responds with a 401 only is determined

 // to be a valid v1 registry endpoint
 func TestValidateEndpoint(t *testing.T) {

 	requireBasicAuthHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		w.Header().Add("WWW-Authenticate", `Basic realm="localhost"`)
 		w.WriteHeader(http.StatusUnauthorized)
 	})
 
 	// Make a test server which should validate as a v1 server.
 	testServer := httptest.NewServer(requireBasicAuthHandler)
 	defer testServer.Close()
 
 	testServerURL, err := url.Parse(testServer.URL)
 	if err != nil {
 		t.Fatal(err)
 	}
 

 	testEndpoint := V1Endpoint{
 		URL:    testServerURL,
 		client: HTTPClient(NewTransport(nil)),

 	}
 
 	if err = validateEndpoint(&testEndpoint); err != nil {
 		t.Fatal(err)
 	}
 

 	if testEndpoint.URL.Scheme != "http" {
 		t.Fatalf("expecting to validate endpoint as http, got url %s", testEndpoint.String())

 	}
 }