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)