protected DLFileEntry getDLFileEntry(Object object) throws Exception { List<FileEntry> fileEntries = new ArrayList<>(); if (object instanceof MBMessage) { fileEntries = ((MBMessage) object).getAttachmentsFileEntries(0, 1); } Assert.assertFalse(fileEntries.isEmpty()); FileEntry fileEntry = fileEntries.get(0); return DLFileEntryLocalServiceUtil.getDLFileEntry(fileEntry.getFileEntryId()); }