def test_delete_zip_file(self): with open(self.download_location, "wb") as zf: zf.write(self.zip_file.getvalue()) delete_zip_file(self.download_location) self.assertFalse(os.path.exists(self.download_location))
def test_delete_zip_file_error(self, mock_remove, mock_logger): e = "Error description" mock_remove.side_effect = Exception(e) error_message = "Failed to remove zip file {}, error {}".format(self.download_location, e) delete_zip_file(self.download_location) mock_logger.assert_called_with(error_message)