github/client-go
1
0
Fork 0
mirror of https://github.com/kubernetes/client-go.git synced 2025-06-25 06:31:35 +00:00
Code Issues Projects Releases Wiki Activity

Tolerate 406 mime-type errors attempting to load new openapi schema

Browse Source 
Kubernetes-commit: 51fb861b42dbac5b0037e525c4b8bfd9db48675c
This commit is contained in:
Jordan Liggitt 2018-03-30 14:20:35 -04:00 committed by Kubernetes Publisher
parent 8c732d02e9
commit 13a6e8222e
2 changed files with 40 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
discovery/discovery_client.go
View File

@ -335,7 +335,7 @@ func (d *DiscoveryClient) ServerVersion() (*version.Info, error) {
func (d *DiscoveryClient) OpenAPISchema() (*openapi_v2.Document, error) { func (d *DiscoveryClient) OpenAPISchema() (*openapi_v2.Document, error) {
data, err := d.restClient.Get().AbsPath("/openapi/v2").SetHeader("Accept", mimePb).Do().Raw() data, err := d.restClient.Get().AbsPath("/openapi/v2").SetHeader("Accept", mimePb).Do().Raw()
if err != nil { if err != nil {
if errors.IsForbidden(err) || errors.IsNotFound(err) { if errors.IsForbidden(err) || errors.IsNotFound(err) || errors.IsNotAcceptable(err) {
// single endpoint not found/registered in old server, try to fetch old endpoint // single endpoint not found/registered in old server, try to fetch old endpoint
// TODO(roycaihw): remove this in 1.11 // TODO(roycaihw): remove this in 1.11
data, err = d.restClient.Get().AbsPath("/swagger-2.0.0.pb-v1").Do().Raw() data, err = d.restClient.Get().AbsPath("/swagger-2.0.0.pb-v1").Do().Raw()

44
discovery/discovery_client_test.go
View File

@ -326,12 +326,12 @@ var returnedOpenAPI = openapi_v2.Document{
}, },
} }
func openapiSchemaDeprecatedFakeServer() (*httptest.Server, error) { func openapiSchemaDeprecatedFakeServer(status int) (*httptest.Server, error) {
var sErr error var sErr error
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
// old server returns 403 on new endpoint request
if req.URL.Path == "/openapi/v2" { if req.URL.Path == "/openapi/v2" {
w.WriteHeader(http.StatusForbidden) // write the error status for the new endpoint request
w.WriteHeader(status)
return return
} }
if req.URL.Path != "/swagger-2.0.0.pb-v1" { if req.URL.Path != "/swagger-2.0.0.pb-v1" {
@ -398,8 +398,42 @@ func TestGetOpenAPISchema(t *testing.T) {
} }
} }
func TestGetOpenAPISchemaFallback(t *testing.T) { func TestGetOpenAPISchemaForbiddenFallback(t *testing.T) {
server, err := openapiSchemaDeprecatedFakeServer() server, err := openapiSchemaDeprecatedFakeServer(http.StatusForbidden)
if err != nil {
t.Errorf("unexpected error starting fake server: %v", err)
}
defer server.Close()
client := NewDiscoveryClientForConfigOrDie(&restclient.Config{Host: server.URL})
got, err := client.OpenAPISchema()
if err != nil {
t.Fatalf("unexpected error getting openapi: %v", err)
}
if e, a := returnedOpenAPI, *got; !reflect.DeepEqual(e, a) {
t.Errorf("expected %v, got %v", e, a)
}
}
func TestGetOpenAPISchemaNotFoundFallback(t *testing.T) {
server, err := openapiSchemaDeprecatedFakeServer(http.StatusNotFound)
if err != nil {
t.Errorf("unexpected error starting fake server: %v", err)
}
defer server.Close()
client := NewDiscoveryClientForConfigOrDie(&restclient.Config{Host: server.URL})
got, err := client.OpenAPISchema()
if err != nil {
t.Fatalf("unexpected error getting openapi: %v", err)
}
if e, a := returnedOpenAPI, *got; !reflect.DeepEqual(e, a) {
t.Errorf("expected %v, got %v", e, a)
}
}
func TestGetOpenAPISchemaNotAcceptableFallback(t *testing.T) {
server, err := openapiSchemaDeprecatedFakeServer(http.StatusNotAcceptable)
if err != nil { if err != nil {
t.Errorf("unexpected error starting fake server: %v", err) t.Errorf("unexpected error starting fake server: %v", err)
} }