@Test
public void testGetContextElement_OK() throws Exception {
ArgumentCaptor<QueryContext> queryContextArg = ArgumentCaptor.forClass(QueryContext.class);
List attributes = Collections.singletonList(new ContextAttribute("test", "string", "OK"));
ContextElement contextElement = new ContextElement();
contextElement.setEntityId(new EntityId("12345678", "", false));
contextElement.setContextAttributeList(attributes);
ContextElementResponse contextElementResponse =
new ContextElementResponse(contextElement, new StatusCode(CodeEnum.CODE_200));
QueryContextResponse response = new QueryContextResponse();
response.setContextElementResponses(Collections.singletonList(contextElementResponse));
when(ngsiController.queryContext(any())).thenReturn(response);
mockMvc
.perform(get("/v1/contextEntities/12345678").accept(MediaType.APPLICATION_JSON))
.andExpect(status().isOk())
.andExpect(jsonPath("$.statusCode.code").value("200"))
.andExpect(jsonPath("$.contextElement.attributes[0].name").value("test"))
.andExpect(jsonPath("$.contextElement.attributes[0].type").value("string"))
.andExpect(jsonPath("$.contextElement.attributes[0].value").value("OK"));
verify(ngsiController).queryContext(queryContextArg.capture());
QueryContext queryContext = queryContextArg.getValue();
assertNotNull(queryContext);
assertNotNull(queryContext.getEntityIdList());
assertEquals(1, queryContext.getEntityIdList().size());
assertNull(queryContext.getAttributeList());
assertEquals("12345678", queryContext.getEntityIdList().get(0).getId());
assertEquals("", queryContext.getEntityIdList().get(0).getType());
assertEquals(false, queryContext.getEntityIdList().get(0).getIsPattern());
}
private QueryContextResponse createRoomQueryResponse(EntityId queryEntityId) {
QueryContextResponse queryContextResponse = new QueryContextResponse();
List<ContextElementResponse> contextElementResponseList = new ArrayList<>();
if ((queryEntityId.getIsPattern()) && (queryEntityId.getId().equals("Room*"))) {
for (int varTemp : tempTab) {
for (int varRoom = 1; varRoom < 5; varRoom++) {
for (int varFloor = 1; varFloor < 4; varFloor++) {
ContextElement contextElement = new ContextElement();
String name = "Room" + varFloor + varRoom;
EntityId entityId = new EntityId(name, "Room", false);
contextElement.setEntityId(entityId);
int value = ((varTemp + (2 * varFloor) + varRoom));
ContextAttribute tempContextAttribute =
new ContextAttribute("temperature", "double", String.valueOf(value));
ContextAttribute floorContextAttribute =
new ContextAttribute("floor", "string", "Floor" + varFloor);
List<ContextAttribute> contextAttributeList = new ArrayList<>();
contextAttributeList.add(tempContextAttribute);
contextAttributeList.add(floorContextAttribute);
contextElement.setContextAttributeList(contextAttributeList);
ContextElementResponse contextElementResponse = new ContextElementResponse();
contextElementResponse.setContextElement(contextElement);
contextElementResponse.setStatusCode(new StatusCode(CodeEnum.CODE_200));
contextElementResponseList.add(contextElementResponse);
}
}
}
} else {
String id = queryEntityId.getId();
int varFloor = Integer.parseInt(String.valueOf(id.charAt(4)));
int varRoom = Integer.parseInt(String.valueOf(id.charAt(5)));
Random rand = new Random();
int i = rand.nextInt(tempTab.length);
int varTemp = tempTab[i];
ContextElement contextElement = new ContextElement();
contextElement.setEntityId(queryEntityId);
int value = ((varTemp + (2 * varFloor) + varRoom));
ContextAttribute tempContextAttribute =
new ContextAttribute("temperature", "double", String.valueOf(value));
ContextAttribute floorContextAttribute =
new ContextAttribute("floor", "string", "Floor" + varFloor);
List<ContextAttribute> contextAttributeList = new ArrayList<>();
contextAttributeList.add(tempContextAttribute);
contextAttributeList.add(floorContextAttribute);
contextElement.setContextAttributeList(contextAttributeList);
ContextElementResponse contextElementResponse = new ContextElementResponse();
contextElementResponse.setContextElement(contextElement);
contextElementResponse.setStatusCode(new StatusCode(CodeEnum.CODE_200));
contextElementResponseList.add(contextElementResponse);
}
queryContextResponse.setContextElementResponses(contextElementResponseList);
return queryContextResponse;
}
@Test
public void testGetContextEntityTypes_Error() throws Exception {
QueryContextResponse response = new QueryContextResponse();
response.setErrorCode(new StatusCode(CodeEnum.CODE_500));
when(ngsiController.queryContext(any())).thenReturn(response);
mockMvc
.perform(get("/v1/contextEntityTypes/TempSensor").accept(MediaType.APPLICATION_JSON))
.andExpect(status().isOk())
.andDo(mvcResult -> System.out.println(mvcResult.getResponse().getContentAsString()))
.andExpect(jsonPath("$.errorCode.code").value("500"))
.andExpect(jsonPath("$.contextResponses").doesNotExist());
verify(ngsiController).queryContext(any());
}
@Override
public QueryContextResponse queryContext(final QueryContext query) {
logger.info("=> queryContext: {}", query.toString());
// check if the incoming query is on Room
EntityId queryEntityId = query.getEntityIdList().get(0);
if (queryEntityId.getId().contains("Room")) {
return createRoomQueryResponse(queryEntityId);
} else if (queryEntityId.getId().contains("Shutter")) {
return createShutterQueryResponse(queryEntityId);
} else {
QueryContextResponse queryResponse = new QueryContextResponse();
queryResponse.setErrorCode(new StatusCode(CodeEnum.CODE_404, queryEntityId.getId()));
return queryResponse;
}
}
@Test
public void testGetContextElement_Error() throws Exception {
ArgumentCaptor<UpdateContext> updateContextArg = ArgumentCaptor.forClass(UpdateContext.class);
QueryContextResponse response = new QueryContextResponse();
response.setErrorCode(new StatusCode(CodeEnum.CODE_400));
when(ngsiController.queryContext(any())).thenReturn(response);
mockMvc
.perform(get("/v1/contextEntities/12345678").accept(MediaType.APPLICATION_JSON))
.andExpect(status().isOk())
.andExpect(jsonPath("$.statusCode.code").value("400"))
.andExpect(jsonPath("$.contextElement").doesNotExist());
verify(ngsiController).queryContext(any());
}
@Test
public void testGetContextEntityTypesAttribute_OK() throws Exception {
ArgumentCaptor<QueryContext> queryContextArg = ArgumentCaptor.forClass(QueryContext.class);
List attributes = Collections.singletonList(new ContextAttribute("test", "string", "OK"));
ContextElement contextElement = new ContextElement();
contextElement.setEntityId(new EntityId("12345678", "TempSensor", false));
contextElement.setContextAttributeList(attributes);
ContextElementResponse contextElementResponse =
new ContextElementResponse(contextElement, new StatusCode(CodeEnum.CODE_200));
QueryContextResponse response = new QueryContextResponse();
response.setContextElementResponses(Collections.singletonList(contextElementResponse));
when(ngsiController.queryContext(any())).thenReturn(response);
mockMvc
.perform(
get("/v1/contextEntityTypes/TempSensor/attributes/temp")
.accept(MediaType.APPLICATION_JSON))
.andExpect(status().isOk())
.andDo(mvcResult -> System.out.println(mvcResult.getResponse().getContentAsString()))
.andExpect(jsonPath("$.errorCode").doesNotExist())
.andExpect(jsonPath("$.contextResponses").isArray())
.andExpect(jsonPath("$.contextResponses[0].statusCode.code").value("200"))
.andExpect(jsonPath("$.contextResponses[0].contextElement.id").value("12345678"))
.andExpect(jsonPath("$.contextResponses[0].contextElement.type").value("TempSensor"))
.andExpect(jsonPath("$.contextResponses[0].contextElement.isPattern").value("false"))
.andExpect(
jsonPath("$.contextResponses[0].contextElement.attributes[0].name").value("test"))
.andExpect(
jsonPath("$.contextResponses[0].contextElement.attributes[0].type").value("string"))
.andExpect(
jsonPath("$.contextResponses[0].contextElement.attributes[0].value").value("OK"));
verify(ngsiController).queryContext(queryContextArg.capture());
QueryContext queryContext = queryContextArg.getValue();
assertNotNull(queryContext);
assertNotNull(queryContext.getEntityIdList());
assertEquals(1, queryContext.getEntityIdList().size());
assertNotNull(queryContext.getAttributeList());
assertEquals(".*", queryContext.getEntityIdList().get(0).getId());
assertEquals("TempSensor", queryContext.getEntityIdList().get(0).getType());
assertEquals(true, queryContext.getEntityIdList().get(0).getIsPattern());
assertEquals("temp", queryContext.getAttributeList().get(0));
}
private QueryContextResponse createShutterQueryResponse(EntityId queryEntityId) {
QueryContextResponse queryContextResponse = new QueryContextResponse();
List<ContextElementResponse> contextElementResponseList = new ArrayList<>();
Random rand = new Random();
if ((queryEntityId.getIsPattern()) && (queryEntityId.getId().equals("Shutter*"))) {
for (int varRoom = 1; varRoom < 5; varRoom++) {
for (int varFloor = 1; varFloor < 4; varFloor++) {
ContextElement contextElement = new ContextElement();
String name = "Shutter" + varFloor + varRoom;
EntityId entityId = new EntityId(name, "Room", false);
contextElement.setEntityId(entityId);
int i = rand.nextInt(statusTab.length);
String varStatus = statusTab[i];
ContextAttribute statusContextAttribute =
new ContextAttribute("status", "string", varStatus);
List<ContextAttribute> contextAttributeList = new ArrayList<>();
contextAttributeList.add(statusContextAttribute);
contextElement.setContextAttributeList(contextAttributeList);
ContextElementResponse contextElementResponse = new ContextElementResponse();
contextElementResponse.setContextElement(contextElement);
contextElementResponse.setStatusCode(new StatusCode(CodeEnum.CODE_200));
contextElementResponseList.add(contextElementResponse);
}
}
} else {
int i = rand.nextInt(statusTab.length);
String varStatus = statusTab[i];
ContextElement contextElement = new ContextElement();
contextElement.setEntityId(queryEntityId);
ContextAttribute statusContextAttribute = new ContextAttribute("status", "string", varStatus);
List<ContextAttribute> contextAttributeList = new ArrayList<>();
contextAttributeList.add(statusContextAttribute);
contextElement.setContextAttributeList(contextAttributeList);
ContextElementResponse contextElementResponse = new ContextElementResponse();
contextElementResponse.setContextElement(contextElement);
contextElementResponse.setStatusCode(new StatusCode(CodeEnum.CODE_200));
contextElementResponseList.add(contextElementResponse);
}
queryContextResponse.setContextElementResponses(contextElementResponseList);
return queryContextResponse;
}
@Test
public void testGetContextAttributeValue_NotImplemented() throws Exception {
ArgumentCaptor<QueryContext> queryContextArg = ArgumentCaptor.forClass(QueryContext.class);
List attributes = Collections.singletonList(new ContextAttribute("test", "string", "OK"));
ContextElement contextElement = new ContextElement();
contextElement.setEntityId(new EntityId("12345678", "", false));
contextElement.setContextAttributeList(attributes);
ContextElementResponse contextElementResponse =
new ContextElementResponse(contextElement, new StatusCode(CodeEnum.CODE_200));
QueryContextResponse response = new QueryContextResponse();
response.setContextElementResponses(Collections.singletonList(contextElementResponse));
when(ngsiController.queryContext(any())).thenReturn(response);
mockMvc
.perform(
get("/v1/contextEntities/12345678/attributes/test/DEADBEEF")
.accept(MediaType.APPLICATION_JSON))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value("500"));
verify(ngsiController, never()).queryContext(queryContextArg.capture());
}