62c117421a
- I realized that the hardcoded fakekubeapi 404 not found response was invalid, so we were getting a default error message. I fixed it so the tests follow a higher fidelity code path. - I caved and added a test for making sure the request body was always closed, and believe it or not, we were double closing a body. I don't *think* this will matter in production, since client-go will pass us ioutil.NopReader()'s, but at least we know now. Signed-off-by: Andrew Keesler <akeesler@vmware.com>
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
// Copyright 2021 the Pinniped contributors. All Rights Reserved.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package kubeclient
|
|
|
|
import (
|
|
restclient "k8s.io/client-go/rest"
|
|
"k8s.io/client-go/transport"
|
|
)
|
|
|
|
type Option func(*clientConfig)
|
|
|
|
type clientConfig struct {
|
|
config *restclient.Config
|
|
middlewares []Middleware
|
|
transportWrapper transport.WrapperFunc
|
|
}
|
|
|
|
func WithConfig(config *restclient.Config) Option {
|
|
return func(c *clientConfig) {
|
|
c.config = config
|
|
}
|
|
}
|
|
|
|
func WithMiddleware(middleware Middleware) Option {
|
|
return func(c *clientConfig) {
|
|
if middleware == nil {
|
|
return // support passing in a nil middleware as a no-op
|
|
}
|
|
|
|
c.middlewares = append(c.middlewares, middleware)
|
|
}
|
|
}
|
|
|
|
// WithTransportWrapper will wrap the client-go http.RoundTripper chain *after* the middleware
|
|
// wrapper is applied. I.e., this wrapper has the opportunity to supply an http.RoundTripper that
|
|
// runs first in the client-go http.RoundTripper chain.
|
|
func WithTransportWrapper(wrapper transport.WrapperFunc) Option {
|
|
return func(c *clientConfig) {
|
|
c.transportWrapper = wrapper
|
|
}
|
|
}
|