@Test
 public void shouldReturnCountOfDocuments() {
   // given
   String documentId = randomNumeric(5);
   SampleEntity sampleEntity = new SampleEntity();
   sampleEntity.setId(documentId);
   sampleEntity.setMessage("some message");
   sampleEntity.setVersion(System.currentTimeMillis());
   repository.save(sampleEntity);
   // when
   Long count = repository.count();
   // then
   assertThat(count, is(greaterThanOrEqualTo(1L)));
 }