From 901be3a6d1643fb27712318f3d5d7edc0ea89916 Mon Sep 17 00:00:00 2001
From: gruebel <anton.gruebel@gmail.com>
Date: Fri, 20 Dec 2019 10:54:38 +0100
Subject: [PATCH] Remove unused import

---
 tests/test_codepipeline/test_codepipeline.py | 35 +++++++++++---------
 1 file changed, 19 insertions(+), 16 deletions(-)

diff --git a/tests/test_codepipeline/test_codepipeline.py b/tests/test_codepipeline/test_codepipeline.py
index d853aa3b2..926d7f873 100644
--- a/tests/test_codepipeline/test_codepipeline.py
+++ b/tests/test_codepipeline/test_codepipeline.py
@@ -1,5 +1,5 @@
 import json
-from datetime import datetime, timezone
+from datetime import datetime
 
 import boto3
 import sure  # noqa
@@ -857,18 +857,21 @@ def test_delete_pipeline():
 @mock_iam
 def get_role_arn():
     iam = boto3.client("iam", region_name="us-east-1")
-    return iam.create_role(
-        RoleName="test-role",
-        AssumeRolePolicyDocument=json.dumps(
-            {
-                "Version": "2012-10-17",
-                "Statement": [
-                    {
-                        "Effect": "Allow",
-                        "Principal": {"Service": "codepipeline.amazonaws.com"},
-                        "Action": "sts:AssumeRole",
-                    }
-                ],
-            }
-        ),
-    )["Role"]["Arn"]
+    try:
+        return iam.get_role(RoleName="test-role")["Role"]["Arn"]
+    except ClientError:
+        return iam.create_role(
+            RoleName="test-role",
+            AssumeRolePolicyDocument=json.dumps(
+                {
+                    "Version": "2012-10-17",
+                    "Statement": [
+                        {
+                            "Effect": "Allow",
+                            "Principal": {"Service": "codepipeline.amazonaws.com"},
+                            "Action": "sts:AssumeRole",
+                        }
+                    ],
+                }
+            ),
+        )["Role"]["Arn"]