Examples with User org.apache.dolphinscheduler.dao.entity.User used on opensource projects

Example 21 with User

use of org.apache.dolphinscheduler.dao.entity.User in project dolphinscheduler by apache.

the class UsersServiceTest method testDeleteUserById.

@Test
public void testDeleteUserById() {
    User loginUser = new User();
    try {
        when(userMapper.queryTenantCodeByUserId(1)).thenReturn(getUser());
        when(userMapper.selectById(1)).thenReturn(getUser());
        // no operate
        Map<String, Object> result = usersService.deleteUserById(loginUser, 3);
        logger.info(result.toString());
        Assert.assertEquals(Status.USER_NO_OPERATION_PERM, result.get(Constants.STATUS));
        // user not exist
        loginUser.setUserType(UserType.ADMIN_USER);
        result = usersService.deleteUserById(loginUser, 3);
        logger.info(result.toString());
        Assert.assertEquals(Status.USER_NOT_EXIST, result.get(Constants.STATUS));
        // success
        result = usersService.deleteUserById(loginUser, 1);
        logger.info(result.toString());
        Assert.assertEquals(Status.SUCCESS, result.get(Constants.STATUS));
    } catch (Exception e) {
        logger.error("delete user error", e);
        Assert.assertTrue(false);
    }
}

Example 22 with User

use of org.apache.dolphinscheduler.dao.entity.User in project dolphinscheduler by apache.

the class AccessTokenMapperTest method createAccessTokens.

/**
 * create accessTokens
 * @param count create accessToken count
 * @param userName username
 * @return accessToken map
 * @throws Exception
 */
private Map<Integer, AccessToken> createAccessTokens(Integer count, String userName) throws Exception {
    User user = createUser(userName);
    Map<Integer, AccessToken> accessTokenMap = new HashMap<>();
    for (int i = 1; i <= count; i++) {
        AccessToken accessToken = createAccessToken(user.getId(), userName);
        accessTokenMap.put(accessToken.getId(), accessToken);
    }
    return accessTokenMap;
}

Example 23 with User

use of org.apache.dolphinscheduler.dao.entity.User in project dolphinscheduler by apache.

the class UDFUserMapperTest method testUpdate.

/**
 * test update
 */
@Test
public void testUpdate() {
    // insertOneUser
    User user = insertOneUser();
    // insertOneUdfFunc
    UdfFunc udfFunc = insertOneUdfFunc();
    // insertOne
    UDFUser udfUser = insertOne(user, udfFunc);
    udfUser.setUserId(2);
    udfUser.setUdfId(2);
    int update = udfUserMapper.updateById(udfUser);
    Assert.assertEquals(update, 1);
}

Example 24 with User

use of org.apache.dolphinscheduler.dao.entity.User in project dolphinscheduler by apache.

the class UserAlertGroupMapperTest method testDeleteByAlertgroupId.

/**
 * test delete by alertgroupId
 */
@Test
public void testDeleteByAlertgroupId() {
    // insertOneUser
    User user = insertOneUser();
    // insertOneAlertGroup
    AlertGroup alertGroup = insertOneAlertGroup();
    // insertOne
    UserAlertGroup userAlertGroup = insertOne(user, alertGroup);
    int delete = userAlertGroupMapper.deleteByAlertgroupId(alertGroup.getId());
    Assert.assertEquals(delete, 1);
}

Example 25 with User

use of org.apache.dolphinscheduler.dao.entity.User in project dolphinscheduler by apache.

the class PasswordAuthenticatorTest method testGetAuthUser.

@Test
public void testGetAuthUser() {
    HttpServletRequest request = Mockito.mock(HttpServletRequest.class);
    when(usersService.queryUser(mockUser.getId())).thenReturn(mockUser);
    when(sessionService.getSession(request)).thenReturn(mockSession);
    User user = authenticator.getAuthUser(request);
    Assert.assertNotNull(user);
}