spring mvc @controller unit test

1. spring mvc controller implementation

@Controller
public class Alarm {
@Resource(name = "db.service", shareable = true)
private IDatabaseService dbService;
@RequestMapping(value = "/query_alarmlogs", produces = "application/json;charset=UTF-8")
public @ResponseBody
AlarmLogList queryAlarmLogs(@RequestParam("queryType") String queryType, @RequestParam("lastId") String lastId) {
AlarmQueryType alarmType = AlarmQueryType.fromValue(queryType);
AlarmLogList list = findAlarmLogs(alarmType.getAlarmDate(), Integer.valueOf(lastId));
return list;
}
private AlarmLogList findAlarmLogs(Date date, int lastId) {
List<AlarmLogs> alarmList = dbService.findAlarmLogs(date, lastId);
AlarmLogList alarms = new AlarmLogList(alarmList);
return alarms;
}
private final class AlarmLogList extends ArrayList<AlarmLogs> {
private static final long serialVersionUID = 1L;
public AlarmLogList(List<AlarmLogs> list) {
super(list);
}
}
public void setDbService(IDatabaseService dbService) {
this.dbService = dbService;
}
}

2. test it.

package com.eifesun.mornitor.web;
import static org.hamcrest.Matchers.equalTo;
import static org.hamcrest.Matchers.hasSize;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoMoreInteractions;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
import java.nio.charset.Charset;
import java.util.Arrays;
import java.util.Date;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
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.setup.MockMvcBuilders;
import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerAdapter;
import com.eifesun.monitor.DeviceType;
import com.eifesun.monitor.service.IDatabaseService;
import com.eifesun.monitor.service.MongoDBServiceImpl;
import com.eifesun.monitor.web.Alarm;
import com.eifesun.monitor.web.command.AlarmQueryType;
import com.eifesun.monitor.web.model.AlarmLogs;
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"/spring/mvc-core-config.xml", "/spring/business-config.xml"})
@WebAppConfiguration
public class AlarmControllerTest {
private MockMvc mockMvc;
@Autowired
public RequestMappingHandlerAdapter handlerAdapter;
@Mock
private IDatabaseService mockDbService;
@Autowired
private Alarm alarm;
@Before
public void init() {
mockDbService = Mockito.mock(MongoDBServiceImpl.class);
alarm.setDbService(mockDbService);
this.mockMvc = MockMvcBuilders.standaloneSetup(alarm).build();
}
@Test
public void testQueryAlarmLogs() throws Exception {
AlarmLogs log1 = new AlarmLogs(DeviceType.INVERTER, "inverter1", "panel 1 over volage");
AlarmLogs log2 = new AlarmLogs(DeviceType.INVERTER, "inverter1", "panel 2 over volage");
int lastId = 0;
String queryType = "alarm_auto";
Date date = AlarmQueryType.fromValue(queryType).getAlarmDate();
Mockito.when(mockDbService.findAlarmLogs(date, lastId)).thenReturn(Arrays.asList(log1, log2));
//@formatter:off
mockMvc.perform(get("/query_alarmlogs?queryType=alarm_auto&lastId=0"))
.andExpect(status().isOk())
.andExpect(content().contentType(APPLICATION_JSON_UTF8))
.andExpect(jsonPath("$", hasSize(2)))
.andExpect(jsonPath("$[0].clientId", equalTo("inverter1")))
.andExpect(jsonPath("$[0].deviceType", equalTo("INVERTER")))
.andExpect(jsonPath("$[0].description", equalTo("panel 1 over volage")))
.andExpect(jsonPath("$[1].clientId", equalTo("inverter1")))
.andExpect(jsonPath("$[1].deviceType", equalTo("INVERTER")))
.andExpect(jsonPath("$[1].description", equalTo("panel 2 over volage")));
//@formatter:on
verify(mockDbService, times(1)).findAlarmLogs(date, lastId);
verifyNoMoreInteractions(mockDbService);
}
public static final MediaType APPLICATION_JSON_UTF8 = new MediaType(MediaType.APPLICATION_JSON.getType(), MediaType.APPLICATION_JSON.getSubtype(),
Charset.forName("utf8"));
}