mirror of
https://github.com/rancher/k3k.git
synced 2026-05-07 01:46:43 +00:00
ba1648e560
* first commit * refactor: update golangci configuration and reorder imports in namespace.go * refactor: update golangci configuration for linters and formatters * check restart for logs fetch * rverted a bit the structure * requested changes * WriteLogs rename
107 lines
2.6 KiB
Go
107 lines
2.6 KiB
Go
package cli_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/testcontainers/testcontainers-go"
|
|
"github.com/testcontainers/testcontainers-go/modules/k3s"
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/rest"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
|
|
fwclient "github.com/rancher/k3k/tests/framework/client"
|
|
fwcontainer "github.com/rancher/k3k/tests/framework/container"
|
|
fwlog "github.com/rancher/k3k/tests/framework/log"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
const (
|
|
k3kNamespace = "k3k-system"
|
|
|
|
k3sVersion = "v1.35.2-k3s1"
|
|
k3sOldVersion = "v1.35.0-k3s1"
|
|
)
|
|
|
|
func TestTests(t *testing.T) {
|
|
RegisterFailHandler(Fail)
|
|
RunSpecs(t, "Tests Suite")
|
|
}
|
|
|
|
var (
|
|
k3sContainer *k3s.K3sContainer
|
|
restcfg *rest.Config
|
|
k8s *kubernetes.Clientset
|
|
k8sClient client.Client
|
|
kubeconfigPath string
|
|
)
|
|
|
|
var _ = BeforeSuite(func() {
|
|
ctx := context.Background()
|
|
|
|
_, dockerInstallEnabled := os.LookupEnv("K3K_DOCKER_INSTALL")
|
|
|
|
if dockerInstallEnabled {
|
|
repo := os.Getenv("REPO")
|
|
if repo == "" {
|
|
repo = "rancher"
|
|
}
|
|
|
|
installK3SDocker(ctx, repo+"/k3k", repo+"/k3k-kubelet")
|
|
initKubernetesClient(ctx)
|
|
installK3kChart(repo+"/k3k", repo+"/k3k-kubelet")
|
|
} else {
|
|
initKubernetesClient(ctx)
|
|
}
|
|
})
|
|
|
|
func installK3SDocker(ctx context.Context, controllerImage, kubeletImage string) {
|
|
k3sHostVersion := os.Getenv("K3S_HOST_VERSION")
|
|
if k3sHostVersion == "" {
|
|
k3sHostVersion = k3sVersion
|
|
}
|
|
|
|
k3sContainer, kubeconfigPath = fwcontainer.SetupK3s(ctx, k3sHostVersion, controllerImage, kubeletImage)
|
|
}
|
|
|
|
func initKubernetesClient(ctx context.Context) {
|
|
scheme := fwclient.NewScheme()
|
|
config, err := fwclient.InitFromKubeconfig(ctx, scheme, k3sContainer)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
restcfg = config.RestConfig
|
|
k8s = config.Clientset
|
|
k8sClient = config.Client
|
|
}
|
|
|
|
func installK3kChart(controllerImage, kubeletImage string) {
|
|
installer := fwcontainer.NewHelmInstaller(controllerImage, kubeletImage, kubeconfigPath)
|
|
kubeconfig, err := os.ReadFile(kubeconfigPath)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
restClientGetter, err := fwclient.NewRESTClientGetter(kubeconfig)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
installer.InstallK3kChart(restClientGetter)
|
|
}
|
|
|
|
var _ = AfterSuite(func() {
|
|
ctx := context.Background()
|
|
|
|
if k3sContainer != nil {
|
|
// dump k3s logs
|
|
k3sLogs, err := k3sContainer.Logs(ctx)
|
|
Expect(err).To(Not(HaveOccurred()))
|
|
fwlog.WriteLogs("k3s.log", k3sLogs)
|
|
|
|
// dump k3k controller logs
|
|
k3kLogs := fwlog.GetK3kPodLogs(ctx, k8sClient, k8s, k3kNamespace)
|
|
fwlog.WriteLogs("k3k.log", k3kLogs)
|
|
|
|
testcontainers.CleanupContainer(GinkgoTB(), k3sContainer)
|
|
}
|
|
})
|