Fixed the tests to work with GlacierResponse

2 files changed, 4 insertions, 2 deletions

diff --git a/tests/unit/__init__.py b/tests/unit/__init__.py
index 8c9a5ec3..4e52b766 100644
--- a/tests/unit/__init__.py
+++ b/tests/unit/__init__.py
@@ -45,10 +45,12 @@ class AWSMockServiceTestCase(unittest.TestCase):
         response.reason = reason
 
         response.getheaders.return_value = header
-        def overwrite_header(arg):
+        def overwrite_header(arg, default=None):
             header_dict = dict(header)
             if header_dict.has_key(arg):
                 return header_dict[arg]
+            else:
+                return default
         response.getheader.side_effect = overwrite_header
         
         return response
diff --git a/tests/unit/glacier/test_layer1.py b/tests/unit/glacier/test_layer1.py
index afad4238..7d7b6fc2 100644
--- a/tests/unit/glacier/test_layer1.py
+++ b/tests/unit/glacier/test_layer1.py
@@ -75,4 +75,4 @@ class GlacierJobOperations(GlacierLayer1ConnectionBase):
                                body=self.job_content)
         response = self.service_connection.get_job_output(self.vault_name,
                                                          'example-job-id')
-        self.assertEqual(self.job_content, response['Response'])
+        self.assertEqual(self.job_content, response.read())