mirror of https://github.com/helm/helm
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
3.7 KiB
112 lines
3.7 KiB
/*
|
|
Copyright The Helm Authors.
|
|
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 getter
|
|
|
|
import (
|
|
"net/http"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"helm.sh/helm/v4/pkg/registry"
|
|
)
|
|
|
|
func TestOCIGetter(t *testing.T) {
|
|
g, err := NewOCIGetter(WithURL("oci://example.com"))
|
|
require.NoError(t, err)
|
|
|
|
_, ok := g.(*OCIGetter)
|
|
require.True(t, ok, "Expected NewOCIGetter to produce an *OCIGetter")
|
|
|
|
cd := "../../testdata"
|
|
join := filepath.Join
|
|
ca, pub, priv := join(cd, "rootca.crt"), join(cd, "crt.pem"), join(cd, "key.pem")
|
|
timeout := time.Second * 5
|
|
transport := &http.Transport{}
|
|
insecureSkipVerifyTLS := false
|
|
plainHTTP := false
|
|
|
|
// Test with getterOptions
|
|
g, err = NewOCIGetter(
|
|
WithBasicAuth("I", "Am"),
|
|
WithTLSClientConfig(pub, priv, ca),
|
|
WithTimeout(timeout),
|
|
WithTransport(transport),
|
|
WithInsecureSkipVerifyTLS(insecureSkipVerifyTLS),
|
|
WithPlainHTTP(plainHTTP),
|
|
)
|
|
require.NoError(t, err)
|
|
|
|
og, ok := g.(*OCIGetter)
|
|
require.True(t, ok, "expected NewOCIGetter to produce an *OCIGetter")
|
|
|
|
assert.Equal(t, "I", og.opts.username, "Expected NewOCIGetter to contain %q as the username, got %q", "I", og.opts.username)
|
|
|
|
assert.Equal(t, "Am", og.opts.password, "Expected NewOCIGetter to contain %q as the password, got %q", "Am", og.opts.password)
|
|
|
|
assert.Equal(t, pub, og.opts.certFile, "Expected NewOCIGetter to contain %q as the public key file, got %q", pub, og.opts.certFile)
|
|
|
|
assert.Equal(t, priv, og.opts.keyFile, "Expected NewOCIGetter to contain %q as the private key file, got %q", priv, og.opts.keyFile)
|
|
|
|
assert.Equal(t, ca, og.opts.caFile, "Expected NewOCIGetter to contain %q as the CA file, got %q", ca, og.opts.caFile)
|
|
|
|
assert.Equal(t, timeout, og.opts.timeout, "Expected NewOCIGetter to contain %s as Timeout flag, got %s", timeout, og.opts.timeout)
|
|
|
|
assert.Equal(t, transport, og.opts.transport, "Expected NewOCIGetter to contain %p as Transport, got %p", transport, og.opts.transport)
|
|
|
|
assert.Equal(t, plainHTTP, og.opts.plainHTTP, "Expected NewOCIGetter to have plainHTTP as %t, got %t", plainHTTP, og.opts.plainHTTP)
|
|
|
|
assert.Equal(t, insecureSkipVerifyTLS, og.opts.insecureSkipVerifyTLS, "Expected NewOCIGetter to have insecureSkipVerifyTLS as %t, got %t", insecureSkipVerifyTLS, og.opts.insecureSkipVerifyTLS)
|
|
|
|
// Test if setting registryClient is being passed to the ops
|
|
registryClient, err := registry.NewClient()
|
|
require.NoError(t, err)
|
|
|
|
g, err = NewOCIGetter(
|
|
WithRegistryClient(registryClient),
|
|
)
|
|
require.NoError(t, err)
|
|
og, ok = g.(*OCIGetter)
|
|
require.True(t, ok, "expected NewOCIGetter to produce an *OCIGetter")
|
|
|
|
assert.Equal(t, registryClient, og.opts.registryClient, "Expected NewOCIGetter to contain %p as RegistryClient, got %p", registryClient, og.opts.registryClient)
|
|
}
|
|
|
|
func TestOCIHTTPTransportReuse(t *testing.T) {
|
|
g := OCIGetter{}
|
|
|
|
_, err := g.newRegistryClient()
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.NotNil(t, g.transport, "Expected non nil value for transport")
|
|
|
|
transport1 := g.transport
|
|
|
|
_, err = g.newRegistryClient()
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.NotNil(t, g.transport, "Expected non nil value for transport")
|
|
|
|
transport2 := g.transport
|
|
|
|
require.Equal(t, transport2, transport1, "Expected default transport to be reused")
|
|
}
|