144 lines
3.2 KiB
Go
144 lines
3.2 KiB
Go
/*
|
|
* Copyright 2024 the KubeSphere Authors.
|
|
* Please refer to the LICENSE file in the root directory of the project.
|
|
* https://github.com/kubesphere/kubesphere/blob/master/LICENSE
|
|
*/
|
|
|
|
package v2
|
|
|
|
import (
|
|
"testing"
|
|
|
|
corev1 "k8s.io/api/core/v1"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
v1 "github.com/google/go-containerregistry/pkg/v1"
|
|
)
|
|
|
|
func TestRegistryerConfig(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
secret *corev1.Secret
|
|
image string
|
|
configFile *v1.ConfigFile
|
|
expectErr bool
|
|
}{
|
|
{
|
|
name: "Should fetch image config with public registry",
|
|
secret: nil,
|
|
image: "kubesphere/ks-apiserver:v3.1.0",
|
|
expectErr: false,
|
|
configFile: &v1.ConfigFile{
|
|
Config: v1.Config{
|
|
Image: "sha256:a3006bafb7702494227f7fa69720e65e1324b7bf8ece8ac03d9fe1d0134e7341",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, testCase := range testCases {
|
|
|
|
t.Run(testCase.name, func(t *testing.T) {
|
|
secretAuthenticator, err := NewSecretAuthenticator(testCase.secret)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
// fix platform to linux/amd64 so we could compare config
|
|
platform := &v1.Platform{
|
|
OS: "linux",
|
|
Architecture: "amd64",
|
|
}
|
|
|
|
options := secretAuthenticator.Options()
|
|
options = append(options, WithPlatform(platform))
|
|
|
|
registryer := NewRegistryer(options...)
|
|
|
|
config, err := registryer.Config(testCase.image)
|
|
if testCase.expectErr && err == nil {
|
|
t.Errorf("expected error, but got nil")
|
|
}
|
|
|
|
if !testCase.expectErr && err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if diff := cmp.Diff(testCase.configFile.Config.Image, config.Config.Image); len(diff) != 0 {
|
|
t.Errorf("expected %v, but got %v", testCase.configFile.Config.Image, config.Config.Image)
|
|
}
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
func TestRegistryerListRepoTags(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
secret *corev1.Secret
|
|
image string
|
|
repositoryTags RepositoryTags
|
|
expectErr bool
|
|
}{
|
|
{
|
|
name: "Should fetch config with public registry",
|
|
secret: nil,
|
|
image: "kubesphere/ks-apiserver",
|
|
expectErr: false,
|
|
repositoryTags: RepositoryTags{
|
|
Registry: "index.docker.io",
|
|
Tags: []string{
|
|
"v3.1.1",
|
|
"v3.1.0",
|
|
"latest",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, testCase := range testCases {
|
|
|
|
t.Run(testCase.name, func(t *testing.T) {
|
|
secretAuthenticator, err := NewSecretAuthenticator(testCase.secret)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
// fix platform to linux/amd64 so we could compare config
|
|
platform := &v1.Platform{
|
|
OS: "linux",
|
|
Architecture: "amd64",
|
|
}
|
|
|
|
options := secretAuthenticator.Options()
|
|
options = append(options, WithPlatform(platform))
|
|
|
|
registryer := NewRegistryer(options...)
|
|
|
|
tags, err := registryer.ListRepositoryTags(testCase.image)
|
|
if testCase.expectErr && err == nil {
|
|
t.Errorf("expected error, but got nil")
|
|
}
|
|
|
|
if !testCase.expectErr && err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
cotains := func(s []string, e string) bool {
|
|
for _, a := range s {
|
|
if a == e {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
for _, tag := range testCase.repositoryTags.Tags {
|
|
if !cotains(tags.Tags, tag) {
|
|
t.Errorf("no expected tag %s in result %v", tag, tags.Tags)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|