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

Example 26 with User

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

the class AccessTokenServiceTest method getLoginUser.

private User getLoginUser() {
    User loginUser = new User();
    loginUser.setId(1);
    loginUser.setUserType(UserType.ADMIN_USER);
    return loginUser;
}

Example 27 with User

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

the class SessionServiceTest method testSignOut.

/**
 * sign out
 * remove ip restrictions
 */
@Test
public void testSignOut() {
    int userId = 88888888;
    String ip = "127.0.0.1";
    User user = new User();
    user.setId(userId);
    Mockito.when(sessionMapper.queryByUserIdAndIp(userId, ip)).thenReturn(getSession());
    sessionService.signOut(ip, user);
}

Example 28 with User

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

the class AlertGroupServiceTest method getLoginUser.

/**
 * create admin user
 * @return
 */
private User getLoginUser() {
    User loginUser = new User();
    loginUser.setUserType(UserType.ADMIN_USER);
    loginUser.setId(99999999);
    return loginUser;
}

Example 29 with User

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

the class AlertGroupServiceTest method testUpdateAlertgroup.

@Test
public void testUpdateAlertgroup() {
    User user = new User();
    // no operate
    Map<String, Object> result = alertGroupService.updateAlertgroup(user, 1, groupName, AlertType.SMS, groupName);
    logger.info(result.toString());
    Assert.assertEquals(Status.USER_NO_OPERATION_PERM, result.get(Constants.STATUS));
    user.setUserType(UserType.ADMIN_USER);
    // not exist
    result = alertGroupService.updateAlertgroup(user, 1, groupName, AlertType.SMS, groupName);
    logger.info(result.toString());
    Assert.assertEquals(Status.ALERT_GROUP_NOT_EXIST, result.get(Constants.STATUS));
    // success
    Mockito.when(alertGroupMapper.selectById(2)).thenReturn(getEntity());
    result = alertGroupService.updateAlertgroup(user, 2, groupName, AlertType.SMS, groupName);
    logger.info(result.toString());
    Assert.assertEquals(Status.SUCCESS, result.get(Constants.STATUS));
}

Example 30 with User

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

the class BaseServiceTest method testIsAdmin.

@Test
public void testIsAdmin() {
    User user = new User();
    user.setUserType(UserType.ADMIN_USER);
    // ADMIN_USER
    boolean isAdmin = baseService.isAdmin(user);
    Assert.assertTrue(isAdmin);
    // GENERAL_USER
    user.setUserType(UserType.GENERAL_USER);
    isAdmin = baseService.isAdmin(user);
    Assert.assertFalse(isAdmin);
}