@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_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());
  }
  @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));
  }