From bde734dc751cbfd59b40c1c2f0d60229795fae4a Mon Sep 17 00:00:00 2001
From: Athanasios Voutsadakis
 <athanasios.voutsadakis@digital.cabinet-office.gov.uk>
Date: Thu, 16 Nov 2017 17:02:38 +0000
Subject: [PATCH] Use test_client() as context manager

---
 tests/app/main/test_request_header.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/tests/app/main/test_request_header.py b/tests/app/main/test_request_header.py
index e5dc6beaa..643d33c56 100644
--- a/tests/app/main/test_request_header.py
+++ b/tests/app/main/test_request_header.py
@@ -16,11 +16,11 @@ def test_route_correct_secret_key(app_, check_proxy_header, header_value, expect
         'CHECK_PROXY_HEADER': check_proxy_header,
     }):
 
-        client = app_.test_client()
-        response = client.get(
-            path='/_status?elb=True',
-            headers=[
-                ('X-Custom-forwarder', header_value),
-            ]
-        )
+        with app_.test_client() as client:
+            response = client.get(
+                path='/_status?elb=True',
+                headers=[
+                    ('X-Custom-forwarder', header_value),
+                ]
+            )
         assert response.status_code == expected_code