From cdc34d5256db275c316bd426440c82ba7c9b8ca7 Mon Sep 17 00:00:00 2001
From: Takumi Akiyama <t.akiym@gmail.com>
Date: Thu, 29 Oct 2020 16:42:38 +0900
Subject: [PATCH] add athena:getQueryExecution policy for
 startQueryExecution.sync

---
 .../lib/athena/start-query-execution.ts                        | 2 +-
 .../test/athena/integ.get-query-execution.expected.json        | 3 ++-
 .../test/athena/integ.get-query-results.expected.json          | 3 ++-
 .../test/athena/integ.start-query-execution.expected.json      | 3 ++-
 .../test/athena/integ.stop-query-execution.expected.json       | 3 ++-
 5 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/packages/@aws-cdk/aws-stepfunctions-tasks/lib/athena/start-query-execution.ts b/packages/@aws-cdk/aws-stepfunctions-tasks/lib/athena/start-query-execution.ts
index 4cdd5f29fbc05..9450800fb6d0b 100644
--- a/packages/@aws-cdk/aws-stepfunctions-tasks/lib/athena/start-query-execution.ts
+++ b/packages/@aws-cdk/aws-stepfunctions-tasks/lib/athena/start-query-execution.ts
@@ -87,7 +87,7 @@ export class AthenaStartQueryExecution extends sfn.TaskStateBase {
           }),
 
         ],
-        actions: ['athena:getDataCatalog', 'athena:startQueryExecution'],
+        actions: ['athena:getDataCatalog', 'athena:startQueryExecution', 'athena:getQueryExecution'],
       }),
     ];
 
diff --git a/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.get-query-execution.expected.json b/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.get-query-execution.expected.json
index de3041464cf88..b83da65e87a2d 100644
--- a/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.get-query-execution.expected.json
+++ b/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.get-query-execution.expected.json
@@ -36,7 +36,8 @@
             {
               "Action": [
                 "athena:getDataCatalog",
-                "athena:startQueryExecution"
+                "athena:startQueryExecution",
+                "athena:getQueryExecution"
               ],
               "Effect": "Allow",
               "Resource": [
diff --git a/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.get-query-results.expected.json b/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.get-query-results.expected.json
index f593dc785feab..9718e2fa79df9 100644
--- a/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.get-query-results.expected.json
+++ b/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.get-query-results.expected.json
@@ -36,7 +36,8 @@
             {
               "Action": [
                 "athena:getDataCatalog",
-                "athena:startQueryExecution"
+                "athena:startQueryExecution",
+                "athena:getQueryExecution"
               ],
               "Effect": "Allow",
               "Resource": [
diff --git a/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.start-query-execution.expected.json b/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.start-query-execution.expected.json
index 4cff9888545d8..1f5ad33266295 100644
--- a/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.start-query-execution.expected.json
+++ b/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.start-query-execution.expected.json
@@ -36,7 +36,8 @@
             {
               "Action": [
                 "athena:getDataCatalog",
-                "athena:startQueryExecution"
+                "athena:startQueryExecution",
+                "athena:getQueryExecution"
               ],
               "Effect": "Allow",
               "Resource": [
diff --git a/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.stop-query-execution.expected.json b/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.stop-query-execution.expected.json
index 28aaf5d4ab243..a797335db041c 100644
--- a/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.stop-query-execution.expected.json
+++ b/packages/@aws-cdk/aws-stepfunctions-tasks/test/athena/integ.stop-query-execution.expected.json
@@ -36,7 +36,8 @@
             {
               "Action": [
                 "athena:getDataCatalog",
-                "athena:startQueryExecution"
+                "athena:startQueryExecution",
+                "athena:getQueryExecution"
               ],
               "Effect": "Allow",
               "Resource": [