use aks-runtime from main

e2e/aks-runtime/aks_runtime_test.go

@@ -20,7 +20,6 @@ import (
 	"github.com/edgelesssys/contrast/internal/kuberesource"
 	"github.com/stretchr/testify/require"
 	corev1 "k8s.io/api/core/v1"
-	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
 const testContainer = "testcontainer"
@@ -104,10 +103,10 @@ func TestAKSRuntime(t *testing.T) {
 	require.NoError(err)
 	require.NoError(c.WaitFor(ctx, kubeclient.Ready, kubeclient.Deployment{}, namespace, testContainer))
 
-	pods, err := c.Client.CoreV1().Pods(namespace).List(ctx, metav1.ListOptions{})
+	pods, err := c.PodsFromDeployment(ctx, namespace, testContainer)
 	require.NoError(err)
-	require.GreaterOrEqual(len(pods.Items), 1)
-	pod := pods.Items[0] // only one pod was deployed
+	require.Len(pods, 1)
+	pod := pods[0]
 
 	logs, err := c.Client.CoreV1().Pods(namespace).GetLogs(pod.Name, &corev1.PodLogOptions{}).DoRaw(ctx)
 	require.NoError(err)