zl程序教程

您现在的位置是:首页 >  其它

当前栏目

MockitoTest 单元测试

单元测试
2023-09-11 14:17:05 时间
package com.yiautos.psf.order.service.impl;

import org.junit.Assert;
import org.junit.Test;
import org.mockito.Mockito;

import java.util.ArrayList;
import java.util.List;


public class MockitoTest {


    @Test
    public void testBehavior() {
        //构建mock数据
        List<String> list = Mockito.mock(List.class);
        list.add("1");
        list.add("2");

        System.out.println(list.get(0)); // 会得到null ,前面只是在记录行为而已,没有往list中添加数据

        Mockito.verify(list).add("1"); // 正确,因为该行为被记住了
        Mockito.verify(list).add("3");//报错,因为前面没有记录这个行为

    }

    @Test
    public void testStub() {
        List<Integer> l = Mockito.mock(ArrayList.class);

        Mockito.when(l.get(0)).thenReturn(10);
        Mockito.when(l.get(1)).thenReturn(20);
        Mockito.when(l.get(2)).thenThrow(new RuntimeException("no such element"));

        Assert.assertEquals(10, (int) l.get(0));
        Assert.assertEquals(20, (int) l.get(1));
        Assert.assertNull(l.get(4));
        l.get(2);
    }

    @Test
    public void testSpy() {
        List<String> spyList = Mockito.spy(ArrayList.class);

        spyList.add("one");
        spyList.add("two");

        Mockito.verify(spyList).add("one");
        Mockito.verify(spyList).add("two");

        Assert.assertEquals(2, spyList.size());

        Mockito.doReturn(100).when(spyList).size();
        Assert.assertEquals(100, spyList.size());
    }

    @Test
    public void testVoidStub() {
        List<Integer> l = Mockito.mock(ArrayList.class);
        Mockito.doReturn(10).when(l).get(1);
        Mockito.doThrow(new RuntimeException("you cant clear this List")).when(l).clear();

        Assert.assertEquals(10, (int) l.get(1));
        l.clear();
    }

    @Test
    public void testMatchers() {
        List<Integer> l = Mockito.mock(ArrayList.class);
        Mockito.when(l.get(Mockito.anyInt())).thenReturn(100);

        Assert.assertEquals(100, (int)l.get(999));
    }
}

 mock token

  @Mock
    private HttpServletRequest request;

    @Before
    public void init() {
        BladeUser bladeUser = new BladeUser();
        bladeUser.setUserId("1");
        //部门研发部
        bladeUser.setDeptId("2");
        Mockito.when(request.getRequestURI()).thenReturn("");
        Mockito.when(request.getAttribute("_BLADE_USER_REQUEST_ATTR_")).thenReturn(bladeUser);
        ServletRequestAttributes servletRequestAttributes = new ServletRequestAttributes(request);
        RequestContextHolder.setRequestAttributes(servletRequestAttributes);
    }