Examples with TestDataverseEngine edu.harvard.iq.dataverse.engine.TestDataverseEngine used on opensource projects

Example 6 with TestDataverseEngine

use of edu.harvard.iq.dataverse.engine.TestDataverseEngine in project dataverse by IQSS.

the class RestrictFileCommandTest method setUp.

@Before
public void setUp() {
    dataset = makeDataset();
    file = makeDataFile();
    engine = new TestDataverseEngine(new TestCommandContext() {

        @Override
        public SettingsServiceBean settings() {
            return new SettingsServiceBean() {

                // override for a public install,
                // assume false
                @Override
                public boolean isTrueForKey(SettingsServiceBean.Key key, boolean defaultValue) {
                    return publicInstall;
                }
            };
        }
    });
}

Example 7 with TestDataverseEngine

use of edu.harvard.iq.dataverse.engine.TestDataverseEngine in project dataverse by IQSS.

the class UpdatePermissionRootCommandTest method testNoChange.

@Test
public void testNoChange() throws CommandException {
    Dataverse dv = MocksFactory.makeDataverse();
    DataverseEngine ngn = new TestDataverseEngine(testCommandContext);
    dv.setPermissionRoot(false);
    UpdatePermissionRootCommand sut = new UpdatePermissionRootCommand(false, MocksFactory.makeRequest(), dv);
    Dataverse result = ngn.submit(sut);
    assertFalse(result.isPermissionRoot());
    assertFalse(serviceBeanCalled);
    dv.setPermissionRoot(true);
    sut = new UpdatePermissionRootCommand(true, MocksFactory.makeRequest(), dv);
    result = ngn.submit(sut);
    assertTrue(result.isPermissionRoot());
    assertFalse(serviceBeanCalled);
}

Example 8 with TestDataverseEngine

use of edu.harvard.iq.dataverse.engine.TestDataverseEngine in project dataverse by IQSS.

the class UpdatePermissionRootCommandTest method testChange.

@Test
public void testChange() throws CommandException {
    Dataverse dv = MocksFactory.makeDataverse();
    DataverseEngine ngn = new TestDataverseEngine(testCommandContext);
    dv.setPermissionRoot(false);
    UpdatePermissionRootCommand sut = new UpdatePermissionRootCommand(true, MocksFactory.makeRequest(), dv);
    Dataverse result = ngn.submit(sut);
    assertTrue(result.isPermissionRoot());
    assertTrue(serviceBeanCalled);
    dv.setPermissionRoot(true);
    sut = new UpdatePermissionRootCommand(false, MocksFactory.makeRequest(), dv);
    result = ngn.submit(sut);
    assertFalse(result.isPermissionRoot());
    assertTrue(serviceBeanCalled);
}

Example 9 with TestDataverseEngine

use of edu.harvard.iq.dataverse.engine.TestDataverseEngine in project dataverse by IQSS.

the class DeletePrivateUrlCommandTest method setUp.

@Before
public void setUp() {
    testEngine = new TestDataverseEngine(new TestCommandContext() {

        @Override
        public PrivateUrlServiceBean privateUrl() {
            return new PrivateUrlServiceBean() {

                @Override
                public PrivateUrl getPrivateUrlFromDatasetId(long datasetId) {
                    if (datasetId == noPrivateUrlToDelete) {
                        return null;
                    } else if (datasetId == hasPrivateUrlToDelete) {
                        Dataset dataset = new Dataset();
                        dataset.setId(hasPrivateUrlToDelete);
                        String token = null;
                        PrivateUrlUser privateUrlUser = new PrivateUrlUser(datasetId);
                        RoleAssignment roleAssignment = new RoleAssignment(null, privateUrlUser, dataset, token);
                        return new PrivateUrl(roleAssignment, dataset, "FIXME");
                    } else {
                        return null;
                    }
                }
            };
        }

        @Override
        public DataverseRoleServiceBean roles() {
            return new DataverseRoleServiceBean() {

                @Override
                public List<RoleAssignment> directRoleAssignments(RoleAssignee roas, DvObject dvo) {
                    RoleAssignment roleAssignment = new RoleAssignment();
                    List<RoleAssignment> list = new ArrayList<>();
                    list.add(roleAssignment);
                    return list;
                }

                @Override
                public void revoke(RoleAssignment ra) {
                // no-op
                }
            };
        }
    });
}

Example 10 with TestDataverseEngine

use of edu.harvard.iq.dataverse.engine.TestDataverseEngine in project dataverse by IQSS.

the class MoveDatasetCommandTest method setUp.

@Before
public void setUp() {
    auth = makeAuthenticatedUser("Super", "User");
    auth.setSuperuser(true);
    nobody = makeAuthenticatedUser("Nick", "Nobody");
    nobody.setSuperuser(false);
    root = new Dataverse();
    root.setName("root");
    root.setId(1l);
    root.setPublicationDate(new Timestamp(new Date().getTime()));
    childA = new Dataverse();
    childA.setName("childA");
    childA.setId(2l);
    childA.setPublicationDate(new Timestamp(new Date().getTime()));
    childB = new Dataverse();
    childB.setName("childB");
    childB.setId(3l);
    childB.setPublicationDate(new Timestamp(new Date().getTime()));
    grandchildAA = new Dataverse();
    grandchildAA.setName("grandchildAA");
    grandchildAA.setId(4l);
    grandchildAA.setPublicationDate(new Timestamp(new Date().getTime()));
    childDraft = new Dataverse();
    childDraft.setName("childDraft");
    childDraft.setId(5l);
    grandchildBB = new Dataverse();
    grandchildBB.setName("grandchildBB");
    grandchildBB.setId(6l);
    grandchildBB.setPublicationDate(new Timestamp(new Date().getTime()));
    moved = new Dataset();
    moved.setOwner(root);
    moved.setPublicationDate(new Timestamp(new Date().getTime()));
    moved.setId(1l);
    movedResponses = new Dataset();
    movedResponses.setOwner(root);
    movedResponses.setPublicationDate(new Timestamp(new Date().getTime()));
    movedResponses.setId(2l);
    childA.setOwner(root);
    childB.setOwner(root);
    grandchildAA.setOwner(childA);
    grandchildBB.setOwner(childA);
    childDraft.setOwner(childA);
    gbA = new Guestbook();
    gbA.setId(1l);
    gbB = new Guestbook();
    gbB.setId(2l);
    gbC = new Guestbook();
    gbC.setId(3l);
    moved.setGuestbook(gbA);
    movedResponses.setGuestbook(gbA);
    GuestbookResponse gbResp = new GuestbookResponse();
    gbResp.setGuestbook(gbA);
    gbResp.setDataset(movedResponses);
    List<Guestbook> includeA = new ArrayList();
    includeA.add(gbA);
    includeA.add(gbB);
    grandchildAA.setGuestbooks(includeA);
    List<Guestbook> notIncludeA = new ArrayList();
    notIncludeA.add(gbC);
    notIncludeA.add(gbB);
    childB.setGuestbooks(notIncludeA);
    List<Guestbook> none = new ArrayList();
    root.setGuestbooks(none);
    grandchildBB.setGuestbooks(none);
    grandchildBB.setGuestbookRoot(false);
    childA.setGuestbooks(includeA);
    testEngine = new TestDataverseEngine(new TestCommandContext() {

        @Override
        public DataverseServiceBean dataverses() {
            return new DataverseServiceBean() {

                @Override
                public Dataverse save(Dataverse dataverse) {
                    // no-op. The superclass accesses databases which we don't have.
                    return dataverse;
                }
            };
        }

        @Override
        public GuestbookServiceBean guestbooks() {
            return new GuestbookServiceBean() {

                @Override
                public Long findCountResponsesForGivenDataset(Long guestbookId, Long datasetId) {
                    // We're going to fake a response for a dataset with responses
                    if (datasetId == 1) {
                        return new Long(0);
                    } else {
                        return new Long(1);
                    }
                }
            };
        }

        @Override
        public IndexServiceBean index() {
            return new IndexServiceBean() {

                @Override
                public Future<String> indexDataset(Dataset dataset, boolean doNormalSolrDocCleanUp) {
                    return null;
                }
            };
        }

        @Override
        public EntityManager em() {
            return new MockEntityManager() {
            };
        }
    });
}