单元测试

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-test</artifactId>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.spockframework</groupId>
    <artifactId>spock-core</artifactId>
    <scope>test</scope></dependency>
<dependency>
    <groupId>org.spockframework</groupId>
    <artifactId>spock-spring</artifactId>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>cglib</groupId>
    <artifactId>cglib-nodep</artifactId>
    <version>3.1</version>
    <scope>test</scope>
</dependency>

<groovy.version>2.4.10</groovy.version>
<spock-core.version>1.0-groovy-2.4</spock-core.version>
<spring-core.version>4.3.8.RELEASE</spring-core.version>

@SpringBootTest(classes = Application.class)
@ContextConfiguration(loader = SpringBootContextLoader.class)
class HiamTenantServiceSpec extends Specification {

}

@SpringBootTest(classes = Application.class)
@ContextConfiguration(loader = SpringBootContextLoader.class)
@Title("Unit test for hzero iam user service")
@Unroll
@Transactional
@Rollback
@Timeout(10)
class HiamUserServiceSpec extends Specification {
    @Autowired
    UserRepository userRepository;
    @Autowired
    HiamUserRepository hiamUserRepository;
    @Autowired
    UserCaptchaService userCaptchaService;
    @Autowired
    RedisHelper redisHelper;
    @Autowired
    HiamUserService hiamUserService;

    /**
     * 修改手机号流程测试
     */
    def "test modify phone #loginName"() {
        given: "获取用户信息"
        UserLogin.login(userRepository, loginName);
        UserVO self = hiamUserRepository.selectSelfDetail();
        String captchaKey, preCheckResultKey;
        CommonException e1;
        IllegalArgumentException e2;

        when: "手机格式不正确"
        captchaKey = userCaptchaService.sendPhoneCaptcha(phone1, true);
        then:
        e1 = thrown();
        e1.getMessage() == "validation.phone.incorrect"

        when: "给原手机发送验证码"
        captchaKey = userCaptchaService.sendPhoneCaptcha(self.getPhone(), true);
        then: "发送成功，返回验证码key"
        captchaKey != null

        when: "验证原手机验证码"
        preCheckResultKey = userCaptchaService.validatePreCheckCaptcha(captchaKey, getCaptcha(captchaKey));
        then:
        preCheckResultKey != null

        when: "给新手机发送验证码"
        captchaKey = userCaptchaService.sendPhoneCaptcha(new_phone, false)
        then:
        captchaKey != null;

        when: "验证新手机并修改手机号"
        boolean result = hiamUserService.validateNewPhoneCaptchaAndUpdate(preCheckResultKey, captchaKey, getCaptcha(captchaKey), new_phone);
        then:
        result

        where:
        loginName | phone1    | phone2        | captchaKey1 | new_phone
        "admin"   | "1852322" | "18523225327" | "wrong-key" | "18888888888"
        "hand"   | "1852322" | "18523225327" | "wrong-key" | "16666666666"

    }

    private String getCaptcha(String captchaKey) {
        String group = redisHelper.strGet(Constants.CacheKey.VALIDATE_CAPTCHA + ":" + captchaKey);
        String[] groupArr = StringUtils.split(group, "_", 2);
        return groupArr[0];
    }

}

package org.hzero.iam.infra.util;

import java.util.Collections;

import org.springframework.security.authentication.AbstractAuthenticationToken;
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
import org.springframework.security.core.context.SecurityContextHolder;

import io.choerodon.core.oauth.CustomUserDetails;

import org.hzero.iam.domain.entity.User;
import org.hzero.iam.domain.repository.UserRepository;

public class UserLogin {

    /**
     * 模拟登录，使得可以通过DetailsHelper获取用户ID、租户ID等信息
     *
     * @param userRepository 用户资源库
     * @param loginName 登录名
     */
    public static void login(UserRepository userRepository, String loginName) {
        CustomUserDetails details = new CustomUserDetails(loginName, "", Collections.emptyList());
        User user = userRepository.selectByLoginName(loginName);
        details.setUserId(user.getId());
        details.setLanguage(user.getLanguage());
        details.setTimeZone(user.getTimeZone());
        details.setEmail(user.getEmail());
        details.setOrganizationId(user.getOrganizationId());

        AbstractAuthenticationToken authentication =
                        new UsernamePasswordAuthenticationToken(details, "", Collections.emptyList());
        
        SecurityContextHolder.getContext().setAuthentication(authentication);
    }

    /**
     * 模拟登录
     *
     * @param loginName 登录名
     * @param userId 用户ID
     * @param organizationId 租户ID
     */
    public static void login(String loginName, Long userId, Long organizationId) {
        CustomUserDetails details = new CustomUserDetails(loginName, "", Collections.emptyList());
        details.setUserId(userId);
        details.setLanguage("zh_CN");
        details.setTimeZone("CTT");
        details.setEmail("hand@hand-china.com");
        details.setOrganizationId(organizationId);

        AbstractAuthenticationToken authentication =
                        new UsernamePasswordAuthenticationToken(details, "", Collections.emptyList());
        
        SecurityContextHolder.getContext().setAuthentication(authentication);
    }

}