refactor(trait): nil PodSpec check

* Added a check to raise an error on nil PodSpec for affinity and toleration traits
* Related unit test

Ref #2047

pkg/trait/affinity.go

@@ -75,21 +75,21 @@ func (t *affinityTrait) Configure(e *Environment) (bool, error) {
 func (t *affinityTrait) Apply(e *Environment) (err error) {
 	podSpec := e.GetIntegrationPodSpec()
 
-	if podSpec != nil {
-		if podSpec.Affinity == nil {
-			podSpec.Affinity = &corev1.Affinity{}
-		}
-		if err := t.addNodeAffinity(e, podSpec); err != nil {
-			return err
-		}
-		if err := t.addPodAffinity(e, podSpec); err != nil {
-			return err
-		}
-		if err := t.addPodAntiAffinity(e, podSpec); err != nil {
-			return err
-		}
+	if podSpec == nil {
+		return fmt.Errorf("could not find any integration deployment for %v", e.Integration.Name)
+	}
+	if podSpec.Affinity == nil {
+		podSpec.Affinity = &corev1.Affinity{}
+	}
+	if err := t.addNodeAffinity(e, podSpec); err != nil {
+		return err
+	}
+	if err := t.addPodAffinity(e, podSpec); err != nil {
+		return err
+	}
+	if err := t.addPodAntiAffinity(e, podSpec); err != nil {
+		return err
 	}
-
 	return nil
 }
 

pkg/trait/affinity_test.go

@@ -59,6 +59,15 @@ func TestConfigureDisabledAffinityTraitFails(t *testing.T) {
 	assert.Nil(t, err)
 }
 
+func TestApplyAffinityMissingDeployment(t *testing.T) {
+	tolerationTrait := createNominalAffinityTest()
+
+	environment := createNominalMissingDeploymentTraitTest()
+	err := tolerationTrait.Apply(environment)
+
+	assert.NotNil(t, err)
+}
+
 func TestApplyEmptyAffinityLabelsDoesSucceed(t *testing.T) {
 	affinityTrait := createNominalAffinityTest()
 

pkg/trait/toleration.go

@@ -73,14 +73,13 @@ func (t *tolerationTrait) Apply(e *Environment) (err error) {
 	}
 	podSpec := e.GetIntegrationPodSpec()
 
-	// Add the toleration
-	if podSpec != nil {
-		if podSpec.Tolerations == nil {
-			podSpec.Tolerations = make([]corev1.Toleration, 0)
-		}
-		podSpec.Tolerations = append(podSpec.Tolerations, tolerations...)
+	if podSpec == nil {
+		return fmt.Errorf("could not find any integration deployment for %v", e.Integration.Name)
 	}
-
+	if podSpec.Tolerations == nil {
+		podSpec.Tolerations = make([]corev1.Toleration, 0)
+	}
+	podSpec.Tolerations = append(podSpec.Tolerations, tolerations...)
 	return nil
 }
 

pkg/trait/toleration_test.go

@@ -47,6 +47,16 @@ func TestApplyTolerationTraitMalformedTaint(t *testing.T) {
 	assert.NotNil(t, err)
 }
 
+func TestApplyPodTolerationMissingDeployment(t *testing.T) {
+	tolerationTrait := createNominalTolerationTrait()
+	tolerationTrait.Taints = append(tolerationTrait.Taints, "my-toleration=my-value:NoExecute")
+
+	environment := createNominalMissingDeploymentTraitTest()
+	err := tolerationTrait.Apply(environment)
+
+	assert.NotNil(t, err)
+}
+
 func TestApplyPodTolerationLabelsDefault(t *testing.T) {
 	tolerationTrait := createNominalTolerationTrait()
 	tolerationTrait.Taints = append(tolerationTrait.Taints, "my-toleration=my-value:NoExecute")

pkg/trait/trait_types_test.go

@@ -54,6 +54,22 @@ func createNominalDeploymentTraitTest() (*Environment, *appsv1.Deployment) {
 	return environment, deployment
 }
 
+func createNominalMissingDeploymentTraitTest() *Environment {
+	environment := &Environment{
+		Integration: &v1.Integration{
+			ObjectMeta: metav1.ObjectMeta{
+				Name: "integration-name",
+			},
+			Status: v1.IntegrationStatus{
+				Phase: v1.IntegrationPhaseDeploying,
+			},
+		},
+		Resources: kubernetes.NewCollection(),
+	}
+
+	return environment
+}
+
 func createNominalKnativeServiceTraitTest() (*Environment, *serving.Service) {
 	knativeService := &serving.Service{
 		ObjectMeta: metav1.ObjectMeta{