Fix pod cleanup on exit in k8s connection mode

Author: tsipinakis

internal/kubernetes/kubernetesPodImpl.go

@@ -274,7 +274,12 @@ func (k *kubernetesPodImpl) remove(ctx context.Context) error {
 	k.lock.Lock()
 	k.shuttingDown = true
 	k.lock.Unlock()
-	k.wg.Wait()
+	// Do not wait to exit in connection mode.
+	// In session mode this function is called everytime an exec exits in order to ensure all signals/messages have been handled.
+	// In connection mode this is only called in ssh disconnect, in which case we want to stop all activity
+	if k.config.Pod.Mode == config2.KubernetesExecutionModeSession {
+		k.wg.Wait()
+	}
 	k.lock.Lock()
 	k.shutdown = true
 	k.lock.Unlock()