test: extend unit tests

Signed-off-by: Gabriele Fedi <[email protected]>

Author: GabriFedi97

pkg/podexec/pod_command_executor_test.go

@@ -35,6 +35,7 @@ import (
 	"k8s.io/apimachinery/pkg/runtime/schema"
 	"k8s.io/client-go/rest"
 	"k8s.io/client-go/tools/remotecommand"
+	"k8s.io/utils/ptr"
 
 	v1 "github.com/vmware-tanzu/velero/pkg/apis/velero/v1"
 	velerotest "github.com/vmware-tanzu/velero/pkg/test"
@@ -253,6 +254,15 @@ func TestEnsureContainerExists(t *testing.T) {
 					Name: "foo",
 				},
 			},
+			InitContainers: []corev1api.Container{
+				{
+					Name: "baz",
+				},
+				{
+					Name:          "qux",
+					RestartPolicy: ptr.To(corev1api.ContainerRestartPolicyAlways),
+				},
+			},
 		},
 	}
 
@@ -261,6 +271,12 @@ func TestEnsureContainerExists(t *testing.T) {
 
 	err = ensureContainerExists(pod, "foo")
 	assert.NoError(t, err)
+
+	err = ensureContainerExists(pod, "baz")
+	require.EqualError(t, err, `no such container: "baz"`)
+
+	err = ensureContainerExists(pod, "qux")
+	assert.NoError(t, err)
 }
 
 func TestPodCompeted(t *testing.T) {