Question

我创建了下面给出的JWTCreateImpl类，以使用generatToken（）方法生成jwt令牌，并使用loginUser（）方法响应jwt字符串。 loginUser（）方法将电子邮件和密码作为来自前端的参数。将此电子邮件与数据库中存在的电子邮件进行比较，如果不存在电子邮件，则errorResponse将变为“电子邮件不存在”，同样适用于密码不匹配的情况。最后，如果存在用于特定ID的电子邮件和密码，则从数据库中提取结果并仅传递用户的uuid以生成令牌，并且响应将是jwt字符串。我想使用Junit4对此类进行单元测试，如何为此使用Junit4编写测试用例？

JWTCreateImpl类实现JWTCreate接口，该接口包含抽象方法loginUser（UserInfoJWT userInfoJWT）和generateToken（UUID uuid）以及一个主要用于HMAC256算法的generateToken（）方法中使用的String JWT_TOKEN_KEY。 UserInfoJWT是一个POJO类，其中包含用户的电子邮件和密码。

ResponseJWT是一个POJO类，其中包含UserToken和errorResponse。

import com.auth0.jwt.JWT;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.exceptions.JWTCreationException;
import com.knoldus.magad.jwt.api.JWTCreate;
import com.knoldus.magad.jwt.model.ResponseJWT;
import com.knoldus.magad.jwt.model.UserInfoJWT;
import com.knoldus.magad.dao.model.UserInfo;
import com.knoldus.magad.user.api.UserInfoProcessingAPI;

import org.apache.log4j.Logger;


import java.io.UnsupportedEncodingException;
import java.time.ZonedDateTime;
import java.util.Date;
import java.util.UUID;
import java.util.concurrent.CompletableFuture;
public class JWTCreateImpl implements JWTCreate {
private static final Logger LOGGER = Logger.getLogger(JWTCreateImpl.class);
private UserInfo existingUser;
private ResponseJWT response;
private  UserInfoProcessingAPI userInfoProcessingAPI;
private static final String NO_USER_EMAIL = "Email doesn't exists..";
private static  final String LOGIN_UNSUCCESSFUL = "Wrong Password";


@Override
public CompletableFuture<String>loginUser(final UserInfoJWT userInfoJWT) {
    return userInfoProcessingAPI.getUserInfo(userInfoJWT.getUserEmail()).thenApply(result -> {
        existingUser = result.orElse(null);
        if (existingUser == null) {
            response = ResponseJWT.builder().errorResponse(NO_USER_EMAIL).build();
        } else {
            final String dataBasePassword = existingUser.getUserPassword();
            if (userInfoJWT.getUserPassword().equals(dataBasePassword)) {
                response = ResponseJWT.builder().userToken(generateToken(existingUser.getUuid())).build();
            } else {
                response = ResponseJWT.builder().errorResponse(LOGIN_UNSUCCESSFUL).build();
            }
        }
        return response.toString();
    });
}
@Override
public String generateToken(final UUID uuid) {
    try {
        final Algorithm algorithm = Algorithm.HMAC256(JWT_TOKEN_KEY);
        final Date expirationDate = Date.from(ZonedDateTime.now().plusHours(24).toInstant());
        final Date issuedAt = Date.from(ZonedDateTime.now().toInstant());
        final String id = String.valueOf(uuid);
        final String issuer = "MagADAuth";
        return JWT.create()
                .withIssuedAt(issuedAt) // Issue date.
                .withExpiresAt(expirationDate) // Expiration date.
                .withClaim("userId", id)
                .withIssuer(issuer) // Issuer of the token.
                .sign(algorithm); // And the signing algorithm.
    } catch (UnsupportedEncodingException | JWTCreationException e) {
        LOGGER.error(e.getMessage(), e);
    }
    return null;
}

Answer 1

众所周知，单元测试是独立测试应用程序的一小部分，这意味着在我们的案例中，方法应该独立进行测试

在您的情况下，JWTCreateImpl类的loginUser方法实际上依赖于对象UserInfoJWT，UserInfoProcessingAPI，因此为了对testUserUser方法进行单元测试

我们需要模拟UserInfoJWT类，以便它将向我们返回实际上是方法输入所需的电子邮件
我们还需要模拟UserInfoProcessingAPI，以便它将返回User
最后，我们还需要在单元测试用例中调用真实的方法loginUser方法（此外，我们还需要针对每个可能的条件使用单个单元测试来测试loginUser方法）
对于generateToken方法，我们还需要提供UUID模拟以进行单元测试。

在上述情况下，我更喜欢Mockito和Junit 引用链接

What is Mocking? Why to use Mockito?

按评论编辑

@RunWith(MockitoJUnitRunner.class)
public class JWTCreateImplTest {

    /* class under test should not be mocked */
    private JWTCreateImpl jwtCreateImpl;

    /* Mocking UserInfoProcessingAPI as it is dependency of JWTCreateImpl */
    @Mock
    private UserInfoProcessingAPI infoProcessingAPI;

    /*Mockito method which setup everything before test actually start (All Object Initialization can be done here */
    @Before
    public void setup() {
        jwtCreateImpl = new JWTCreateImpl();
    }

    /* Testing Happy flow */
    @Test
    public void testNotNullJwtToken() {
        Mockito.when(infoProcessingAPI.getUserInfo(getUser().getUserEmail())).thenReturn(getAppUser());
        CompletableFuture<String> jwtToken = jwtCreateImpl.loginUser(getUser());
        Assert.assertNotNull(jwtToken);
    }

    /* Testing when null email is passed */
    @Test
    public void testJwtTokenWithNullEmail() {
        /* this will return null as there are no object exists with email <null> */
        Mockito.when(infoProcessingAPI.getUserInfo(null)).thenReturn(null);
        CompletableFuture<String> jwtToken = jwtCreateImpl.loginUser(getUser());
        /* response of NO_USER_EMAIL should be checked here */
        Assert.assertEquals("NO_USER_EMAIL",jwtToken);
    }

    /*TODO */

    /* similarly you need to check
    * when password is null <AND>
    * when password doesn't match with requester password <AND>
    * generateToken method should be checked for positive scenrio as well as case scenrio where it will throw
    * UnsupportedEncodingException or JWTCreationException
    * all of these in independant test cases
    */

    private UserInfoJwt getUser(){
        UserInfoJwt userInfoJwt = new UserInfoJwt();
        userInfoJwt.setUserEmail("name@java.com");
        userInfoJwt.setUserEmail("myPassword");
        return userInfoJwt;
    }

    private AppUser getAppUser() {

        AppUser user = new AppUser();
        user.setId(1);
        user.setName("name");
        user.setEmail("appuser@java.com");
        return user;
    }
}

如何使用Junit4进行单元测试？

1 个答案: