Search in sources :

Example 16 with Workspace

use of org.pmiops.workbench.model.Workspace in project workbench by all-of-us.

the class WorkspacesControllerTest method testCloneWorkspaceBadRequest.

@Test(expected = BadRequestException.class)
public void testCloneWorkspaceBadRequest() throws Exception {
    Workspace workspace = createDefaultWorkspace();
    workspace = workspacesController.createWorkspace(workspace).getBody();
    stubGetWorkspace(workspace.getNamespace(), workspace.getName(), LOGGED_IN_USER_EMAIL, WorkspaceAccessLevel.OWNER);
    CloneWorkspaceRequest req = new CloneWorkspaceRequest();
    Workspace modWorkspace = new Workspace();
    modWorkspace.setName("cloned");
    modWorkspace.setNamespace("cloned-ns");
    req.setWorkspace(modWorkspace);
    // Missing research purpose.
    workspacesController.cloneWorkspace(workspace.getNamespace(), workspace.getId(), req);
}
Also used : CloneWorkspaceRequest(org.pmiops.workbench.model.CloneWorkspaceRequest) Workspace(org.pmiops.workbench.model.Workspace) DataJpaTest(org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest) Test(org.junit.Test)

Example 17 with Workspace

use of org.pmiops.workbench.model.Workspace in project workbench by all-of-us.

the class WorkspacesControllerTest method testCloneWorkspaceWithCohorts.

@Test
public void testCloneWorkspaceWithCohorts() throws Exception {
    CdrVersionContext.setCdrVersion(cdrVersion);
    Workspace workspace = createDefaultWorkspace();
    workspace = workspacesController.createWorkspace(workspace).getBody();
    Cohort c1 = createDefaultCohort("c1");
    c1 = cohortsController.createCohort(workspace.getNamespace(), workspace.getId(), c1).getBody();
    Cohort c2 = createDefaultCohort("c2");
    c2 = cohortsController.createCohort(workspace.getNamespace(), workspace.getId(), c2).getBody();
    stubBigQueryCohortCalls();
    CreateReviewRequest reviewReq = new CreateReviewRequest();
    reviewReq.setSize(1);
    CohortReview cr1 = cohortReviewController.createCohortReview(workspace.getNamespace(), workspace.getId(), c1.getId(), cdrVersion.getCdrVersionId(), reviewReq).getBody();
    reviewReq.setSize(2);
    CohortReview cr2 = cohortReviewController.createCohortReview(workspace.getNamespace(), workspace.getId(), c2.getId(), cdrVersion.getCdrVersionId(), reviewReq).getBody();
    stubGetWorkspace(workspace.getNamespace(), workspace.getName(), LOGGED_IN_USER_EMAIL, WorkspaceAccessLevel.OWNER);
    CloneWorkspaceRequest req = new CloneWorkspaceRequest();
    Workspace modWorkspace = new Workspace();
    modWorkspace.setName("cloned");
    modWorkspace.setNamespace("cloned-ns");
    ResearchPurpose modPurpose = new ResearchPurpose();
    modPurpose.setAncestry(true);
    modWorkspace.setResearchPurpose(modPurpose);
    req.setWorkspace(modWorkspace);
    stubGetWorkspace(modWorkspace.getNamespace(), modWorkspace.getName(), LOGGED_IN_USER_EMAIL, WorkspaceAccessLevel.OWNER);
    Workspace cloned = workspacesController.cloneWorkspace(workspace.getNamespace(), workspace.getId(), req).getBody().getWorkspace();
    List<Cohort> cohorts = cohortsController.getCohortsInWorkspace(cloned.getNamespace(), cloned.getId()).getBody().getItems();
    Map<String, Cohort> cohortsByName = Maps.uniqueIndex(cohorts, c -> c.getName());
    assertThat(cohortsByName.keySet()).containsExactlyElementsIn(ImmutableSet.of("c1", "c2"));
    assertThat(cohorts.stream().map(c -> c.getId()).collect(Collectors.toList())).containsNoneOf(c1.getId(), c2.getId());
    CohortReview gotCr1 = cohortReviewController.getParticipantCohortStatuses(cloned.getNamespace(), cloned.getId(), cohortsByName.get("c1").getId(), cdrVersion.getCdrVersionId(), new ParticipantCohortStatuses().pageFilterType(PageFilterType.PARTICIPANTCOHORTSTATUSES)).getBody();
    assertThat(gotCr1.getReviewSize()).isEqualTo(cr1.getReviewSize());
    assertThat(gotCr1.getParticipantCohortStatuses()).isEqualTo(cr1.getParticipantCohortStatuses());
    CohortReview gotCr2 = cohortReviewController.getParticipantCohortStatuses(cloned.getNamespace(), cloned.getId(), cohortsByName.get("c2").getId(), cdrVersion.getCdrVersionId(), new ParticipantCohortStatuses().pageFilterType(PageFilterType.PARTICIPANTCOHORTSTATUSES)).getBody();
    assertThat(gotCr2.getReviewSize()).isEqualTo(cr2.getReviewSize());
    assertThat(gotCr2.getParticipantCohortStatuses()).isEqualTo(cr2.getParticipantCohortStatuses());
    assertThat(ImmutableSet.of(gotCr1.getCohortReviewId(), gotCr2.getCohortReviewId())).containsNoneOf(cr1.getCohortReviewId(), cr2.getCohortId());
}
Also used : DirtiesContext(org.springframework.test.annotation.DirtiesContext) FieldValue(com.google.cloud.bigquery.FieldValue) Provider(javax.inject.Provider) ReviewTabQueryBuilder(org.pmiops.workbench.cohortreview.ReviewTabQueryBuilder) FireCloudService(org.pmiops.workbench.firecloud.FireCloudService) GenderRaceEthnicityType(org.pmiops.workbench.cdr.cache.GenderRaceEthnicityType) Autowired(org.springframework.beans.factory.annotation.Autowired) BlobId(com.google.cloud.storage.BlobId) ShareWorkspaceResponse(org.pmiops.workbench.model.ShareWorkspaceResponse) CohortReviewServiceImpl(org.pmiops.workbench.cohortreview.CohortReviewServiceImpl) PageFilterType(org.pmiops.workbench.model.PageFilterType) DataJpaTest(org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest) TestConfiguration(org.springframework.boot.test.context.TestConfiguration) Blob(com.google.cloud.storage.Blob) GenderRaceEthnicityConcept(org.pmiops.workbench.cdr.cache.GenderRaceEthnicityConcept) org.pmiops.workbench.db.dao(org.pmiops.workbench.db.dao) ResearchPurpose(org.pmiops.workbench.model.ResearchPurpose) Gson(com.google.gson.Gson) Map(java.util.Map) WorkspaceACLUpdate(org.pmiops.workbench.firecloud.model.WorkspaceACLUpdate) CloudStorageService(org.pmiops.workbench.google.CloudStorageService) WorkspaceACLUpdateResponseList(org.pmiops.workbench.firecloud.model.WorkspaceACLUpdateResponseList) SpringRunner(org.springframework.test.context.junit4.SpringRunner) User(org.pmiops.workbench.db.model.User) MockBean(org.springframework.boot.test.mock.mockito.MockBean) ImmutableSet(com.google.common.collect.ImmutableSet) ImmutableMap(com.google.common.collect.ImmutableMap) Truth.assertWithMessage(com.google.common.truth.Truth.assertWithMessage) Timestamp(java.sql.Timestamp) CdrVersion(org.pmiops.workbench.db.model.CdrVersion) Cohort(org.pmiops.workbench.model.Cohort) ConflictException(org.pmiops.workbench.exceptions.ConflictException) Instant(java.time.Instant) Collectors(java.util.stream.Collectors) ZoneId(java.time.ZoneId) UpdateWorkspaceRequest(org.pmiops.workbench.model.UpdateWorkspaceRequest) List(java.util.List) CohortMaterializationService(org.pmiops.workbench.cohorts.CohortMaterializationService) AutoConfigureTestDatabase(org.springframework.boot.test.autoconfigure.jdbc.AutoConfigureTestDatabase) CreateReviewRequest(org.pmiops.workbench.model.CreateReviewRequest) WorkspaceAccessLevel(org.pmiops.workbench.model.WorkspaceAccessLevel) ResearchPurposeReviewRequest(org.pmiops.workbench.model.ResearchPurposeReviewRequest) QueryResult(com.google.cloud.bigquery.QueryResult) CloneWorkspaceRequest(org.pmiops.workbench.model.CloneWorkspaceRequest) ShareWorkspaceRequest(org.pmiops.workbench.model.ShareWorkspaceRequest) TestCase.assertEquals(junit.framework.TestCase.assertEquals) UserRole(org.pmiops.workbench.model.UserRole) ParticipantCohortStatuses(org.pmiops.workbench.model.ParticipantCohortStatuses) Mock(org.mockito.Mock) Workspace(org.pmiops.workbench.model.Workspace) RunWith(org.junit.runner.RunWith) ApiException(org.pmiops.workbench.firecloud.ApiException) HashMap(java.util.HashMap) LiquibaseAutoConfiguration(org.springframework.boot.autoconfigure.liquibase.LiquibaseAutoConfiguration) Matchers.anyString(org.mockito.Matchers.anyString) ArrayList(java.util.ArrayList) SearchRequests(org.pmiops.workbench.test.SearchRequests) ParticipantCounter(org.pmiops.workbench.cohortbuilder.ParticipantCounter) ImmutableList(com.google.common.collect.ImmutableList) CohortReview(org.pmiops.workbench.model.CohortReview) Propagation(org.springframework.transaction.annotation.Propagation) FailedPreconditionException(org.pmiops.workbench.exceptions.FailedPreconditionException) Qualifier(org.springframework.beans.factory.annotation.Qualifier) Before(org.junit.Before) CdrVersionContext(org.pmiops.workbench.cdr.CdrVersionContext) BadRequestException(org.pmiops.workbench.exceptions.BadRequestException) Iterator(java.util.Iterator) ClassMode(org.springframework.test.annotation.DirtiesContext.ClassMode) Matchers.anyListOf(org.mockito.Matchers.anyListOf) TestCase.fail(junit.framework.TestCase.fail) Import(org.springframework.context.annotation.Import) Test(org.junit.Test) FileDetail(org.pmiops.workbench.model.FileDetail) Truth.assertThat(com.google.common.truth.Truth.assertThat) Maps(com.google.common.collect.Maps) FakeClock(org.pmiops.workbench.test.FakeClock) Mockito(org.mockito.Mockito) DataAccessLevel(org.pmiops.workbench.model.DataAccessLevel) ForbiddenException(org.pmiops.workbench.exceptions.ForbiddenException) NotFoundException(org.pmiops.workbench.exceptions.NotFoundException) EmailVerificationStatus(org.pmiops.workbench.model.EmailVerificationStatus) Clock(java.time.Clock) Bean(org.springframework.context.annotation.Bean) Transactional(org.springframework.transaction.annotation.Transactional) CloneWorkspaceRequest(org.pmiops.workbench.model.CloneWorkspaceRequest) Cohort(org.pmiops.workbench.model.Cohort) CreateReviewRequest(org.pmiops.workbench.model.CreateReviewRequest) CohortReview(org.pmiops.workbench.model.CohortReview) Matchers.anyString(org.mockito.Matchers.anyString) Workspace(org.pmiops.workbench.model.Workspace) ResearchPurpose(org.pmiops.workbench.model.ResearchPurpose) ParticipantCohortStatuses(org.pmiops.workbench.model.ParticipantCohortStatuses) DataJpaTest(org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest) Test(org.junit.Test)

Example 18 with Workspace

use of org.pmiops.workbench.model.Workspace in project workbench by all-of-us.

the class WorkspacesControllerTest method testUnableToShareWithNonExistentUser.

@Test(expected = BadRequestException.class)
public void testUnableToShareWithNonExistentUser() throws Exception {
    Workspace workspace = createDefaultWorkspace();
    workspacesController.createWorkspace(workspace);
    ShareWorkspaceRequest shareWorkspaceRequest = new ShareWorkspaceRequest();
    UserRole creator = new UserRole();
    creator.setEmail(LOGGED_IN_USER_EMAIL);
    creator.setRole(WorkspaceAccessLevel.OWNER);
    shareWorkspaceRequest.addItemsItem(creator);
    UserRole writer = new UserRole();
    writer.setEmail("writerfriend@gmail.com");
    writer.setRole(WorkspaceAccessLevel.WRITER);
    shareWorkspaceRequest.addItemsItem(writer);
    workspacesController.shareWorkspace(workspace.getNamespace(), workspace.getName(), shareWorkspaceRequest);
}
Also used : UserRole(org.pmiops.workbench.model.UserRole) ShareWorkspaceRequest(org.pmiops.workbench.model.ShareWorkspaceRequest) Workspace(org.pmiops.workbench.model.Workspace) DataJpaTest(org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest) Test(org.junit.Test)

Example 19 with Workspace

use of org.pmiops.workbench.model.Workspace in project workbench by all-of-us.

the class WorkspacesControllerTest method testUpdateWorkspaceStaleThrows.

@Test(expected = ConflictException.class)
public void testUpdateWorkspaceStaleThrows() throws Exception {
    Workspace ws = createDefaultWorkspace();
    ws = workspacesController.createWorkspace(ws).getBody();
    UpdateWorkspaceRequest request = new UpdateWorkspaceRequest();
    request.setWorkspace(new Workspace().name("updated-name").etag(ws.getEtag()));
    stubGetWorkspace(ws.getNamespace(), ws.getId(), ws.getCreator(), WorkspaceAccessLevel.OWNER);
    workspacesController.updateWorkspace(ws.getNamespace(), ws.getId(), request).getBody();
    // Still using the initial now-stale etag; this should throw.
    stubGetWorkspace(ws.getNamespace(), ws.getId(), ws.getCreator(), WorkspaceAccessLevel.OWNER);
    request.setWorkspace(new Workspace().name("updated-name2").etag(ws.getEtag()));
    workspacesController.updateWorkspace(ws.getNamespace(), ws.getId(), request).getBody();
}
Also used : UpdateWorkspaceRequest(org.pmiops.workbench.model.UpdateWorkspaceRequest) Workspace(org.pmiops.workbench.model.Workspace) DataJpaTest(org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest) Test(org.junit.Test)

Example 20 with Workspace

use of org.pmiops.workbench.model.Workspace in project workbench by all-of-us.

the class WorkspacesControllerTest method testUpdateWorkspace.

@Test
public void testUpdateWorkspace() throws Exception {
    Workspace ws = createDefaultWorkspace();
    ws = workspacesController.createWorkspace(ws).getBody();
    ws.setName("updated-name");
    stubGetWorkspace(ws.getNamespace(), ws.getId(), ws.getCreator(), WorkspaceAccessLevel.OWNER);
    UpdateWorkspaceRequest request = new UpdateWorkspaceRequest();
    request.setWorkspace(ws);
    Workspace updated = workspacesController.updateWorkspace(ws.getNamespace(), ws.getId(), request).getBody();
    ws.setEtag(updated.getEtag());
    assertThat(updated).isEqualTo(ws);
    ws.setName("updated-name2");
    stubGetWorkspace(ws.getNamespace(), ws.getId(), ws.getCreator(), WorkspaceAccessLevel.OWNER);
    updated = workspacesController.updateWorkspace(ws.getNamespace(), ws.getId(), request).getBody();
    ws.setEtag(updated.getEtag());
    assertThat(updated).isEqualTo(ws);
    stubGetWorkspace(ws.getNamespace(), ws.getId(), ws.getCreator(), WorkspaceAccessLevel.OWNER);
    Workspace got = workspacesController.getWorkspace(ws.getNamespace(), ws.getId()).getBody().getWorkspace();
    assertThat(got).isEqualTo(ws);
}
Also used : UpdateWorkspaceRequest(org.pmiops.workbench.model.UpdateWorkspaceRequest) Workspace(org.pmiops.workbench.model.Workspace) DataJpaTest(org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest) Test(org.junit.Test)

Aggregations

Workspace (org.pmiops.workbench.model.Workspace)31 Test (org.junit.Test)23 DataJpaTest (org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest)23 ResearchPurpose (org.pmiops.workbench.model.ResearchPurpose)13 User (org.pmiops.workbench.db.model.User)9 CloneWorkspaceRequest (org.pmiops.workbench.model.CloneWorkspaceRequest)8 UserRole (org.pmiops.workbench.model.UserRole)7 ConflictException (org.pmiops.workbench.exceptions.ConflictException)6 ShareWorkspaceRequest (org.pmiops.workbench.model.ShareWorkspaceRequest)6 BadRequestException (org.pmiops.workbench.exceptions.BadRequestException)5 WorkspaceACLUpdate (org.pmiops.workbench.firecloud.model.WorkspaceACLUpdate)5 WorkspaceACLUpdateResponseList (org.pmiops.workbench.firecloud.model.WorkspaceACLUpdateResponseList)5 UpdateWorkspaceRequest (org.pmiops.workbench.model.UpdateWorkspaceRequest)5 Blob (com.google.cloud.storage.Blob)4 Matchers.anyString (org.mockito.Matchers.anyString)4 ApiException (org.pmiops.workbench.firecloud.ApiException)4 ResearchPurposeReviewRequest (org.pmiops.workbench.model.ResearchPurposeReviewRequest)4 Timestamp (java.sql.Timestamp)3 FirecloudWorkspaceId (org.pmiops.workbench.db.model.Workspace.FirecloudWorkspaceId)3 ShareWorkspaceResponse (org.pmiops.workbench.model.ShareWorkspaceResponse)3