1. quickstart

最新推荐文章于 2024-08-10 08:38:11 发布

原创最新推荐文章于 2024-08-10 08:38:11 发布 · 90 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#redirect #junit #web #单元测试 #smtp

Mockito 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一个基于Mockito的单元测试案例，针对登录控制器进行成功登录、失败登录及异常情况的测试。通过模拟请求参数和数据库操作，验证控制器的行为是否符合预期。

package lesson1_quickstart;

import common.Account;
import common.AccountDao;
import common.AccountLoginController;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.mockito.runners.MockitoJUnitRunner;
import javax.servlet.http.HttpServletRequest;
import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.when;


@RunWith(MockitoJUnitRunner.class)
public class AccountLoginControllerTest {

    private AccountDao accountDao;
    private HttpServletRequest request;
    private AccountLoginController accountLoginController;

    @Before
    public void setUp() {
        accountDao = Mockito.mock(AccountDao.class);
        request = Mockito.mock(HttpServletRequest.class);
        accountLoginController = new AccountLoginController(accountDao);
    }

    @Test
    public void testLoginSuccess() {
        when(request.getParameter("username")).thenReturn("cain");
        when(request.getParameter("password")).thenReturn("123456");

        Account account = new Account();
        when(accountDao.getAccount(anyString(), anyString())).thenReturn(account);

        String login = accountLoginController.login(request);
        Assert.assertEquals("/index", login);
    }

    @Test
    public void testLoginFailure() {
        when(request.getParameter("username")).thenReturn("cain");
        when(request.getParameter("password")).thenReturn("123456");

        when(accountDao.getAccount(anyString(), anyString())).thenReturn(null);

        String login = accountLoginController.login(request);
        Assert.assertEquals("/login", login);
    }

    @Test
    public void test505() {
        when(request.getParameter("username")).thenReturn("cain");
        when(request.getParameter("password")).thenReturn("123456");

        when(accountDao.getAccount(anyString(), anyString())).thenThrow(UnsupportedOperationException.class);

        String login = accountLoginController.login(request);
        Assert.assertEquals("500", login);
    }
}

AccountLoginControllerTest

package common;

import javax.servlet.http.HttpServletRequest;

public class AccountLoginController {
    private AccountDao accountDao;

    public AccountLoginController(AccountDao accountDao) {
        this.accountDao = accountDao;
    }

    public String login(HttpServletRequest request) {
        final String username = request.getParameter("username");
        final String password = request.getParameter("password");
        try {
            Account account = accountDao.getAccount(username, password);
            if (account == null) {
                return "/login";
            } else {
                return "/index";
            }
        } catch (Exception e) {
            return "500";
        }

    }
}

AccountLoginController

package common;

public class AccountDao {
    public Account getAccount(String username, String pwd) {
        throw new UnsupportedOperationException("DB is down");
    }
}

AccountDao

package common;

public class Account {

}

Account