mirror of https://github.com/tailscale/tailscale/
cmd/k8s-operator: add a basic unit test.
The test verifies one of the successful reconcile paths, where a client requests an exposed service via a LoadBalancer class. Updates #502. Signed-off-by: David Anderson <danderson@tailscale.com>pull/6719/head
David Anderson
2 years ago
committed by
Dave Anderson
parent
3b7ae39a06
commit
bc8f5a7734
@ -0,0 +1,334 @@
|
|||||||
|
// Copyright (c) 2022 Tailscale Inc & AUTHORS All rights reserved.
|
||||||
|
// Use of this source code is governed by a BSD-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/go-cmp/cmp"
|
||||||
|
appsv1 "k8s.io/api/apps/v1"
|
||||||
|
corev1 "k8s.io/api/core/v1"
|
||||||
|
v1 "k8s.io/api/core/v1"
|
||||||
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||||
|
"k8s.io/apimachinery/pkg/types"
|
||||||
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
||||||
|
"sigs.k8s.io/controller-runtime/pkg/client/fake"
|
||||||
|
"sigs.k8s.io/controller-runtime/pkg/reconcile"
|
||||||
|
"tailscale.com/client/tailscale"
|
||||||
|
"tailscale.com/types/ptr"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestController(t *testing.T) {
|
||||||
|
fc := fake.NewFakeClient()
|
||||||
|
ft := &fakeTSClient{}
|
||||||
|
sr := &ServiceReconciler{
|
||||||
|
Client: fc,
|
||||||
|
tsClient: ft,
|
||||||
|
defaultTags: []string{"tag:k8s"},
|
||||||
|
operatorNamespace: "operator-ns",
|
||||||
|
proxyImage: "tailscale/tailscale",
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a service that we should manage, and check that the initial round
|
||||||
|
// of objects looks right.
|
||||||
|
mustCreate(t, fc, &corev1.Service{
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
Name: "test",
|
||||||
|
Namespace: "default",
|
||||||
|
// The apiserver is supposed to set the UID, but the fake client
|
||||||
|
// doesn't. So, set it explicitly because other code later depends
|
||||||
|
// on it being set.
|
||||||
|
UID: types.UID("1234-UID"),
|
||||||
|
},
|
||||||
|
Spec: corev1.ServiceSpec{
|
||||||
|
ClusterIP: "10.20.30.40",
|
||||||
|
Type: corev1.ServiceTypeLoadBalancer,
|
||||||
|
LoadBalancerClass: ptr.To("tailscale"),
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
expectRequeue(t, sr, "default", "test")
|
||||||
|
|
||||||
|
fullName, shortName := findGenName(t, fc, "default", "test")
|
||||||
|
|
||||||
|
expectEqual(t, fc, &corev1.Secret{
|
||||||
|
TypeMeta: metav1.TypeMeta{
|
||||||
|
Kind: "Secret",
|
||||||
|
APIVersion: "v1",
|
||||||
|
},
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
Name: fullName,
|
||||||
|
Namespace: "operator-ns",
|
||||||
|
ResourceVersion: "1",
|
||||||
|
Labels: map[string]string{
|
||||||
|
"tailscale.com/managed": "true",
|
||||||
|
"tailscale.com/parent-resource": "test",
|
||||||
|
"tailscale.com/parent-resource-ns": "default",
|
||||||
|
"tailscale.com/parent-resource-type": "svc",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
StringData: map[string]string{
|
||||||
|
"authkey": "secret-authkey",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
expectEqual(t, fc, &corev1.Service{
|
||||||
|
TypeMeta: metav1.TypeMeta{
|
||||||
|
Kind: "Service",
|
||||||
|
APIVersion: "v1",
|
||||||
|
},
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
Name: shortName,
|
||||||
|
GenerateName: "ts-test-",
|
||||||
|
Namespace: "operator-ns",
|
||||||
|
ResourceVersion: "1",
|
||||||
|
Labels: map[string]string{
|
||||||
|
"tailscale.com/managed": "true",
|
||||||
|
"tailscale.com/parent-resource": "test",
|
||||||
|
"tailscale.com/parent-resource-ns": "default",
|
||||||
|
"tailscale.com/parent-resource-type": "svc",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Spec: corev1.ServiceSpec{
|
||||||
|
Selector: map[string]string{
|
||||||
|
"app": "1234-UID",
|
||||||
|
},
|
||||||
|
ClusterIP: "None",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
expectEqual(t, fc, &appsv1.StatefulSet{
|
||||||
|
TypeMeta: metav1.TypeMeta{
|
||||||
|
Kind: "StatefulSet",
|
||||||
|
APIVersion: "apps/v1",
|
||||||
|
},
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
Name: shortName,
|
||||||
|
Namespace: "operator-ns",
|
||||||
|
ResourceVersion: "1",
|
||||||
|
Labels: map[string]string{
|
||||||
|
"tailscale.com/managed": "true",
|
||||||
|
"tailscale.com/parent-resource": "test",
|
||||||
|
"tailscale.com/parent-resource-ns": "default",
|
||||||
|
"tailscale.com/parent-resource-type": "svc",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Spec: appsv1.StatefulSetSpec{
|
||||||
|
Replicas: ptr.To[int32](1),
|
||||||
|
Selector: &metav1.LabelSelector{
|
||||||
|
MatchLabels: map[string]string{"app": "1234-UID"},
|
||||||
|
},
|
||||||
|
ServiceName: shortName,
|
||||||
|
Template: corev1.PodTemplateSpec{
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
DeletionGracePeriodSeconds: ptr.To[int64](10),
|
||||||
|
Labels: map[string]string{"app": "1234-UID"},
|
||||||
|
},
|
||||||
|
Spec: corev1.PodSpec{
|
||||||
|
ServiceAccountName: "proxies",
|
||||||
|
InitContainers: []corev1.Container{
|
||||||
|
{
|
||||||
|
Name: "sysctler",
|
||||||
|
Image: "busybox",
|
||||||
|
Command: []string{"/bin/sh"},
|
||||||
|
Args: []string{"-c", "sysctl -w net.ipv4.ip_forward=1 net.ipv6.conf.all.forwarding=1"},
|
||||||
|
SecurityContext: &corev1.SecurityContext{
|
||||||
|
Privileged: ptr.To(true),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Containers: []v1.Container{
|
||||||
|
{
|
||||||
|
Name: "tailscale",
|
||||||
|
Image: "tailscale/tailscale",
|
||||||
|
Env: []v1.EnvVar{
|
||||||
|
{Name: "TS_USERSPACE", Value: "false"},
|
||||||
|
{Name: "TS_AUTH_ONCE", Value: "true"},
|
||||||
|
{Name: "TS_DEST_IP", Value: "10.20.30.40"},
|
||||||
|
{Name: "TS_KUBE_SECRET", Value: fullName},
|
||||||
|
},
|
||||||
|
SecurityContext: &corev1.SecurityContext{
|
||||||
|
Capabilities: &corev1.Capabilities{
|
||||||
|
Add: []corev1.Capability{"NET_ADMIN"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
ImagePullPolicy: "Always",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
// Normally the Tailscale proxy pod would come up here and write its info
|
||||||
|
// into the secret. Simulate that, then verify reconcile again and verify
|
||||||
|
// that we get to the end.
|
||||||
|
mustUpdate(t, fc, "operator-ns", fullName, func(s *corev1.Secret) {
|
||||||
|
if s.Data == nil {
|
||||||
|
s.Data = map[string][]byte{}
|
||||||
|
}
|
||||||
|
s.Data["device_id"] = []byte("ts-id-1234")
|
||||||
|
s.Data["device_fqdn"] = []byte("tailscale.device.name.")
|
||||||
|
})
|
||||||
|
expectReconciled(t, sr, "default", "test")
|
||||||
|
expectEqual(t, fc, &corev1.Service{
|
||||||
|
TypeMeta: metav1.TypeMeta{
|
||||||
|
Kind: "Service",
|
||||||
|
APIVersion: "v1",
|
||||||
|
},
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
Name: "test",
|
||||||
|
Namespace: "default",
|
||||||
|
ResourceVersion: "4",
|
||||||
|
Finalizers: []string{"tailscale.com/finalizer"},
|
||||||
|
UID: types.UID("1234-UID"),
|
||||||
|
},
|
||||||
|
Spec: corev1.ServiceSpec{
|
||||||
|
ClusterIP: "10.20.30.40",
|
||||||
|
Type: corev1.ServiceTypeLoadBalancer,
|
||||||
|
LoadBalancerClass: ptr.To("tailscale"),
|
||||||
|
},
|
||||||
|
Status: corev1.ServiceStatus{
|
||||||
|
LoadBalancer: corev1.LoadBalancerStatus{
|
||||||
|
Ingress: []corev1.LoadBalancerIngress{
|
||||||
|
{
|
||||||
|
Hostname: "tailscale.device.name",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func findGenName(t *testing.T, client client.Client, ns, name string) (full, noSuffix string) {
|
||||||
|
t.Helper()
|
||||||
|
labels := map[string]string{
|
||||||
|
LabelManaged: "true",
|
||||||
|
LabelParentName: name,
|
||||||
|
LabelParentNamespace: ns,
|
||||||
|
LabelParentType: "svc",
|
||||||
|
}
|
||||||
|
s, err := getSingleObject[corev1.Secret](context.Background(), client, "operator-ns", labels)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("finding secret for %q: %v", name, err)
|
||||||
|
}
|
||||||
|
return s.GetName(), strings.TrimSuffix(s.GetName(), "-0")
|
||||||
|
}
|
||||||
|
|
||||||
|
func mustCreate(t *testing.T, client client.Client, obj client.Object) {
|
||||||
|
t.Helper()
|
||||||
|
if err := client.Create(context.Background(), obj); err != nil {
|
||||||
|
t.Fatalf("creating %q: %v", obj.GetName(), err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func mustUpdate[T any, O ptrObject[T]](t *testing.T, client client.Client, ns, name string, update func(O)) {
|
||||||
|
t.Helper()
|
||||||
|
obj := O(new(T))
|
||||||
|
if err := client.Get(context.Background(), types.NamespacedName{
|
||||||
|
Name: name,
|
||||||
|
Namespace: ns,
|
||||||
|
}, obj); err != nil {
|
||||||
|
t.Fatalf("getting %q: %v", name, err)
|
||||||
|
}
|
||||||
|
update(obj)
|
||||||
|
if err := client.Update(context.Background(), obj); err != nil {
|
||||||
|
t.Fatalf("updating %q: %v", name, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func expectEqual[T any, O ptrObject[T]](t *testing.T, client client.Client, want O) {
|
||||||
|
t.Helper()
|
||||||
|
got := O(new(T))
|
||||||
|
if err := client.Get(context.Background(), types.NamespacedName{
|
||||||
|
Name: want.GetName(),
|
||||||
|
Namespace: want.GetNamespace(),
|
||||||
|
}, got); err != nil {
|
||||||
|
t.Fatalf("getting %q: %v", want.GetName(), err)
|
||||||
|
}
|
||||||
|
if diff := cmp.Diff(got, want); diff != "" {
|
||||||
|
t.Fatalf("unexpected object (-got +want):\n%s", diff)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func expectReconciled(t *testing.T, sr *ServiceReconciler, ns, name string) {
|
||||||
|
t.Helper()
|
||||||
|
req := reconcile.Request{
|
||||||
|
NamespacedName: types.NamespacedName{
|
||||||
|
Name: name,
|
||||||
|
Namespace: ns,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
res, err := sr.Reconcile(context.Background(), req)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Reconcile: unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
if res.Requeue {
|
||||||
|
t.Fatalf("unexpected immediate requeue")
|
||||||
|
}
|
||||||
|
if res.RequeueAfter != 0 {
|
||||||
|
t.Fatalf("unexpected timed requeue")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func expectRequeue(t *testing.T, sr *ServiceReconciler, ns, name string) {
|
||||||
|
t.Helper()
|
||||||
|
req := reconcile.Request{
|
||||||
|
NamespacedName: types.NamespacedName{
|
||||||
|
Name: name,
|
||||||
|
Namespace: ns,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
res, err := sr.Reconcile(context.Background(), req)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Reconcile: unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
if res.Requeue {
|
||||||
|
t.Fatalf("unexpected immediate requeue")
|
||||||
|
}
|
||||||
|
if res.RequeueAfter == 0 {
|
||||||
|
t.Fatalf("expected timed requeue, got success")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type fakeTSClient struct {
|
||||||
|
sync.Mutex
|
||||||
|
keyRequests []tailscale.KeyCapabilities
|
||||||
|
deleted []string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *fakeTSClient) CreateKey(ctx context.Context, caps tailscale.KeyCapabilities) (string, *tailscale.Key, error) {
|
||||||
|
c.Lock()
|
||||||
|
defer c.Unlock()
|
||||||
|
c.keyRequests = append(c.keyRequests, caps)
|
||||||
|
k := &tailscale.Key{
|
||||||
|
ID: "key",
|
||||||
|
Created: time.Now(),
|
||||||
|
Expires: time.Now().Add(24 * time.Hour),
|
||||||
|
Capabilities: caps,
|
||||||
|
}
|
||||||
|
return "secret-authkey", k, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *fakeTSClient) DeleteDevice(ctx context.Context, deviceID string) error {
|
||||||
|
c.Lock()
|
||||||
|
defer c.Unlock()
|
||||||
|
c.deleted = append(c.deleted, deviceID)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *fakeTSClient) KeyRequests() []tailscale.KeyCapabilities {
|
||||||
|
c.Lock()
|
||||||
|
defer c.Unlock()
|
||||||
|
return c.keyRequests
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *fakeTSClient) Deleted() []string {
|
||||||
|
c.Lock()
|
||||||
|
defer c.Unlock()
|
||||||
|
return c.deleted
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue