Pop Age if it is there, otherwise ignore it

betamaxpy · Apr 7, 2016 · dbfd3d9 · dbfd3d9
1 parent f67b2cb
commit dbfd3d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/integration/test_record_modes.py b/tests/integration/test_record_modes.py
@@ -28,10 +28,10 @@ def test_replays_response_from_cassette(self):
             assert r1.status_code == 200
             r0_headers = r0.headers.copy()
             r0_headers.pop('Date')
-            r0_headers.pop('Age')
+            r0_headers.pop('Age', None)
             r1_headers = r1.headers.copy()
             r1_headers.pop('Date')
-            r1_headers.pop('Age')
+            r1_headers.pop('Age', None)
             # NOTE(sigmavirus24): This fails if the second request is
             # technically a second later. Ignoring the Date headers allows
             # this test to succeed.