/*
 * Copyright (C) 2021 eXo Platform SAS.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License
 * as published by the Free Software Foundation; either version 3
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see<http://www.gnu.org/licenses/>.
 */
package org.exoplatform.documents.service;

import static org.junit.Assert.*;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.*;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.*;

import org.junit.Before;
import org.junit.Test;

import org.exoplatform.commons.exception.ObjectNotFoundException;
import org.exoplatform.documents.constant.FileListingType;
import org.exoplatform.documents.listener.AttachmentsActivityCacheUpdater;
import org.exoplatform.documents.model.*;
import org.exoplatform.documents.rest.model.AbstractNodeEntity;
import org.exoplatform.documents.rest.model.FileNodeEntity;
import org.exoplatform.documents.rest.util.EntityBuilder;
import org.exoplatform.documents.storage.DocumentFileStorage;
import org.exoplatform.documents.storage.JCRDeleteFileStorage;
import org.exoplatform.services.listener.Event;
import org.exoplatform.services.listener.ListenerService;
import org.exoplatform.services.security.Authenticator;
import org.exoplatform.services.security.ConversationState;
import org.exoplatform.services.security.IdentityRegistry;
import org.exoplatform.services.thumbnail.ImageThumbnailService;
import org.exoplatform.social.core.identity.model.Identity;
import org.exoplatform.social.core.identity.model.Profile;
import org.exoplatform.social.core.identity.provider.OrganizationIdentityProvider;
import org.exoplatform.social.core.identity.provider.SpaceIdentityProvider;
import org.exoplatform.social.core.manager.IdentityManager;
import org.exoplatform.social.core.space.model.Space;
import org.exoplatform.social.core.space.spi.SpaceService;
import org.exoplatform.social.core.storage.api.ActivityStorage;
import org.exoplatform.social.core.storage.cache.CachedActivityStorage;

import io.meeds.analytics.api.service.AnalyticsService;

public class DocumentFileServiceTest {

  private DocumentFileStorage     documentFileStorage;

  private IdentityManager         identityManager;

  private SpaceService            spaceService;

  private IdentityRegistry        identityRegistry;

  private Authenticator           authenticator;

  private DocumentFileServiceImpl documentFileService;

  private JCRDeleteFileStorage    jcrDeleteFileStorage;

  private ActivityStorage         activityStorage;

  private ListenerService         listenerService;

  private ImageThumbnailService imageThumbnailService;

  private CachedActivityStorage   cachedActivityStorage;

  private AnalyticsService        analyticsService;

  private Identity                currentIdentity;

  String                          userName       = "testuser";

  long                            currentOwnerId = 2;

  @Before
  public void setUp() {
    spaceService = mock(SpaceService.class);
    identityManager = mock(IdentityManager.class);
    identityRegistry = mock(IdentityRegistry.class);
    authenticator = mock(Authenticator.class);
    documentFileStorage = mock(DocumentFileStorage.class);
    activityStorage = mock(ActivityStorage.class);
    listenerService = mock(ListenerService.class);
    jcrDeleteFileStorage = mock(JCRDeleteFileStorage.class);
    cachedActivityStorage = mock(CachedActivityStorage.class);
    analyticsService = mock(AnalyticsService.class);
    imageThumbnailService = mock(ImageThumbnailService.class);
    documentFileService = new DocumentFileServiceImpl(documentFileStorage,
                                                      jcrDeleteFileStorage,
                                                      authenticator,
                                                      spaceService,
                                                      identityManager,
                                                      identityRegistry,
                                                      listenerService,
                                                      analyticsService,
                                                      imageThumbnailService);

    currentIdentity = new Identity(OrganizationIdentityProvider.NAME, userName);
    currentIdentity.setId(String.valueOf(currentOwnerId));
    org.exoplatform.services.security.Identity userID = new org.exoplatform.services.security.Identity(userName);
    when(identityRegistry.getIdentity(userName)).thenReturn(userID);
    when(identityManager.getIdentity(currentOwnerId)).thenReturn(currentIdentity);
    when(identityManager.getOrCreateIdentity(eq(OrganizationIdentityProvider.NAME), eq(userName))).thenReturn(currentIdentity);
  }

  @Test
  public void testClearActivityCacheWhenFileRenamed() throws Exception {

    String documentID = "2";
    Event<String, String> renameFileEvent = new Event<>("rename_file_event", null, documentID);
    lenient().doNothing().when(documentFileStorage).renameDocument(anyLong(), anyString(), anyString(), any());
    documentFileService.renameDocument(1, documentID, "newName", currentOwnerId);
    cachedActivityStorage = mock(CachedActivityStorage.class);
    AttachmentsActivityCacheUpdater attachmentsActivityCacheUpdater = new AttachmentsActivityCacheUpdater(cachedActivityStorage);
    attachmentsActivityCacheUpdater.onEvent(renameFileEvent);
    verify(cachedActivityStorage, times(1)).clearActivityCachedByAttachmentId(documentID);
  }

  @Test
  public void testGetDocumentItems() throws Exception {
    Profile currentProfile = new Profile();
    currentProfile.setProperty(Profile.FULL_NAME, userName);
    currentIdentity.setProfile(currentProfile);

    org.exoplatform.services.security.Identity userID = new org.exoplatform.services.security.Identity(userName);
    DocumentTimelineFilter filter = null;
    Exception exception = assertThrows(IllegalArgumentException.class, () -> {
      documentFileService.getDocumentItems(FileListingType.TIMELINE, null, 0, 0, Long.valueOf(currentIdentity.getId()),false);
    });
    assertEquals(exception.getMessage(), "File filter is mandatory");

    filter = new DocumentTimelineFilter(0L, null);
    DocumentTimelineFilter finalFilter1 = filter;
    exception = assertThrows(IllegalArgumentException.class, () -> {
      documentFileService.getDocumentItems(FileListingType.TIMELINE, finalFilter1, 0, 0, Long.valueOf(currentIdentity.getId()),false);
    });
    assertEquals(exception.getMessage(), "OwnerId is mandatory");

    filter = new DocumentTimelineFilter(Long.valueOf(currentIdentity.getId()), null);
    DocumentTimelineFilter finalFilter = filter;
    exception = assertThrows(IllegalAccessException.class, () -> {
      documentFileService.getDocumentItems(FileListingType.TIMELINE, finalFilter, 0, 0, 0,false);
    });
    assertEquals(exception.getMessage(), "User Identity is mandatory");

    DocumentFolderFilter docFilter = new DocumentFolderFilter("", "", 0L, "");
    DocumentFolderFilter finalDocFilter = docFilter;
    exception = assertThrows(IllegalArgumentException.class, () -> {
      documentFileService.getDocumentItems(FileListingType.TIMELINE, finalDocFilter, 0, 0, Long.valueOf(currentIdentity.getId()),false);
    });
    assertEquals(exception.getMessage(), "filter must be an instance of DocumentTimelineFilter");

    exception = assertThrows(IllegalArgumentException.class, () -> {
      documentFileService.getDocumentItems(FileListingType.FOLDER, finalFilter, 0, 0, Long.valueOf(currentIdentity.getId()),false);
    });
    assertEquals(exception.getMessage(), "filter must be an instance of DocumentFolderFilter");

    exception = assertThrows(IllegalArgumentException.class, () -> {
      documentFileService.getDocumentItems(FileListingType.FOLDER, finalDocFilter, 0, 0, Long.valueOf(currentIdentity.getId()),false);
    });
    assertEquals(exception.getMessage(), "ParentFolderId or OwnerId is mandatory");

    when(identityRegistry.getIdentity(userName)).thenReturn(userID);
    when(identityManager.getIdentity(currentOwnerId)).thenReturn(currentIdentity);
    when(identityManager.getOrCreateIdentity(eq(OrganizationIdentityProvider.NAME), eq(userName))).thenReturn(currentIdentity);

    String spacePrettyName = "spacetest";
    currentIdentity.setRemoteId(spacePrettyName);
    Space space = new Space();
    org.exoplatform.services.security.Identity spaceID = new org.exoplatform.services.security.Identity(spacePrettyName);
    when(identityRegistry.getIdentity(spacePrettyName)).thenReturn(spaceID);
    when(spaceService.getSpaceByPrettyName(eq(spacePrettyName))).thenReturn(space);
    when(spaceService.hasAccessPermission(space, userName)).thenReturn(true);

    List<FileNode> files = new ArrayList<>();

    FileNode file1 = new FileNode();
    FileNode file2 = new FileNode();
    FileNode file3 = new FileNode();
    FileNode file4 = new FileNode();

    files.add(file1);
    files.add(file2);
    files.add(file3);
    files.add(file4);

    when(documentFileStorage.getFilesTimeline(filter, spaceID, 0, 0)).thenReturn(files);
    List<? extends AbstractNode> documentItems = documentFileService.getDocumentItems(FileListingType.TIMELINE, filter, 0, 0, Long.parseLong(currentIdentity.getId()),false);
    assertEquals(documentItems.size(), 4);
  }

  @Test
  public void testGetFilesTimeline() throws Exception { // NOSONAR

    String username = "testuser";
    long currentOwnerId = 2;
    Identity currentIdentity = new Identity(OrganizationIdentityProvider.NAME, username);
    currentIdentity.setId(String.valueOf(currentOwnerId));
    Profile currentProfile = new Profile();
    currentProfile.setProperty(Profile.FULL_NAME, username);
    currentIdentity.setProfile(currentProfile);

    org.exoplatform.services.security.Identity userID = new org.exoplatform.services.security.Identity(username);
    DocumentTimelineFilter filter = new DocumentTimelineFilter(Long.valueOf(currentIdentity.getId()), null);

    when(identityRegistry.getIdentity(username)).thenReturn(userID);
    when(identityManager.getIdentity(currentOwnerId)).thenReturn(currentIdentity);
    when(identityManager.getOrCreateIdentity(eq(OrganizationIdentityProvider.NAME), eq(username))).thenReturn(currentIdentity);

    DocumentTimelineFilter filter_ = new DocumentTimelineFilter(0L, null);
    DocumentTimelineFilter finalFilter1 = filter_;
    Exception exception = assertThrows(ObjectNotFoundException.class, () -> {
      documentFileService.getFilesTimeline(finalFilter1, 0, 0, Long.valueOf(currentIdentity.getId()));
    });
    assertEquals(exception.getMessage(), "Owner Identity with id : " + finalFilter1.getOwnerId() + " isn't found");

    String spacePrettyName = "spacetest";
    currentIdentity.setRemoteId(spacePrettyName);
    Space space = new Space();
    org.exoplatform.services.security.Identity spaceID = new org.exoplatform.services.security.Identity(spacePrettyName);
    when(identityRegistry.getIdentity(spacePrettyName)).thenReturn(spaceID);
    when(spaceService.getSpaceByPrettyName(eq(spacePrettyName))).thenReturn(space);
    when(spaceService.hasAccessPermission(space, username)).thenReturn(true);

    List<FileNode> files = new ArrayList<>();

    FileNode file1 = new FileNode();
    FileNode file2 = new FileNode();
    FileNode file3 = new FileNode();
    FileNode file4 = new FileNode();

    files.add(file1);
    files.add(file2);
    files.add(file3);
    files.add(file4);

    when(documentFileStorage.getFilesTimeline(filter, spaceID, 0, 0)).thenReturn(files);
    List<FileNode> files_ = new ArrayList<>();
    files_ = documentFileService.getFilesTimeline(filter, 0, 0, Long.valueOf(currentIdentity.getId()));
    assertEquals(files_.size(), 4);
  }

  @Test
  public void testGetGroupDocumentsCount() throws Exception { // NOSONAR

    String username = "testuser";
    long currentOwnerId = 2;
    Identity currentIdentity = new Identity(OrganizationIdentityProvider.NAME, username);
    currentIdentity.setId(String.valueOf(currentOwnerId));
    Profile currentProfile = new Profile();
    currentProfile.setProperty(Profile.FULL_NAME, username);
    currentIdentity.setProfile(currentProfile);

    org.exoplatform.services.security.Identity userID = new org.exoplatform.services.security.Identity(username);
    DocumentTimelineFilter filter = new DocumentTimelineFilter(Long.valueOf(currentIdentity.getId()), null);

    when(identityRegistry.getIdentity(username)).thenReturn(userID);
    when(identityManager.getIdentity(currentOwnerId)).thenReturn(currentIdentity);
    when(identityManager.getOrCreateIdentity(eq(OrganizationIdentityProvider.NAME), eq(username))).thenReturn(currentIdentity);

    DocumentTimelineFilter filter_ = new DocumentTimelineFilter(0L, null);
    DocumentTimelineFilter finalFilter1 = filter_;
    Exception exception = assertThrows(ObjectNotFoundException.class, () -> {
      documentFileService.getFilesTimeline(finalFilter1, 0, 0, Long.valueOf(currentIdentity.getId()));
    });
    assertEquals(exception.getMessage(), "Owner Identity with id : " + finalFilter1.getOwnerId() + " isn't found");

    String spacePrettyName = "spacetest";
    currentIdentity.setRemoteId(spacePrettyName);
    Space space = new Space();
    org.exoplatform.services.security.Identity spaceID = new org.exoplatform.services.security.Identity(spacePrettyName);
    when(identityRegistry.getIdentity(spacePrettyName)).thenReturn(spaceID);
    when(spaceService.getSpaceByPrettyName(eq(spacePrettyName))).thenReturn(space);
    when(spaceService.hasAccessPermission(space, username)).thenReturn(true);

    DocumentGroupsSize documentGroupsSize = new DocumentGroupsSize();
    documentGroupsSize.setThisDay(4);

    when(documentFileStorage.getGroupDocumentsCount(filter, spaceID)).thenReturn(documentGroupsSize);
    DocumentGroupsSize documentGroupsSize_ = new DocumentGroupsSize();
    documentGroupsSize_ = documentFileService.getGroupDocumentsCount(filter, Long.valueOf(currentIdentity.getId()));

    currentIdentity.setProviderId("space");
    when(spaceService.getSpaceByPrettyName(spacePrettyName)).thenReturn(space);
    when(spaceService.hasAccessPermission(space, spacePrettyName)).thenReturn(true);
    DocumentGroupsSize documentGroupsSize_2 = documentFileService.getGroupDocumentsCount(filter,
                                                                                         Long.valueOf(currentIdentity.getId()));

    assertEquals(4, documentGroupsSize_.getThisDay());
    assertEquals(4, documentGroupsSize_2.getThisDay());

  }

  @Test
  public void testGetFolderChildNodes() throws Exception { // NOSONAR
    String username = "testuser";
    long currentOwnerId = 2;
    long spaceIdentityId = 4;
    Identity currentIdentity = new Identity(OrganizationIdentityProvider.NAME, username);
    currentIdentity.setId(String.valueOf(currentOwnerId));
    String user2name = "testuser2";
    long user2Id = 3;
    Identity user2Identity = new Identity(OrganizationIdentityProvider.NAME, user2name);
    user2Identity.setId(String.valueOf(user2Id));
    Profile user2Profile = new Profile();
    user2Profile.setProperty(Profile.FULL_NAME, user2name);
    user2Identity.setProfile(user2Profile);

    org.exoplatform.services.security.Identity userID = new org.exoplatform.services.security.Identity(username);
    org.exoplatform.services.security.Identity user2ID = new org.exoplatform.services.security.Identity(user2name);

    when(identityRegistry.getIdentity(username)).thenReturn(userID);
    when(identityManager.getIdentity(currentOwnerId)).thenReturn(currentIdentity);
    when(identityManager.getOrCreateIdentity(eq(OrganizationIdentityProvider.NAME), eq(username))).thenReturn(currentIdentity);

    when(identityRegistry.getIdentity(user2name)).thenReturn(user2ID);
    when(identityManager.getIdentity(user2Id)).thenReturn(user2Identity);
    when(identityManager.getOrCreateIdentity(eq(OrganizationIdentityProvider.NAME), eq(user2name))).thenReturn(user2Identity);

    DocumentFolderFilter filter_ = new DocumentFolderFilter(null, null, 0L, "");
    DocumentFolderFilter finalFilter1 = filter_;
    Exception exception = assertThrows(ObjectNotFoundException.class, () -> {
      documentFileService.getFolderChildNodes(finalFilter1, 0, 0, Long.valueOf(currentIdentity.getId()));
    });
    assertEquals(exception.getMessage(), "Owner Identity with id : " + finalFilter1.getOwnerId() + " isn't found");

    String spacePrettyName = "spacetest";

    Space space = new Space();
    org.exoplatform.services.security.Identity spaceID = new org.exoplatform.services.security.Identity(spacePrettyName);

    Identity spaceIdentity = new Identity(OrganizationIdentityProvider.NAME, username);
    spaceIdentity.setId(String.valueOf(spaceIdentityId));
    spaceIdentity.setRemoteId(spacePrettyName);
    spaceIdentity.setProviderId(SpaceIdentityProvider.NAME);

    when(identityRegistry.getIdentity(spacePrettyName)).thenReturn(spaceID);
    when(identityManager.getIdentity(spaceIdentityId)).thenReturn(spaceIdentity);
    when(spaceService.getSpaceByPrettyName(eq(spacePrettyName))).thenReturn(space);
    when(spaceService.hasAccessPermission(space, username)).thenReturn(true);
    when(spaceService.hasAccessPermission(space, user2name)).thenReturn(false);

    filter_ = new DocumentFolderFilter(null, null, Long.valueOf(spaceIdentity.getId()), "");
    DocumentFolderFilter finalFilter2 = filter_;
    List<AbstractNode> files = new ArrayList<>();

    AbstractNode file1 = new FileNode();
    AbstractNode file2 = new FileNode();
    AbstractNode file3 = new FileNode();
    AbstractNode file4 = new FileNode();

    files.add(file1);
    files.add(file2);
    files.add(file3);
    files.add(file4);

    exception = assertThrows(IllegalAccessException.class, () -> {
      documentFileService.getFolderChildNodes(finalFilter2, 0, 0, Long.valueOf(user2Identity.getId()));
    });
    assertEquals(exception.getMessage(),
                 "User " + user2name + " attempts to access documents of space " + space.getDisplayName()
                     + "while it's not a member");

    when(documentFileStorage.getFolderChildNodes(finalFilter2, userID, 0, 0)).thenReturn(files);
    List<AbstractNode> files_ = new ArrayList<>();
    files_ = documentFileService.getFolderChildNodes(finalFilter2, 0, 0, Long.valueOf(currentIdentity.getId()));
    assertEquals(files_.size(), 4);
  }

  @Test
  public void testGetBreadCrumbs() throws Exception { // NOSONAR
    String username = "testuser";
    org.exoplatform.services.security.Identity root = new org.exoplatform.services.security.Identity(username);
    ConversationState.setCurrent(new ConversationState(root));
    long currentOwnerId = 2;
    Identity currentIdentity = new Identity(OrganizationIdentityProvider.NAME, username);
    currentIdentity.setId(String.valueOf(currentOwnerId));
    Profile currentProfile = new Profile();
    currentProfile.setProperty(Profile.FULL_NAME, username);
    currentIdentity.setProfile(currentProfile);

    org.exoplatform.services.security.Identity userID = new org.exoplatform.services.security.Identity(username);

    when(identityRegistry.getIdentity(username)).thenReturn(userID);
    when(identityManager.getIdentity(currentOwnerId)).thenReturn(currentIdentity);

    BreadCrumbItem breadCrumbItem1 = new BreadCrumbItem("1", "Folder1", "Folder1", "", false,new HashMap<>());
    BreadCrumbItem breadCrumbItem2 = new BreadCrumbItem("2", "Folder2", "Folder1", "", false,new HashMap<>());
    BreadCrumbItem breadCrumbItem3 = new BreadCrumbItem();
    breadCrumbItem3.setId("3");
    breadCrumbItem3.setName("Folder3");
    BreadCrumbItem breadCrumbItem4 = new BreadCrumbItem();
    breadCrumbItem4.setId("4");
    breadCrumbItem4.setName("Folder4");

    List<BreadCrumbItem> breadCrumbItems = new ArrayList<>();
    breadCrumbItems.add(breadCrumbItem1);
    breadCrumbItems.add(breadCrumbItem2);
    breadCrumbItems.add(breadCrumbItem3);
    breadCrumbItems.add(breadCrumbItem4);

    List<BreadCrumbItem> breadCrumbItems_ = new ArrayList<>();

    when(documentFileStorage.getBreadcrumb(2, "Folder1", "", userID)).thenReturn(breadCrumbItems);

    Exception exception = assertThrows(IllegalAccessException.class, () -> {
      documentFileService.getBreadcrumb(0, "", "", 0);
    });
    assertEquals(exception.getMessage(), "Can't find user identity with id 0");

    when(identityManager.getOrCreateUserIdentity(username)).thenReturn(currentIdentity);
    breadCrumbItems_ = documentFileService.getBreadcrumb(Long.valueOf(2), "Folder1", "", 2);
    assertEquals(breadCrumbItems_.size(), 4);
    assertEquals(breadCrumbItems_.get(0).getId(), "1");
    assertEquals(breadCrumbItems_.get(0).getName(), "Folder1");
  }

  @Test
  public void testUpdatePermissions() throws IllegalAccessException {
    NodePermission nodePermission = new NodePermission();
    Map<Long, String> toShare = new HashMap<>();
    toShare.put(1L, "read");
    nodePermission.setToShare(toShare);
    org.exoplatform.services.security.Identity identity = mock(org.exoplatform.services.security.Identity.class);
    Identity socialIdentity = mock(Identity.class);
    when(identityRegistry.getIdentity("username")).thenReturn(identity);
    when(socialIdentity.getRemoteId()).thenReturn("username");
    when(identityManager.getIdentity(1)).thenReturn(socialIdentity);
    documentFileService.updatePermissions("123", nodePermission, 1L);
    verify(documentFileStorage, times(1)).updatePermissions("123", nodePermission, identity);
    verify(documentFileStorage, times(1)).shareDocument("123", 1L,identity, false);
    //
    Map<Long, String> toNotify = new HashMap<>();
    toNotify.put(1L, "read");
    nodePermission.setToNotify(toNotify);
    documentFileService.updatePermissions("123", nodePermission, 1L);
    verify(documentFileStorage, atLeast(1)).updatePermissions("123", nodePermission, identity);
    verify(documentFileStorage, atLeast(1)).shareDocument("123", 1L,identity, true);

  }

  @Test
  public void testDeleteDocuments() throws IllegalAccessException {

    org.exoplatform.services.security.Identity identity = mock(org.exoplatform.services.security.Identity.class);
    Identity socialIdentity = mock(Identity.class);
    when(identityRegistry.getIdentity("username")).thenReturn(identity);
    when(socialIdentity.getRemoteId()).thenReturn("username");
    when(identityManager.getIdentity(1)).thenReturn(socialIdentity);
    FileNodeEntity file1 = new FileNodeEntity();
    file1.setId("1");
    file1.setName("oldFile");
    file1.setPath("/document/oldFile");
    file1.setDatasource("datasource");
    file1.setMimeType(":file");
    file1.setSize(50);
    FileNodeEntity file2 = new FileNodeEntity();
    file2.setId("2");
    file2.setName("oldFile2");
    file2.setPath("/document/oldFile2");
    file2.setDatasource("datasource");
    file2.setMimeType(":file");
    file2.setSize(50);
    FileNodeEntity file3 = new FileNodeEntity();
    file3.setId("3");
    file3.setName("oldFile3");
    file3.setPath("/document/oldFile3");
    file3.setDatasource("datasource");
    file3.setMimeType(":file");
    file3.setSize(50);

    List<AbstractNodeEntity> nodeEntities = new ArrayList<>();

    nodeEntities.add(file1);
    nodeEntities.add(file2);
    nodeEntities.add(file3);
    List<AbstractNode> nodes = EntityBuilder.toAbstractNodes(nodeEntities);

    documentFileService.deleteDocuments(123456, nodes, 1L);
    verify(jcrDeleteFileStorage, times(1)).deleteDocuments(123456, nodes, identity, 1L);
  }

  @Test
  public void testDownloadDocuments() throws IllegalAccessException {

    org.exoplatform.services.security.Identity identity = mock(org.exoplatform.services.security.Identity.class);
    Identity socialIdentity = mock(Identity.class);
    when(identityRegistry.getIdentity("username")).thenReturn(identity);
    when(socialIdentity.getRemoteId()).thenReturn("username");
    when(identityManager.getIdentity(1)).thenReturn(socialIdentity);
    FileNodeEntity file1 = new FileNodeEntity();
    file1.setId("1");
    file1.setName("oldFile");
    file1.setPath("/document/oldFile");
    file1.setDatasource("datasource");
    file1.setMimeType(":file");
    file1.setSize(50);
    FileNodeEntity file2 = new FileNodeEntity();
    file2.setId("2");
    file2.setName("oldFile2");
    file2.setPath("/document/oldFile2");
    file2.setDatasource("datasource");
    file2.setMimeType(":file");
    file2.setSize(50);
    FileNodeEntity file3 = new FileNodeEntity();
    file3.setId("3");
    file3.setName("oldFile3");
    file3.setPath("/document/oldFile3");
    file3.setDatasource("datasource");
    file3.setMimeType(":file");
    file3.setSize(50);

    List<AbstractNodeEntity> nodeEntities = new ArrayList<>();

    nodeEntities.add(file1);
    nodeEntities.add(file2);
    nodeEntities.add(file3);
    List<AbstractNode> nodes = EntityBuilder.toAbstractNodes(nodeEntities);
    doNothing().when(documentFileStorage).downloadDocuments(123456, nodes, identity, 1L);
    documentFileService.downloadDocuments(123456, nodes, 1L);
    verify(documentFileStorage, times(1)).downloadDocuments(123456, nodes, identity, 1L);
  }

  @Test
  public void testGetDownloadZipBytes() throws IOException {

    org.exoplatform.services.security.Identity identity = mock(org.exoplatform.services.security.Identity.class);
    Identity socialIdentity = mock(Identity.class);
    when(identityRegistry.getIdentity("username")).thenReturn(identity);
    when(socialIdentity.getRemoteId()).thenReturn("username");
    when(identityManager.getIdentity(1)).thenReturn(socialIdentity);

    when(documentFileStorage.getDownloadZipBytes(123456, "userName")).thenReturn(null);
    documentFileService.getDownloadZipBytes(123456,"userName");
    verify(documentFileStorage, times(1)).getDownloadZipBytes(123456, "userName");
  }

  @Test
  public void getAclUserIdentity() throws Exception {
    org.exoplatform.services.security.Identity identity = mock(org.exoplatform.services.security.Identity.class);
    when(identityRegistry.getIdentity("user")).thenReturn(null);
    when(authenticator.createIdentity("user")).thenReturn(identity);
    org.exoplatform.services.security.Identity aclIdentity = documentFileService.getAclUserIdentity("user");
    assertNotNull(aclIdentity);
    when(identityRegistry.getIdentity("user")).thenReturn(identity);
    org.exoplatform.services.security.Identity aclIdentity1 = documentFileService.getAclUserIdentity("user");
    assertNotNull(aclIdentity1);
  }

  @Test
  public void testCreateShortcut() throws Exception {
    String docId = "11111111";
    String docPath = "/Groups/spaces/test/Documents/test";
    documentFileService.createShortcut(docId, "/Groups/spaces/test/Documents/test", "user", "keepBoth");
    verify(documentFileStorage, times(1)).createShortcut(docId, docPath, "user", "keepBoth");
  }

  @Test
  public void getFileVersions() {
    Throwable exception =
                        assertThrows(IllegalArgumentException.class, () -> this.documentFileService.getFileVersions(null, null));
    assertEquals("file node id is mandatory", exception.getMessage());
    verify(documentFileStorage, times(0)).getFileVersions(anyString(), anyString());
    this.documentFileService.getFileVersions("5456456456", "user");
    verify(documentFileStorage, times(1)).getFileVersions("5456456456", "user");
  }

  @Test
  public void updateVersionSummary() {
    Throwable exception = assertThrows(IllegalArgumentException.class,
                                       () -> this.documentFileService.updateVersionSummary("123", null, "test", "user"));
    assertEquals("version id is mandatory", exception.getMessage());
    verify(documentFileStorage, times(0)).updateVersionSummary(anyString(), anyString(), anyString(), anyString());
    Throwable exception1 = assertThrows(IllegalArgumentException.class,
                                        () -> this.documentFileService.updateVersionSummary(null, "333", "test", "user"));
    assertEquals("original file id is mandatory", exception1.getMessage());
    verify(documentFileStorage, times(0)).updateVersionSummary(anyString(), anyString(), anyString(), anyString());
    this.documentFileService.updateVersionSummary("5456456456", "333erf", "new summary", "user");
    verify(documentFileStorage, times(1)).updateVersionSummary(anyString(), anyString(), anyString(), anyString());
  }

  @Test
  public void restoreVersion() {
    Throwable exception =
                        assertThrows(IllegalArgumentException.class, () -> this.documentFileService.restoreVersion(null, "user"));
    assertEquals("version id is mandatory", exception.getMessage());
    verify(documentFileStorage, times(0)).restoreVersion(anyString(), anyString());
    documentFileService.restoreVersion("123", "user");
    verify(documentFileStorage, times(1)).restoreVersion("123", "user");
  }

  @Test
  public void createNewVersion() {
    Throwable exception = assertThrows(IllegalArgumentException.class,
                                       () -> this.documentFileService.createNewVersion(null, null, null));
    assertEquals("node id is mandatory", exception.getMessage());
    exception = assertThrows(IllegalArgumentException.class, () -> this.documentFileService.createNewVersion("123", null, null));
    assertEquals("User identity id is mandatory", exception.getMessage());
    verify(documentFileStorage, times(0)).createNewVersion(anyString(), anyString(), any());
    InputStream newContent = new ByteArrayInputStream("test".getBytes());
    documentFileService.createNewVersion("123", "user", newContent);
    verify(documentFileStorage, times(1)).createNewVersion("123", "user", newContent);
  }

  @Test
  public void moveDocument() throws IllegalAccessException {
    org.exoplatform.services.security.Identity identity = mock(org.exoplatform.services.security.Identity.class);
    Identity userIdentity = mock(Identity.class);
    AbstractNode abstractNode = mock(AbstractNode.class);
    List<AbstractNode> documents = List.of(abstractNode);
    when(identityManager.getIdentity(anyLong())).thenReturn(userIdentity);
    when(userIdentity.getRemoteId()).thenReturn("user");
    when(identityRegistry.getIdentity("user")).thenReturn(identity);
    documentFileService.moveDocuments(1, 1L, documents, "destPath", 1L);
    verify(documentFileStorage, times(1)).moveDocuments(1, 1L, documents, "destPath", identity, 1L);
  }

  @Test
  public void getDocumentsSizeStats() throws Exception {
    org.exoplatform.services.security.Identity identity = mock(org.exoplatform.services.security.Identity.class);
    Exception exception = assertThrows(ObjectNotFoundException.class, () -> {
      documentFileService.getDocumentsSizeStat(1, 1);
    });
    assertEquals("Owner Identity with id : 1 isn't found", exception.getMessage());
    Identity ownerIdentity = mock(Identity.class);
    Identity currentUserIdentity = mock(Identity.class);
    when(identityManager.getIdentity(1)).thenReturn(ownerIdentity);
    when(identityManager.getIdentity(2)).thenReturn(currentUserIdentity);
    when(ownerIdentity.getRemoteId()).thenReturn("user");
    when(identityRegistry.getIdentity("user")).thenReturn(identity);
    when(ownerIdentity.isUser()).thenReturn(true);
    when(ownerIdentity.isSpace()).thenReturn(false);
    exception = assertThrows(IllegalAccessException.class, () -> {
      documentFileService.getDocumentsSizeStat(1, 2);
    });
    assertEquals("Current user with identity id : 2 attempts to get the size of private documents of user  with identity id  1",
                 exception.getMessage());
    when(ownerIdentity.isUser()).thenReturn(false);
    when(ownerIdentity.isSpace()).thenReturn(true);
    when(identity.getUserId()).thenReturn("user");
    when(currentUserIdentity.getRemoteId()).thenReturn("user");
    when(ownerIdentity.getRemoteId()).thenReturn("test1");
    Space space = new Space();
    space.setPrettyName("test1");
    space.setDisplayName("test1");
    when(spaceService.getSpaceByPrettyName("test1")).thenReturn(space);
    when(spaceService.hasAccessPermission(space, currentUserIdentity.getRemoteId())).thenReturn(false);
    exception = assertThrows(IllegalAccessException.class, () -> {
      documentFileService.getDocumentsSizeStat(1, 2);
    });
    assertEquals("Current user with identity id : 2 attempts to get size of documents of space with identity id 1 while it's not a member",
                 exception.getMessage());
    when(spaceService.hasAccessPermission(space, currentUserIdentity.getRemoteId())).thenReturn(true);
    documentFileService.getDocumentsSizeStat(1, 2);
    verify(analyticsService, times(1)).retrieveData(any());
  }

  @Test
  public void addDocumentsSizeStat() throws Exception {
    org.exoplatform.services.security.Identity identity = mock(org.exoplatform.services.security.Identity.class);
    Exception exception = assertThrows(ObjectNotFoundException.class, () -> {
      documentFileService.addDocumentsSizeStat(1, 1);
    });
    assertEquals("Owner Identity with id : 1 isn't found", exception.getMessage());
    Identity ownerIdentity = mock(Identity.class);
    Identity currentUserIdentity = mock(Identity.class);
    when(identityManager.getIdentity(1)).thenReturn(ownerIdentity);
    when(identityManager.getIdentity(2)).thenReturn(currentUserIdentity);
    when(ownerIdentity.getRemoteId()).thenReturn("user");
    when(identityRegistry.getIdentity("user")).thenReturn(identity);
    when(ownerIdentity.isUser()).thenReturn(true);
    when(ownerIdentity.isSpace()).thenReturn(false);
    exception = assertThrows(IllegalAccessException.class, () -> {
      documentFileService.addDocumentsSizeStat(1, 2);
    });
    assertEquals("Current user with identity id : 2 attempts to calculate the size of private documents of user  with identity id  1",
                 exception.getMessage());
    when(ownerIdentity.isUser()).thenReturn(false);
    when(ownerIdentity.isSpace()).thenReturn(true);
    when(identity.getUserId()).thenReturn("user");
    when(currentUserIdentity.getRemoteId()).thenReturn("user");
    when(ownerIdentity.getRemoteId()).thenReturn("test1");
    Space space = new Space();
    space.setPrettyName("test1");
    space.setDisplayName("test1");
    when(spaceService.getSpaceByPrettyName("test1")).thenReturn(space);
    when(spaceService.hasAccessPermission(space, currentUserIdentity.getRemoteId())).thenReturn(false);
    exception = assertThrows(IllegalAccessException.class, () -> {
      documentFileService.addDocumentsSizeStat(1, 2);
    });
    assertEquals("Current user with identity id : 2 attempts to calculate size of documents of space with identity id 1 while it's not a member",
                 exception.getMessage());
    when(spaceService.hasAccessPermission(space, currentUserIdentity.getRemoteId())).thenReturn(true);
    documentFileService.addDocumentsSizeStat(1, 2);
    verify(documentFileStorage, times(1)).calculateFilesSize(1L, identity);
  }

  @Test
  public void hasEditPermissionOnDocument() throws IllegalAccessException {
    org.exoplatform.services.security.Identity identity = mock(org.exoplatform.services.security.Identity.class);
    Identity userIdentity = mock(Identity.class);
    when(identityManager.getIdentity(anyLong())).thenReturn(userIdentity);
    when(userIdentity.getRemoteId()).thenReturn("user");
    when(identityRegistry.getIdentity("user")).thenReturn(identity);
    when(documentFileStorage.hasEditPermissions("123", identity)).thenReturn(true, false);
    assertTrue(documentFileService.hasEditPermissionOnDocument("123", 1L));
    assertFalse(documentFileService.hasEditPermissionOnDocument("123", 1L));
  }

  @Test
  public void importDocuments() throws Exception {
    org.exoplatform.services.security.Identity identity = mock(org.exoplatform.services.security.Identity.class);
    Exception exception = assertThrows(ObjectNotFoundException.class, () -> {
      documentFileService.importFiles("1", null, null, "1", "ignore", identity, 1L);
    });
    assertEquals("Owner Identity with id : 1 isn't found", exception.getMessage());
    Identity ownerIdentity = mock(Identity.class);
    when(identityManager.getIdentity(anyLong())).thenReturn(ownerIdentity);
    when(ownerIdentity.getRemoteId()).thenReturn("user");
    when(identityRegistry.getIdentity("user")).thenReturn(identity);
    documentFileService.importFiles("1", null, null, "1", "ignore", identity, 1L);
    verify(documentFileStorage, times(1)).importFiles("1", null, "", null, null, "ignore", identity, "1", 1L);
    when(ownerIdentity.isUser()).thenReturn(true);
    when(ownerIdentity.getRemoteId()).thenReturn("test");
    when(identity.getUserId()).thenReturn("test1");
    exception = assertThrows(IllegalAccessException.class, () -> {
      documentFileService.importFiles("1", null, null, "1", "ignore", identity, 1L);
    });
    assertEquals("User test1 attempts to access private documents of user test", exception.getMessage());
    when(ownerIdentity.isSpace()).thenReturn(true);
    when(identity.getUserId()).thenReturn("user");
    when(ownerIdentity.getRemoteId()).thenReturn("test1");
    Space space = new Space();
    space.setPrettyName("test1");
    space.setDisplayName("test1");
    when(spaceService.getSpaceByPrettyName("test1")).thenReturn(space);
    when(spaceService.hasAccessPermission(space, identity.getUserId())).thenReturn(false);
    exception = assertThrows(IllegalAccessException.class, () -> {
      documentFileService.importFiles("1", null, null, "1", "ignore", identity, 1L);
    });
    assertEquals("User user attempts to access documents of space test1 while it's not a member", exception.getMessage());
  }

  @Test
  public void getDeletedDocumentsTest() throws Exception {
    TrashElementNode trashElementNode = mock(TrashElementNode.class);
    List<TrashElementNode> trashElementNodes = Arrays.asList(trashElementNode);
    when(jcrDeleteFileStorage.getDeletedDocuments(any(TrashElementNodeFilter.class))).thenReturn(trashElementNodes);

    assertEquals(1, documentFileService.getDeletedDocuments(new TrashElementNodeFilter()).size());
  }

}
