mirror of
https://github.com/sablierapp/sablier.git
synced 2025-12-21 13:23:03 +01:00
102 lines
3.3 KiB
Go
102 lines
3.3 KiB
Go
package kubernetes_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/neilotoole/slogt"
|
|
"github.com/sablierapp/sablier/pkg/config"
|
|
"github.com/sablierapp/sablier/pkg/provider/kubernetes"
|
|
"gotest.tools/v3/assert"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
func TestKubernetesProvider_NotifyInstanceStopped(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping test in short mode.")
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(t.Context(), 30*time.Second)
|
|
defer cancel()
|
|
kind := setupKinD(t, ctx)
|
|
conf := config.NewProviderConfig().Kubernetes
|
|
conf.QPS = 100
|
|
conf.Burst = 100
|
|
p, err := kubernetes.New(ctx, kind.client, slogt.New(t), conf)
|
|
assert.NilError(t, err)
|
|
|
|
waitC := make(chan string)
|
|
go p.NotifyInstanceStopped(ctx, waitC)
|
|
|
|
t.Run("deployment is scaled to 0 replicas", func(t *testing.T) {
|
|
d, err := kind.CreateMimicDeployment(ctx, MimicOptions{})
|
|
assert.NilError(t, err)
|
|
|
|
err = WaitForDeploymentReady(ctx, kind.client, d.Namespace, d.Name)
|
|
assert.NilError(t, err)
|
|
|
|
s, err := p.Client.AppsV1().Deployments(d.Namespace).GetScale(ctx, d.Name, metav1.GetOptions{})
|
|
assert.NilError(t, err)
|
|
|
|
s.Spec.Replicas = 0
|
|
_, err = p.Client.AppsV1().Deployments(d.Namespace).UpdateScale(ctx, d.Name, s, metav1.UpdateOptions{})
|
|
assert.NilError(t, err)
|
|
|
|
name := <-waitC
|
|
|
|
// Docker container name is prefixed with a slash, but we don't use it
|
|
assert.Equal(t, name, kubernetes.DeploymentName(d, kubernetes.ParseOptions{Delimiter: "_"}).Original)
|
|
})
|
|
t.Run("deployment is removed", func(t *testing.T) {
|
|
d, err := kind.CreateMimicDeployment(ctx, MimicOptions{})
|
|
assert.NilError(t, err)
|
|
|
|
err = WaitForDeploymentReady(ctx, kind.client, d.Namespace, d.Name)
|
|
assert.NilError(t, err)
|
|
|
|
err = p.Client.AppsV1().Deployments(d.Namespace).Delete(ctx, d.Name, metav1.DeleteOptions{})
|
|
assert.NilError(t, err)
|
|
|
|
name := <-waitC
|
|
|
|
// Docker container name is prefixed with a slash, but we don't use it
|
|
assert.Equal(t, name, kubernetes.DeploymentName(d, kubernetes.ParseOptions{Delimiter: "_"}).Original)
|
|
})
|
|
t.Run("statefulSet is scaled to 0 replicas", func(t *testing.T) {
|
|
ss, err := kind.CreateMimicStatefulSet(ctx, MimicOptions{})
|
|
assert.NilError(t, err)
|
|
|
|
err = WaitForStatefulSetReady(ctx, kind.client, ss.Namespace, ss.Name)
|
|
assert.NilError(t, err)
|
|
|
|
s, err := p.Client.AppsV1().StatefulSets(ss.Namespace).GetScale(ctx, ss.Name, metav1.GetOptions{})
|
|
assert.NilError(t, err)
|
|
|
|
s.Spec.Replicas = 0
|
|
_, err = p.Client.AppsV1().StatefulSets(ss.Namespace).UpdateScale(ctx, ss.Name, s, metav1.UpdateOptions{})
|
|
assert.NilError(t, err)
|
|
|
|
name := <-waitC
|
|
|
|
// Docker container name is prefixed with a slash, but we don't use it
|
|
assert.Equal(t, name, kubernetes.StatefulSetName(ss, kubernetes.ParseOptions{Delimiter: "_"}).Original)
|
|
})
|
|
|
|
t.Run("statefulSet is removed", func(t *testing.T) {
|
|
ss, err := kind.CreateMimicStatefulSet(ctx, MimicOptions{})
|
|
assert.NilError(t, err)
|
|
|
|
err = WaitForStatefulSetReady(ctx, kind.client, ss.Namespace, ss.Name)
|
|
assert.NilError(t, err)
|
|
|
|
err = p.Client.AppsV1().StatefulSets(ss.Namespace).Delete(ctx, ss.Name, metav1.DeleteOptions{})
|
|
assert.NilError(t, err)
|
|
|
|
name := <-waitC
|
|
|
|
// Docker container name is prefixed with a slash, but we don't use it
|
|
assert.Equal(t, name, kubernetes.StatefulSetName(ss, kubernetes.ParseOptions{Delimiter: "_"}).Original)
|
|
})
|
|
}
|