Add more assertions around the mocked file

sigmavirus24 · sigmavirus24 · commit fe0e027fe286 · 2014-04-19T15:28:50.000-05:00
diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py
@@ -72,6 +72,8 @@ def test_opens_a_new_file(self, mocked_open, response):
             stream_response_to_file(response, 'some_file')
 
         mocked_open.assert_called_once_with('some_file', 'wb')
+        mocked_open().write.assert_called_once_with(b'fake data')
+        mocked_open().close.assert_called_once_with()
 
     def test_uses_existing_file(self, response):
         fd = OpenFile()
@@ -84,3 +86,5 @@ def test_finds_filename_in_headers(self, mocked_open, response):
             stream_response_to_file(response)
 
         mocked_open.assert_called_once_with('a_file_name', 'wb')
+        mocked_open().write.assert_called_once_with(b'fake data')
+        mocked_open().close.assert_called_once_with()
