Update controller.go (#233)

Author: bakito

pkg/controller/controller.go

@@ -72,7 +72,7 @@ func (r *PodReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.R
 	node := pod.Spec.NodeName
 
 	if pod.Status.Phase == corev1.PodSucceeded || pod.Status.Phase == corev1.PodFailed {
-		if r.Controller.Config().SavePodLog && pod.DeletionTimestamp == nil {
+		if r.Controller.Config().SavePodLog && r.Controller.Has(node, executionID) {
 			r.savePodLogs(ctx, pod, executionID)
 		}
 		if err := r.Controller.PodTerminated(executionID, node, pod.Status.Phase); err != nil {

pkg/controller/controller_test.go

@@ -124,6 +124,7 @@ var _ = Describe("Controller", func() {
 		It("should update controller on pod succeeded with logs of 2 containers", func() {
 			cfg.SavePodLog = true
 			mockController.EXPECT().Config().Return(cfg).AnyTimes()
+			mockController.EXPECT().Has(gm.Any(), executionID).Return(true)
 			mockSink.EXPECT().WithValues(gm.Any()).Return(mockSink).AnyTimes()
 			mockSink.EXPECT().Info(gm.Any(), gm.Any()).AnyTimes()
 			mockClient.EXPECT().Get(gm.Any(), gm.Any(), gm.AssignableToTypeOf(&corev1.Pod{})).