package com.neu.cse.powercloud.serviceImpl.sysmanage;
import com.github.pagehelper.Page;
import com.neu.cse.powercloud.pojo.sysmanage.SysUser;
import com.neu.cse.powercloud.service.sysmanage.UserManageService;
import com.neu.cse.powercloud.util.ResponseResult;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
import org.springframework.mock.web.MockHttpSession;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.test.context.web.WebAppConfiguration;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.MvcResult;
import org.springframework.test.web.servlet.result.MockMvcResultHandlers;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;
import javax.servlet.http.HttpSession;
import java.util.Random;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
import static org.springframework.test.web.servlet.result.MockMvcResultHandlers.print;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
@RunWith(SpringJUnit4ClassRunner.class)
@WebAppConfiguration(value = "src/main/webapp")
@ContextConfiguration(locations = { "classpath:spring/springmvc.xml", "classpath:mybatis/SqlMapConfig.xml", "classpath:spring/applicationContext.xml"})
public class UserManageServiceImplTest {
@Autowired
private UserManageService userManageService;
@Autowired
private WebApplicationContext wac;
private MockMvc mockMvc;
private MockHttpSession session;
@Before
public void setUp() {
mockMvc = MockMvcBuilders.webAppContextSetup(wac).build();
session = new MockHttpSession();
}
@Test
public void addUser() throws Exception {
// 需要登录
// 发送请求
String sysUserJson = "{ \"username\": \"admin" + new Random().nextInt() + "\", \"password\": \"admin1\", \"passwordrepeat\": \"admin1\", " +
"\"customerid\": \"100010\", \"tel\": \"191717171\", \"status\": \"1\"}";
System.out.println(sysUserJson);
MvcResult result = this.mockMvc.perform(
post("/user/add")
.contentType(MediaType.APPLICATION_JSON)
.content(sysUserJson)
.session((MockHttpSession)getLoginSession()))
.andDo(print())
.andExpect(status().isOk())
// .andExpect(jsonPath("$.msg").value("OK"))
.andReturn();
String content = result.getResponse().getContentAsString();
System.out.println(content);
// System.out.println(userManageService.getUsers(0, 10));
}
@Test
public void getUsers() throws Exception {
ResponseResult result = userManageService.getUsers(0, 3);
Page page = (Page)result.getData();
System.out.println(page.getTotal());
System.out.println(page.getResult());
System.out.println(page.size()); // 该页记录数量
for (int i=0; i < page.size(); ++i) {
SysUser tUser = (SysUser) page.get(i);
System.out.println(tUser);
}
}
@Test
public void deleteUser() throws Exception {
ResponseResult result = userManageService.getUsers(0, 3);
System.out.println(result.getData());
Assert.assertTrue(result.getData() instanceof Page);
Page page = (Page)result.getData();
System.out.println(page);
long totalBefore = page.getTotal();
if (page.size() > 0) {
SysUser deleteUser = (SysUser) page.get(0);
System.out.println("删除用户:" + deleteUser);
userManageService.deleteUser(deleteUser.getId());
result = userManageService.getUsers(0, 3);
page = (Page)result.getData();
if (page.size() > 0) {
Assert.assertNotEquals(((SysUser) page.get(0)).getId(), deleteUser.getId());
}
Assert.assertEquals(totalBefore, page.getTotal() + 1);
}
}
/**
* 获取登录session
* @return Session
* @throws Exception E
*/
private HttpSession getLoginSession() throws Exception {
MvcResult result = this.mockMvc.perform(
post("/login/userLogin")
.contentType(MediaType.APPLICATION_FORM_URLENCODED)
.param("account", "admin")
.param("password", "admin"))
.andExpect(status().isOk())
.andDo(print())
.andReturn();
return result.getRequest().getSession();
}
}