mirror of
https://github.com/rancher/k3k.git
synced 2026-08-19 04:16:16 +00:00
64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
package k3k_test
|
|
|
|
import (
|
|
"time"
|
|
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
|
|
fwk3k "github.com/rancher/k3k/tests/framework/k3k"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = When("creating a shared mode cluster", Label(e2eTestLabel), Label(slowTestsLabel), func() {
|
|
var virtualCluster *VirtualCluster
|
|
|
|
BeforeEach(func() {
|
|
namespace := fwk3k.CreateNamespace(k8s)
|
|
|
|
DeferCleanup(func() {
|
|
fwk3k.DeleteNamespaces(k8s, namespace.Name)
|
|
})
|
|
|
|
cluster := NewCluster(namespace.Name)
|
|
CreateCluster(cluster)
|
|
client, restConfig := NewVirtualK8sClientAndConfig(cluster)
|
|
|
|
virtualCluster = &VirtualCluster{
|
|
Cluster: cluster,
|
|
RestConfig: restConfig,
|
|
Client: client,
|
|
}
|
|
})
|
|
|
|
It("creates nodes with the worker role", func() {
|
|
Eventually(func(g Gomega) {
|
|
ctx := GinkgoT().Context()
|
|
|
|
nodes, err := virtualCluster.Client.CoreV1().Nodes().List(ctx, metav1.ListOptions{})
|
|
g.Expect(err).To(Not(HaveOccurred()))
|
|
g.Expect(nodes.Items).To(HaveLen(1))
|
|
g.Expect(nodes.Items[0].Labels).To(HaveKeyWithValue("node-role.kubernetes.io/worker", "true"))
|
|
}).
|
|
WithTimeout(time.Minute).
|
|
WithPolling(time.Second).
|
|
Should(Succeed())
|
|
})
|
|
|
|
It("has the provider.cattle.io label set to k3k", func() {
|
|
Eventually(func(g Gomega) {
|
|
ctx := GinkgoT().Context()
|
|
|
|
key := client.ObjectKeyFromObject(virtualCluster.Cluster)
|
|
g.Expect(k8sClient.Get(ctx, key, virtualCluster.Cluster)).To(Succeed())
|
|
g.Expect(virtualCluster.Cluster.Labels).To(HaveKeyWithValue("provider.cattle.io", "k3k"))
|
|
}).
|
|
WithTimeout(time.Minute).
|
|
WithPolling(time.Second).
|
|
Should(Succeed())
|
|
})
|
|
})
|