-
Notifications
You must be signed in to change notification settings - Fork 163
/
scyllacluster_upgrades.go
107 lines (90 loc) · 3.38 KB
/
scyllacluster_upgrades.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
// Copyright (C) 2021 ScyllaDB
package scyllacluster
import (
"context"
"fmt"
"time"
g "github.com/onsi/ginkgo"
gt "github.com/onsi/ginkgo/extensions/table"
o "github.com/onsi/gomega"
scyllaclusterfixture "github.com/scylladb/scylla-operator/test/e2e/fixture/scyllacluster"
"github.com/scylladb/scylla-operator/test/e2e/framework"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/types"
)
var _ = g.Describe("ScyllaCluster upgrades", func() {
defer g.GinkgoRecover()
f := framework.NewFramework("scyllacluster")
type entry struct {
rackSize int32
initialVersion string
updatedVersion string
}
describeEntry := func(e *entry) string {
return fmt.Sprintf("with %d member(s) from %q to %q", e.rackSize, e.initialVersion, e.updatedVersion)
}
gt.DescribeTable("should deploy and update",
func(e *entry) {
ctx, cancel := context.WithTimeout(context.Background(), testTimout)
defer cancel()
sc := scyllaclusterfixture.BasicScyllaCluster.ReadOrFail()
sc.Spec.Version = e.initialVersion
sc.Spec.Datacenter.Racks[0].Members = e.rackSize
framework.By("Creating a ScyllaCluster")
err := framework.SetupScyllaClusterSA(ctx, f.KubeClient().CoreV1(), f.KubeClient().RbacV1(), f.Namespace(), sc.Name)
o.Expect(err).NotTo(o.HaveOccurred())
sc, err = f.ScyllaClient().ScyllaV1().ScyllaClusters(f.Namespace()).Create(ctx, sc, metav1.CreateOptions{})
o.Expect(err).NotTo(o.HaveOccurred())
o.Expect(sc.Spec.Version).To(o.Equal(e.initialVersion))
framework.By("Waiting for the ScyllaCluster to deploy")
waitCtx1, waitCtx1Cancel := contextForRollout(ctx, sc)
defer waitCtx1Cancel()
sc, err = waitForScyllaClusterState(waitCtx1, f.ScyllaClient().ScyllaV1(), sc.Namespace, sc.Name, scyllaClusterRolledOut)
o.Expect(err).NotTo(o.HaveOccurred())
verifyScyllaCluster(ctx, f.KubeClient(), sc)
framework.By("triggering and update")
sc, err = f.ScyllaClient().ScyllaV1().ScyllaClusters(f.Namespace()).Patch(
ctx,
sc.Name,
types.MergePatchType,
[]byte(fmt.Sprintf(
`{"metadata":{"uid":"%s"},"spec":{"version":"%s"}}`,
sc.UID,
e.updatedVersion,
)),
metav1.PatchOptions{},
)
o.Expect(err).NotTo(o.HaveOccurred())
o.Expect(sc.Spec.Version).To(o.Equal(e.updatedVersion))
// FIXME: Remove when we have observedGeneration. Until then this is racy.
time.Sleep(10 * time.Second)
framework.By("Waiting for the ScyllaCluster to re-deploy")
waitCtx2, waitCtx2Cancel := contextForRollout(ctx, sc)
defer waitCtx2Cancel()
sc, err = waitForScyllaClusterState(waitCtx2, f.ScyllaClient().ScyllaV1(), sc.Namespace, sc.Name, scyllaClusterRolledOut)
o.Expect(err).NotTo(o.HaveOccurred())
verifyScyllaCluster(ctx, f.KubeClient(), sc)
},
// Test 1 and 3 member rack to cover e.g. handling PDBs correctly.
gt.Entry(describeEntry, &entry{
rackSize: 1,
initialVersion: updateFromScyllaVersion,
updatedVersion: updateToScyllaVersion,
}),
gt.Entry(describeEntry, &entry{
rackSize: 3,
initialVersion: updateFromScyllaVersion,
updatedVersion: updateToScyllaVersion,
}),
gt.Entry(describeEntry, &entry{
rackSize: 1,
initialVersion: upgradeFromScyllaVersion,
updatedVersion: upgradeToScyllaVersion,
}),
gt.Entry(describeEntry, &entry{
rackSize: 3,
initialVersion: upgradeFromScyllaVersion,
updatedVersion: upgradeToScyllaVersion,
}),
)
})