JUNIT4 模拟session

àì夳堔傛蜴生んèń 2022-11-16 13:57 150阅读 0赞

# 一、方式一 #

//需要Servlet 3.0，如果项目是Servlet 2.5，引入测试依赖Servlet 3.0即可。
    //MockHttpServletRequest request =new MockHttpServletRequest();
    //MockHttpServletResponse response = new MockHttpServletResponse();
    MockHttpSession session = new MockHttpSession();

依赖

<dependency>
    	<groupId>javax.servlet</groupId>
    	<artifactId>javax.servlet-api</artifactId>
    	<version>3.1.0</version>
    	<scope>test</scope>
    </dependency>

如果不引入Servlet3.0会报错：

java.lang.NoClassDefFoundError: javax/servlet/AsyncContext  
java.lang.NoClassDefFoundError: javax/servlet/SessionCookieConfig

# 二、方式二（未测试） #

HttpServletRequest requestMock = Mockito.mock(HttpServletRequest.class);
    HttpServletResponse responseMock = Mockito.mock(HttpServletResponse.class);
    HttpSession sessionMock = Mockito.mock(HttpSession.class);
    Mockito.when(requestMock.getSession()).thenReturn(sessionMock);

依赖

<dependency>
    	<groupId>org.mockito</groupId>
    	<artifactId>mockito-core</artifactId>
    	<version>1.9.5</version>
    	<scope>test</scope>
    </dependency>

# 三、示例代码 #

package test.cms;
    
    import static org.hamcrest.MatcherAssert.assertThat;
    import static org.hamcrest.Matchers.greaterThan;
    
    import java.util.HashMap;
    import java.util.Map;
    
    import org.apache.commons.logging.Log;
    import org.apache.commons.logging.LogFactory;
    import org.junit.After;
    import org.junit.Before;
    import org.junit.Test;
    import org.junit.runner.RunWith;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.mock.web.MockHttpSession;
    import org.springframework.test.annotation.Rollback;
    import org.springframework.test.context.ContextConfiguration;
    import org.springframework.test.context.junit4.AbstractTransactionalJUnit4SpringContextTests;
    import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
    
    import com.urthink.framework.page.EuiPage;
    import com.urthink.framework.web.UserSession;
    import com.urthink.module.cms.entity.Catalog;
    import com.urthink.module.cms.service.CatalogService;
    import com.urthink.platform.authority.entity.User;
    
    /**
     * 栏目测试类
     * @author happyqing
     *
     */
    @RunWith(SpringJUnit4ClassRunner.class) // 整合
    @Rollback // 事务自动回滚
    @ContextConfiguration(locations = {"classpath:spring-context.xml", "classpath:spring-mybatis.xml"}) // 加载配置
    public class CatalogTest extends AbstractTransactionalJUnit4SpringContextTests {
    	
    	private static final Log log = LogFactory.getLog(CatalogTest.class);
    	
    	@Autowired
    	private CatalogService catalogService;
    	
    	/**
    	 * 测试前执行
    	 */
    	@Before
    	public void beforeTest(){
    		User user = new User();
    		user.setId("1");
    		
    		//需要Servlet 3.0，如果项目是Servlet 2.5，引入测试依赖Servlet 3.0即可。
    		//MockHttpServletRequest request =new MockHttpServletRequest();
    		//MockHttpServletResponse response = new MockHttpServletResponse();
    		MockHttpSession session = new MockHttpSession();
    		
    		session.setAttribute("user", user);
    	}
    
    	/**
    	 * 测试后执行
    	 */
    	@After
    	public void afterTest(){
    		
    	}
    	
    	/**
    	 * 分页查询测试
    	 */
    	@Test
    	public void pageListTest() {
    		Map paramMap = new HashMap();
    		paramMap.put("page", 1);
    		paramMap.put("rows", 10);
    		paramMap.put("orderBy", "id asc");
    		EuiPage euiPage = catalogService.getPageList(paramMap);
    		assertThat(euiPage.getTotal(), greaterThan(0L));
    	}
    	
    	/**
    	 * 插入测试
    	 */
    	@Test
    	public void insertTest() {
    		Catalog catalog = new Catalog();
    		catalog.setParentId(14);
    		catalog.setSiteId(2);
    		catalog.setName("测试栏目");
    		catalog.setDescription("测试栏目描述");
    		catalog.setEnable("1");
    		int r = catalogService.insert(catalog);
    		assertThat(r, greaterThan(0));
    	}
    	
    }

参考：

[Junit4中的新断言assertThat的使用方法][Junit4_assertThat]

[Junit4_assertThat]: https://blog.csdn.net/smxjant/article/details/78206435