@Before
 public void setUp() {
   testWorkspace =
       session.createDocumentModel("/default-domain/workspaces", "testWorkspace", "Workspace");
   testWorkspace = session.createDocument(testWorkspace);
   // Create a list of test documents
   listDocuments = createTestFiles(session, 5);
   // Add them in the favorites
   for (DocumentModel doc : listDocuments) {
     favoritesManager.addToFavorites(doc, session);
   }
 }
  @Test
  public void addRemoveToFavoritesTest() {
    DocumentModel testWorkspace =
        session.createDocumentModel("/default-domain/workspaces", "testWorkspace", "Workspace");
    testWorkspace = session.createDocument(testWorkspace);
    DocumentModel testFile =
        session.createDocumentModel(testWorkspace.getPathAsString(), TEST_FILE_NAME, "File");
    testFile = session.createDocument(testFile);

    favoritesManager.addToFavorites(testFile, session);

    testFile = session.getDocument(testFile.getRef());

    assertTrue(favoritesManager.isFavorite(testFile, session));

    favoritesManager.removeFromFavorites(testFile, session);

    testFile = session.getDocument(testFile.getRef());

    assertFalse(favoritesManager.isFavorite(testFile, session));
  }