@Test
public void testSaveSingle() throws Exception {

    Document expectedDocument = new Document();
    expectedDocument.setId(UUID.randomUUID().toString());
    expectedDocument.setTimestamp(System.currentTimeMillis());
    JsonNode data = mapper.valueToTree(Collections.singletonMap("TEST_NAME", "SINGLE_SAVE_TEST"));
    expectedDocument.setData(data);
    queryStore.save(TestUtils.TEST_TABLE_NAME, expectedDocument);
    GetResponse getResponse = elasticsearchServer
            .getClient()
            .prepareGet(ElasticsearchUtils.getCurrentIndex(TestUtils.TEST_TABLE_NAME, expectedDocument.getTimestamp()),
                    ElasticsearchUtils.TYPE_NAME,
                    expectedDocument.getId())
            .setFields("_timestamp").execute().actionGet();
    assertTrue("Id should exist in ES", getResponse.isExists());
    assertEquals("Id should match requestId", expectedDocument.getId(), getResponse.getId());
    assertEquals("Timestamp should match request timestamp", expectedDocument.getTimestamp(), getResponse.getField("_timestamp").getValue());
}