From 9404623248f0d7b9ab956ab40103cfee4838f1d2 Mon Sep 17 00:00:00 2001
From: Matthew Wong <mattwon@amazon.com>
Date: Fri, 4 Jun 2021 10:50:30 -0700
Subject: [PATCH 1/2] Update driver capabilities

---
 tests/e2e-kubernetes/manifests.yaml | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tests/e2e-kubernetes/manifests.yaml b/tests/e2e-kubernetes/manifests.yaml
index e6d302be64..cdee83b1fe 100644
--- a/tests/e2e-kubernetes/manifests.yaml
+++ b/tests/e2e-kubernetes/manifests.yaml
@@ -22,8 +22,10 @@ DriverInfo:
     fsGroup: true
     block: true
     exec: true
-    volumeLimits: false
-    controllerExpansion: true
-    nodeExpansion: true
     snapshotDataSource: true
+    pvcDataSource: false
     multipods: true
+    controllerExpansion: true
+    nodeExpansion: true
+    volumeLimits: true
+    topology: true

From b7876b7fd0ddda15a6848aa570515ff339887a70 Mon Sep 17 00:00:00 2001
From: Matthew Wong <mattwon@amazon.com>
Date: Fri, 4 Jun 2021 10:53:55 -0700
Subject: [PATCH 2/2] Create nodes in all zones to exercise more topology tests

---
 Makefile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Makefile b/Makefile
index 61a014d411..db65078bc2 100644
--- a/Makefile
+++ b/Makefile
@@ -93,7 +93,7 @@ test-e2e-multi-az:
 .PHONY: test-e2e-migration
 test-e2e-migration:
 	AWS_REGION=us-west-2 \
-	AWS_AVAILABILITY_ZONES=us-west-2a \
+	AWS_AVAILABILITY_ZONES=us-west-2a,us-west-2b,us-west-2c \
 	TEST_PATH=./tests/e2e-kubernetes/... \
 	GINKGO_FOCUS="\[ebs-csi-migration\]" \
 	EBS_CHECK_MIGRATION=true \
@@ -102,7 +102,7 @@ test-e2e-migration:
 .PHONY: test-e2e-external
 test-e2e-external:
 	AWS_REGION=us-west-2 \
-	AWS_AVAILABILITY_ZONES=us-west-2a \
+	AWS_AVAILABILITY_ZONES=us-west-2a,us-west-2b,us-west-2c \
 	TEST_PATH=./tests/e2e-kubernetes/... \
 	GINKGO_FOCUS="External.Storage" \
 	GINKGO_SKIP="\[Disruptive\]|\[Serial\]" \