Examples with DbUser org.pmiops.workbench.db.model.DbUser used on opensource projects

Example 1 with DbUser

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

the class OfflineAuditControllerTest method setUp.

@BeforeEach
public void setUp() {
    DbUser user = new DbUser();
    user.setUsername(USER_EMAIL);
    user.setUserId(123L);
    user.setDisabled(false);
    user = userDao.save(user);
    DbCdrVersion cdrV1 = TestMockFactory.createDefaultCdrVersion(cdrVersionDao, accessTierDao, 1);
    cdrV1.setBigqueryProject(CDR_V1_PROJECT_ID);
    cdrV1 = cdrVersionDao.save(cdrV1);
    DbCdrVersion cdrV2 = TestMockFactory.createDefaultCdrVersion(cdrVersionDao, accessTierDao, 2);
    cdrV2.setBigqueryProject(CDR_V2_PROJECT_ID);
    cdrV2 = cdrVersionDao.save(cdrV2);
    when(workspaceDao.findAllWorkspaceNamespaces()).thenReturn(ImmutableSet.of(FC_PROJECT_ID));
    CLOCK.setInstant(NOW);
}

Example 2 with DbUser

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

the class AuthDomainControllerTest method testDisableUser.

@Test
public void testDisableUser() {
    final boolean oldDisabledValue = false;
    final DbUser createdUser = createUser(oldDisabledValue);
    final boolean newDisabledValue = true;
    UpdateUserDisabledRequest request = new UpdateUserDisabledRequest().email(PRIMARY_EMAIL).disabled(newDisabledValue);
    ResponseEntity<Void> response = this.authDomainController.updateUserDisabledStatus(request);
    verify(mockAuthDomainAuditAdapter).fireSetAccountDisabledStatus(createdUser.getUserId(), newDisabledValue, oldDisabledValue);
    assertThat(response.getStatusCode()).isEqualTo(HttpStatus.NO_CONTENT);
    DbUser updatedUser = userDao.findUserByUsername(PRIMARY_EMAIL);
    assertThat(updatedUser.getDisabled()).isTrue();
}

Example 3 with DbUser

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

the class AuthDomainControllerTest method createUser.

private DbUser createUser(boolean disabled) {
    DbUser user = new DbUser();
    user.setGivenName(GIVEN_NAME);
    user.setFamilyName(FAMILY_NAME);
    user.setUsername(PRIMARY_EMAIL);
    user.setContactEmail(CONTACT_EMAIL);
    user.setAreaOfResearch(RESEARCH_PURPOSE);
    user.setDisabled(disabled);
    return userDao.save(user);
}

Example 4 with DbUser

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

the class FreeTierBillingServiceTest method createUser.

private DbUser createUser(String email) {
    DbUser user = new DbUser();
    user.setUsername(email);
    return userDao.save(user);
}

Example 5 with DbUser

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

the class WorkspacesControllerTest method testShareWorkspaceAddBillingProjectUser.

@Test
public void testShareWorkspaceAddBillingProjectUser() {
    stubFcGetGroup();
    DbUser writerUser = createAndSaveUser("writerfriend@gmail.com", 124L);
    DbUser ownerUser = createAndSaveUser("ownerfriend@gmail.com", 125L);
    stubFcGetWorkspaceACL();
    Workspace workspace = createWorkspace();
    workspace = workspacesController.createWorkspace(workspace).getBody();
    ShareWorkspaceRequest shareWorkspaceRequest = new ShareWorkspaceRequest().workspaceEtag(workspace.getEtag()).addItemsItem(new UserRole().email(LOGGED_IN_USER_EMAIL).role(WorkspaceAccessLevel.OWNER)).addItemsItem(new UserRole().email(writerUser.getUsername()).role(WorkspaceAccessLevel.WRITER)).addItemsItem(new UserRole().email(ownerUser.getUsername()).role(WorkspaceAccessLevel.OWNER));
    stubFcUpdateWorkspaceACL();
    workspacesController.shareWorkspace(workspace.getNamespace(), workspace.getName(), shareWorkspaceRequest);
    verify(fireCloudService, times(1)).addOwnerToBillingProject(ownerUser.getUsername(), workspace.getNamespace());
    verify(fireCloudService, never()).addOwnerToBillingProject(eq(writerUser.getUsername()), any());
    verify(fireCloudService, never()).removeOwnerFromBillingProject(any(), any(), eq(Optional.empty()));
}