def test_process_file_doesnt_handle_file_if_doesnt_fit_in_buffer(self):
self.file_fits_in_buffer.return_value = False
with self.assertRaises(OutOfSpaceError):
process_file(self.file_entry, self.remote_driver)
def test_process_file_handles_file_if_filts_in_buffer(self):
process_file(self.file_entry, self.remote_driver)
self.handle_file.assert_called_once_with(self.file_entry, self.remote_driver)
