Skip to content

Commit

Permalink
[#402] Support Floating x or x.y version is Artemis CRD version string
Browse files Browse the repository at this point in the history
  • Loading branch information
@howardgao
howardgao committed Dec 2, 2022
1 parent 7bd65cf commit f701f1a
Show file tree
Hide file tree
Showing 5 changed files with 320 additions and 22 deletions.
207 changes: 204 additions & 3 deletions controllers/activemqartemis_controller_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ import (
"strconv"
"strings"

"github.com/Masterminds/semver"
brokerv2alpha4 "github.com/artemiscloud/activemq-artemis-operator/api/v2alpha4"
"github.com/artemiscloud/activemq-artemis-operator/api/v2alpha5"
"github.com/artemiscloud/activemq-artemis-operator/pkg/client/clientset/versioned/typed/broker/v1beta1"
Expand Down Expand Up @@ -118,6 +119,203 @@ var _ = Describe("artemis controller", func() {
}
})

Context("broker versions", Label("broker-versions"), func() {
versionList := []string{
"7.8.1",
"7.8.2",
"7.8.3",
"7.10.0",
"7.10.1",
"7.10.2",
"7.9.0",
"7.9.1",
"7.9.2",
"7.9.3",
"7.9.4",
}
versionsMap := make(map[string]*semver.Version)
for _, raw := range versionList {
v, _ := semver.NewVersion(raw)
versionsMap[v.String()] = v
}

It("resolve various versions", func() {
requested, _ := semver.NewVersion("7")
resolved := common.ResolveBrokerVersion(versionsMap, requested)
Expect(resolved).NotTo(BeNil())
Expect(resolved.String()).To(Equal("7.10.2"))

requested, _ = semver.NewVersion("7.8")
resolved = common.ResolveBrokerVersion(versionsMap, requested)
Expect(resolved).NotTo(BeNil())
Expect(resolved.String()).To(Equal("7.8.3"))

requested, _ = semver.NewVersion("7.9")
resolved = common.ResolveBrokerVersion(versionsMap, requested)
Expect(resolved).NotTo(BeNil())
Expect(resolved.String()).To(Equal("7.9.4"))

requested, _ = semver.NewVersion("7.9.0")
resolved = common.ResolveBrokerVersion(versionsMap, requested)
Expect(resolved).NotTo(BeNil())
Expect(resolved.String()).To(Equal("7.9.0"))

requested, _ = semver.NewVersion("7.10")
resolved = common.ResolveBrokerVersion(versionsMap, requested)
Expect(resolved).NotTo(BeNil())
Expect(resolved.String()).To(Equal("7.10.2"))

requested, _ = semver.NewVersion("7.10.1")
resolved = common.ResolveBrokerVersion(versionsMap, requested)
Expect(resolved).NotTo(BeNil())
Expect(resolved.String()).To(Equal("7.10.1"))

requested, _ = semver.NewVersion("7.11")
resolved = common.ResolveBrokerVersion(versionsMap, requested)
Expect(resolved).To(BeNil())

requested, _ = semver.NewVersion("8")
resolved = common.ResolveBrokerVersion(versionsMap, requested)
Expect(resolved).To(BeNil())

})

It("version ignored if images are explicitly specified", func() {
By("deploy a broker with images specified")
brokerCr, createdBrokerCr := DeployCustomBroker(defaultNamespace, func(candidate *brokerv1beta1.ActiveMQArtemis) {
candidate.Spec.Version = "2"
candidate.Spec.DeploymentPlan.Image = "myrepo/my-image:1.0"
candidate.Spec.DeploymentPlan.InitImage = "myrepo/my-init-image:1.0"
})

By("checking the image should respect the CR values")
createdSs := &appsv1.StatefulSet{}
ssKey := types.NamespacedName{Name: namer.CrToSS(brokerCr.Name), Namespace: defaultNamespace}
Eventually(func(g Gomega) {
g.Expect(k8sClient.Get(ctx, ssKey, createdSs)).Should(Succeed())
mainContainer := createdSs.Spec.Template.Spec.Containers[0]
g.Expect(mainContainer.Image).To(Equal("myrepo/my-image:1.0"))
initContainer := createdSs.Spec.Template.Spec.InitContainers[0]
g.Expect(initContainer.Image).To(Equal("myrepo/my-init-image:1.0"))

}, timeout, interval).Should(Succeed())

// cleanup
Expect(k8sClient.Delete(ctx, brokerCr)).Should(Succeed())
By("check it has gone")
Eventually(func() bool {
return checkCrdDeleted(brokerCr.Name, defaultNamespace, createdBrokerCr)
}, timeout, interval).Should(BeTrue())
})

It("specify only major version", func() {
os.Setenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Kubernetes_"+version.CompactLatestVersion, "quay.io/artemiscloud/fake-broker:latest")
os.Setenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Init_"+version.CompactLatestVersion, "quay.io/artemiscloud/fake-broker-init:latest")
defer func() {
os.Unsetenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Kubernetes_" + version.CompactLatestVersion)
os.Unsetenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Init_" + version.CompactLatestVersion)
}()
By("deploy a broker")
verFields := strings.Split(version.LatestVersion, ".")
major := verFields[0]
brokerCr, createdBrokerCr := DeployCustomBroker(defaultNamespace, func(candidate *brokerv1beta1.ActiveMQArtemis) {
candidate.Spec.Version = major
candidate.Spec.DeploymentPlan.Image = ""
candidate.Spec.DeploymentPlan.InitImage = ""
})

By("checking the default broker version would be the latest")
createdSs := &appsv1.StatefulSet{}
ssKey := types.NamespacedName{Name: namer.CrToSS(brokerCr.Name), Namespace: defaultNamespace}
Eventually(func(g Gomega) {
g.Expect(k8sClient.Get(ctx, ssKey, createdSs)).Should(Succeed())
mainContainer := createdSs.Spec.Template.Spec.Containers[0]
g.Expect(mainContainer.Image).To(Equal("quay.io/artemiscloud/fake-broker:latest"))
initContainer := createdSs.Spec.Template.Spec.InitContainers[0]
g.Expect(initContainer.Image).To(Equal("quay.io/artemiscloud/fake-broker-init:latest"))

}, timeout, interval).Should(Succeed())

// cleanup
Expect(k8sClient.Delete(ctx, brokerCr)).Should(Succeed())
By("check it has gone")
Eventually(func() bool {
return checkCrdDeleted(brokerCr.Name, defaultNamespace, createdBrokerCr)
}, timeout, interval).Should(BeTrue())
})

It("specify only major.minor version", func() {
os.Setenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Kubernetes_"+version.CompactLatestVersion, "quay.io/artemiscloud/fake-broker:latest")
os.Setenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Init_"+version.CompactLatestVersion, "quay.io/artemiscloud/fake-broker-init:latest")
defer func() {
os.Unsetenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Kubernetes_" + version.CompactLatestVersion)
os.Unsetenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Init_" + version.CompactLatestVersion)
}()
By("deploy a broker")
verFields := strings.Split(version.LatestVersion, ".")
major := verFields[0]
minor := verFields[1]
brokerCr, createdBrokerCr := DeployCustomBroker(defaultNamespace, func(candidate *brokerv1beta1.ActiveMQArtemis) {
candidate.Spec.Version = major + "." + minor
candidate.Spec.DeploymentPlan.Image = ""
candidate.Spec.DeploymentPlan.InitImage = ""
})

By("checking the default broker version would be the latest")
createdSs := &appsv1.StatefulSet{}
ssKey := types.NamespacedName{Name: namer.CrToSS(brokerCr.Name), Namespace: defaultNamespace}
Eventually(func(g Gomega) {
g.Expect(k8sClient.Get(ctx, ssKey, createdSs)).Should(Succeed())
mainContainer := createdSs.Spec.Template.Spec.Containers[0]
g.Expect(mainContainer.Image).To(Equal("quay.io/artemiscloud/fake-broker:latest"))
initContainer := createdSs.Spec.Template.Spec.InitContainers[0]
g.Expect(initContainer.Image).To(Equal("quay.io/artemiscloud/fake-broker-init:latest"))

}, timeout, interval).Should(Succeed())

// cleanup
Expect(k8sClient.Delete(ctx, brokerCr)).Should(Succeed())
By("check it has gone")
Eventually(func() bool {
return checkCrdDeleted(brokerCr.Name, defaultNamespace, createdBrokerCr)
}, timeout, interval).Should(BeTrue())
})

It("default broker versions", func() {
os.Setenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Kubernetes_"+version.CompactLatestVersion, "quay.io/artemiscloud/fake-broker:latest")
os.Setenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Init_"+version.CompactLatestVersion, "quay.io/artemiscloud/fake-broker-init:latest")
defer func() {
os.Unsetenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Kubernetes_" + version.CompactLatestVersion)
os.Unsetenv("RELATED_IMAGE_ActiveMQ_Artemis_Broker_Init_" + version.CompactLatestVersion)
}()
By("deploy a broker")
brokerCr, createdBrokerCr := DeployCustomBroker(defaultNamespace, func(candidate *brokerv1beta1.ActiveMQArtemis) {
candidate.Spec.Version = ""
candidate.Spec.DeploymentPlan.Image = ""
candidate.Spec.DeploymentPlan.InitImage = ""
})

By("checking the default broker version would be the latest")
createdSs := &appsv1.StatefulSet{}
ssKey := types.NamespacedName{Name: namer.CrToSS(brokerCr.Name), Namespace: defaultNamespace}
Eventually(func(g Gomega) {
g.Expect(k8sClient.Get(ctx, ssKey, createdSs)).Should(Succeed())
mainContainer := createdSs.Spec.Template.Spec.Containers[0]
g.Expect(mainContainer.Image).To(Equal("quay.io/artemiscloud/fake-broker:latest"))
initContainer := createdSs.Spec.Template.Spec.InitContainers[0]
g.Expect(initContainer.Image).To(Equal("quay.io/artemiscloud/fake-broker-init:latest"))

}, timeout, interval).Should(Succeed())

// cleanup
Expect(k8sClient.Delete(ctx, brokerCr)).Should(Succeed())
By("check it has gone")
Eventually(func() bool {
return checkCrdDeleted(brokerCr.Name, defaultNamespace, createdBrokerCr)
}, timeout, interval).Should(BeTrue())
})
})

Context("broker resource tracking", Label("broker-resource-tracking-context"), func() {
It("default user credential secret", func() {
By("deploy a broker")
Expand Down Expand Up @@ -690,13 +888,16 @@ var _ = Describe("artemis controller", func() {
Context("Versions Test", func() {
It("default image to use latest", func() {
crd := generateArtemisSpec(defaultNamespace)
imageToUse := determineImageToUse(&crd, "Kubernetes")
imageToUse, verr := determineImageToUse(&crd, "Kubernetes")
Expect(verr).To(BeNil())
Expect(imageToUse).To(Equal(version.LatestKubeImage), "actual", imageToUse)

imageToUse = determineImageToUse(&crd, "Init")
imageToUse, verr = determineImageToUse(&crd, "Init")
Expect(verr).To(BeNil())
Expect(imageToUse).To(Equal(version.LatestInitImage), "actual", imageToUse)
brokerCr := generateArtemisSpec(defaultNamespace)
compactVersionToUse := determineCompactVersionToUse(&brokerCr)
compactVersionToUse, verr := determineCompactVersionToUse(&brokerCr)
Expect(verr).To(BeNil())
yacfgProfileVersion = version.YacfgProfileVersionFromFullVersion[version.FullVersionFromCompactVersion[compactVersionToUse]]
Expect(yacfgProfileVersion).To(Equal("2.21.0"))
})
Expand Down

0 comments on commit f701f1a

Please sign in to comment.