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());
  }