Fix issue #374 with determining the cassandra version from image

pkg/controller/cassandracluster/generator.go

@@ -631,8 +631,8 @@ func initContainerEnvVar(cc *api.CassandraCluster, status *api.CassandraClusterS
 	image := strings.Split(cc.Spec.CassandraImage, ":")
 	serverVersion := cc.Spec.ServerVersion
 	if serverVersion == "" {
-		if len(image) == 2 {
-			version := strings.Split(image[1], "-")
+		if len(image) >= 2 {
+			version := strings.Split(image[len(image)-1], "-")
 			serverVersion = version[0]
 			if len(version) != 1 {
 				serverVersion += ".0"

pkg/controller/cassandracluster/generator_test.go

@@ -65,6 +65,15 @@ func helperInitCluster(t *testing.T, name string) (*ReconcileCassandraCluster, *
 	return &rcc, &cc
 }
 
+func GetEnvVarByName(envVars []v1.EnvVar, name string) *v1.EnvVar {
+	for _, envVar := range envVars {
+		if envVar.Name == name {
+			return &envVar
+		}
+	}
+	return nil
+}
+
 func TestCreateNodeAffinity(t *testing.T) {
 	assert := assert.New(t)
 
@@ -330,6 +339,30 @@ func TestInitContainerConfigFileData(t *testing.T) {
 	assert.Equal(configFileData.String(), initEnvVar[0].Value)
 }
 
+func TestInitContainerServerVersionDetect(t *testing.T) {
+	dcName := "dc1"
+	rackName := "rack1"
+	dcRackName := fmt.Sprintf("%s-%s", dcName, rackName)
+
+	_, cc := helperInitCluster(t, "cassandracluster-1DC.yaml")
+	// Cassandra image contains tag with cassandra version
+	imageTag := "1.0.0"
+	cc.Spec.CassandraImage = fmt.Sprintf("artifactory-name/repo/cassandra:%s", imageTag)
+	cassieResources := cc.Spec.Resources
+
+	assert := assert.New(t)
+	initEnvVar := initContainerEnvVar(cc, &cc.Status, cassieResources, dcRackName)
+	prodVer := GetEnvVarByName(initEnvVar, "PRODUCT_VERSION")
+	assert.Equal(imageTag, prodVer.Value)
+
+	// Cassandra image contains custom port for artifactory and cassandra version
+	imageTag = "1.0.1"
+	cc.Spec.CassandraImage = fmt.Sprintf("artifactory-name:44301/repo/cassandra:%s", imageTag)
+	initEnvVar = initContainerEnvVar(cc, &cc.Status, cassieResources, dcRackName)
+	prodVer = GetEnvVarByName(initEnvVar, "PRODUCT_VERSION")
+	assert.Equal(imageTag, prodVer.Value)
+}
+
 func TestGenerateCassandraStatefulSet(t *testing.T) {
 	assert := assert.New(t)
 	dcName := "dc1"