github/k3k
1
0
Fork 0
mirror of https://github.com/rancher/k3k.git synced 2026-08-15 02:16:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor the service tests.

Browse Source
This commit is contained in:
Kevin McDermott
2026-07-06 08:42:20 +01:00
parent f1dbf33435
commit 4cd5bd85d0
1 changed files with 153 additions and 183 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/controller/cluster/server/service_test.go
+153 -183
View File
@@ -13,7 +13,156 @@ import (
"github.com/rancher/k3k/pkg/apis/k3k.io/v1beta1"
)
func TestNoExposeService(t *testing.T) {
func TestService(t *testing.T) {
tests := map[string]struct {
clusterOpts []func(*v1beta1.Cluster)
serviceOpts []func(*corev1.Service)
}{
"no expose": {
serviceOpts: []func(*corev1.Service){
func(s *corev1.Service) {
s.Spec.Type = corev1.ServiceTypeClusterIP
s.Spec.Ports = []corev1.ServicePort{
{
Name: "k3s-server-port",
Protocol: corev1.ProtocolTCP,
Port: int32(443),
TargetPort: intstr.FromInt(6443),
},
{
Name: "k3s-etcd-port",
Protocol: corev1.ProtocolTCP,
Port: int32(2379),
},
}
},
},
},
"expose load balancer": {
clusterOpts: []func(*v1beta1.Cluster){
func(c *v1beta1.Cluster) {
c.Spec = v1beta1.ClusterSpec{
Expose: &v1beta1.ExposeConfig{
LoadBalancer: &v1beta1.LoadBalancerConfig{
ServerPort: ptr.To[int32](9443),
},
},
}
},
},
serviceOpts: []func(*corev1.Service){
func(s *corev1.Service) {
s.Spec.Type = corev1.ServiceTypeLoadBalancer
s.Spec.Ports = []corev1.ServicePort{
{
Name: "k3s-server-port",
Protocol: corev1.ProtocolTCP,
Port: int32(9443),
TargetPort: intstr.FromInt(6443),
},
{
Name: "k3s-etcd-port",
Protocol: corev1.ProtocolTCP,
Port: int32(2379),
},
}
},
},
},
"expose load balancer with annotations": {
clusterOpts: []func(*v1beta1.Cluster){
func(c *v1beta1.Cluster) {
c.Spec.Expose = &v1beta1.ExposeConfig{
Annotations: map[string]string{
"example.com/testing": "test-annotation",
},
LoadBalancer: &v1beta1.LoadBalancerConfig{
ServerPort: ptr.To[int32](9443),
},
}
},
},
serviceOpts: []func(*corev1.Service){
func(s *corev1.Service) {
s.Annotations = map[string]string{
"example.com/testing": "test-annotation",
}
s.Spec.Type = corev1.ServiceTypeLoadBalancer
s.Spec.Ports = []corev1.ServicePort{
{
Name: "k3s-server-port",
Protocol: corev1.ProtocolTCP,
Port: int32(9443),
TargetPort: intstr.FromInt(6443),
},
{
Name: "k3s-etcd-port",
Protocol: corev1.ProtocolTCP,
Port: int32(2379),
},
}
},
},
},
"expose node port": {
clusterOpts: []func(*v1beta1.Cluster){
func(c *v1beta1.Cluster) {
c.Spec.Expose = &v1beta1.ExposeConfig{
NodePort: &v1beta1.NodePortConfig{
ServerPort: ptr.To[int32](7443),
},
}
},
},
serviceOpts: []func(*corev1.Service){
func(s *corev1.Service) {
s.Spec.Type = corev1.ServiceTypeNodePort
s.Spec.Ports = []corev1.ServicePort{
{
Name: "k3s-etcd-port",
Protocol: corev1.ProtocolTCP,
Port: int32(2379),
},
}
},
},
},
}
for name, tt := range tests {
t.Run(name, func(t *testing.T) {
cluster := newTestCluster(tt.clusterOpts...)
want := newTestService(cluster, tt.serviceOpts...)
assert.Equal(t, want, Service(cluster))
})
}
}
func newTestService(cluster *v1beta1.Cluster, opts ...func(*corev1.Service)) *corev1.Service {
svc := &corev1.Service{
TypeMeta: metav1.TypeMeta{
Kind: "Service",
APIVersion: "v1",
},
ObjectMeta: metav1.ObjectMeta{
Name: "k3k-test-cluster-service",
Namespace: cluster.Namespace,
},
Spec: corev1.ServiceSpec{
Selector: map[string]string{
"cluster": "test-cluster",
"role": "server",
},
},
}
for _, opt := range opts {
opt(svc)
}
return svc
}
func newTestCluster(opts ...func(*v1beta1.Cluster)) *v1beta1.Cluster {
cluster := &v1beta1.Cluster{
ObjectMeta: metav1.ObjectMeta{
Name: "test-cluster",
@@ -21,188 +170,9 @@ func TestNoExposeService(t *testing.T) {
},
Spec: v1beta1.ClusterSpec{},
}
want := &corev1.Service{
TypeMeta: metav1.TypeMeta{
Kind: "Service",
APIVersion: "v1",
},
ObjectMeta: metav1.ObjectMeta{
Name: "k3k-test-cluster-service",
Namespace: cluster.Namespace,
},
Spec: corev1.ServiceSpec{
Selector: map[string]string{
"cluster": "test-cluster",
"role": "server",
},
Type: corev1.ServiceTypeClusterIP,
Ports: []corev1.ServicePort{
{
Name: "k3s-server-port",
Protocol: corev1.ProtocolTCP,
Port: int32(443),
TargetPort: intstr.FromInt(6443),
},
{
Name: "k3s-etcd-port",
Protocol: corev1.ProtocolTCP,
Port: int32(2379),
},
},
},
for _, opt := range opts {
opt(cluster)
}
service := Service(cluster)
assert.Equal(t, want, service)
}
func TestExposeLoadBalancerService(t *testing.T) {
cluster := &v1beta1.Cluster{
ObjectMeta: metav1.ObjectMeta{
Name: "test-cluster",
Namespace: "test-namespace",
},
Spec: v1beta1.ClusterSpec{
Expose: &v1beta1.ExposeConfig{
LoadBalancer: &v1beta1.LoadBalancerConfig{
ServerPort: ptr.To[int32](9443),
},
},
},
}
want := &corev1.Service{
TypeMeta: metav1.TypeMeta{
Kind: "Service",
APIVersion: "v1",
},
ObjectMeta: metav1.ObjectMeta{
Name: "k3k-test-cluster-service",
Namespace: cluster.Namespace,
},
Spec: corev1.ServiceSpec{
Selector: map[string]string{
"cluster": "test-cluster",
"role": "server",
},
Type: corev1.ServiceTypeLoadBalancer,
Ports: []corev1.ServicePort{
{
Name: "k3s-server-port",
Protocol: corev1.ProtocolTCP,
Port: int32(9443),
TargetPort: intstr.FromInt(6443),
},
{
Name: "k3s-etcd-port",
Protocol: corev1.ProtocolTCP,
Port: int32(2379),
},
},
},
}
service := Service(cluster)
assert.Equal(t, want, service)
}
func TestExposeLoadBalancerServiceWithAnnotations(t *testing.T) {
cluster := &v1beta1.Cluster{
ObjectMeta: metav1.ObjectMeta{
Name: "test-cluster",
Namespace: "test-namespace",
},
Spec: v1beta1.ClusterSpec{
Expose: &v1beta1.ExposeConfig{
Annotations: map[string]string{
"example.com/testing": "test-annotation",
},
LoadBalancer: &v1beta1.LoadBalancerConfig{
ServerPort: ptr.To[int32](9443),
},
},
},
}
want := &corev1.Service{
TypeMeta: metav1.TypeMeta{
Kind: "Service",
APIVersion: "v1",
},
ObjectMeta: metav1.ObjectMeta{
Name: "k3k-test-cluster-service",
Namespace: cluster.Namespace,
Annotations: map[string]string{
"example.com/testing": "test-annotation",
},
},
Spec: corev1.ServiceSpec{
Selector: map[string]string{
"cluster": "test-cluster",
"role": "server",
},
Type: corev1.ServiceTypeLoadBalancer,
Ports: []corev1.ServicePort{
{
Name: "k3s-server-port",
Protocol: corev1.ProtocolTCP,
Port: int32(9443),
TargetPort: intstr.FromInt(6443),
},
{
Name: "k3s-etcd-port",
Protocol: corev1.ProtocolTCP,
Port: int32(2379),
},
},
},
}
service := Service(cluster)
assert.Equal(t, want, service)
}
func TestExposeNodePortService(t *testing.T) {
cluster := &v1beta1.Cluster{
ObjectMeta: metav1.ObjectMeta{
Name: "test-cluster",
Namespace: "test-namespace",
},
Spec: v1beta1.ClusterSpec{
Expose: &v1beta1.ExposeConfig{
NodePort: &v1beta1.NodePortConfig{
ServerPort: ptr.To[int32](7443),
},
},
},
}
want := &corev1.Service{
TypeMeta: metav1.TypeMeta{
Kind: "Service",
APIVersion: "v1",
},
ObjectMeta: metav1.ObjectMeta{
Name: "k3k-test-cluster-service",
Namespace: cluster.Namespace,
},
Spec: corev1.ServiceSpec{
Selector: map[string]string{
"cluster": "test-cluster",
"role": "server",
},
Type: corev1.ServiceTypeNodePort,
Ports: []corev1.ServicePort{
{
Name: "k3s-etcd-port",
Protocol: corev1.ProtocolTCP,
Port: int32(2379),
},
},
},
}
service := Service(cluster)
assert.Equal(t, want, service)
return cluster
}