adds test to asyn

Author: otaviojava

elasticsearch-driver/src/test/java/org/jnosql/diana/elasticsearch/document/DocumentConfigurationUtils.java

@@ -34,7 +34,7 @@ final class DocumentConfigurationUtils {
     private DocumentConfigurationUtils() {
     }
 
-    public static ElasticsearchDocumentCollectionManagerFactory getConfiguration() {
+    public static ElasticsearchDocumentCollectionManagerFactory getFactory() {
         es.start();
         ElasticsearchDocumentConfiguration configuration = new ElasticsearchDocumentConfiguration();
         Map<String, Object> settings = new HashMap<>();

elasticsearch-driver/src/test/java/org/jnosql/diana/elasticsearch/document/ElasticsearchDocumentCollectionManagerAsyncTest.java

@@ -14,7 +14,6 @@
  */
 package org.jnosql.diana.elasticsearch.document;
 
-import org.awaitility.Awaitility;
 import org.elasticsearch.index.query.TermQueryBuilder;
 import org.jnosql.diana.api.document.Document;
 import org.jnosql.diana.api.document.DocumentCollectionManager;
@@ -32,13 +31,13 @@
 import java.util.concurrent.atomic.AtomicReference;
 import java.util.function.Consumer;
 
+import static org.awaitility.Awaitility.await;
 import static org.elasticsearch.index.query.QueryBuilders.termQuery;
 import static org.jnosql.diana.api.document.query.DocumentQueryBuilder.delete;
 import static org.jnosql.diana.api.document.query.DocumentQueryBuilder.select;
 import static org.jnosql.diana.elasticsearch.document.DocumentEntityGerator.COLLECTION_NAME;
 import static org.jnosql.diana.elasticsearch.document.DocumentEntityGerator.INDEX;
 import static org.jnosql.diana.elasticsearch.document.DocumentEntityGerator.getEntity;
-import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.api.Assertions.assertTrue;
@@ -53,8 +52,7 @@ public class ElasticsearchDocumentCollectionManagerAsyncTest {
 
     @BeforeEach
     public void setUp() {
-        ElasticsearchDocumentConfiguration configuration = new ElasticsearchDocumentConfiguration();
-        ElasticsearchDocumentCollectionManagerFactory managerFactory = configuration.get();
+        ElasticsearchDocumentCollectionManagerFactory managerFactory = DocumentConfigurationUtils.getFactory();
         entityManagerAsync = managerFactory.getAsync(COLLECTION_NAME);
         entityManager = managerFactory.get(INDEX);
         DocumentEntity documentEntity = getEntity();
@@ -101,9 +99,15 @@ public void shouldRemoveEntityAsync() throws InterruptedException {
         DocumentDeleteQuery deleteQuery = delete().from(COLLECTION_NAME).where(id.getName()).eq(id.get()).build();
         entityManagerAsync.delete(deleteQuery);
 
-        Thread.sleep(1_000L);
-        assertTrue(entityManager.select(query).isEmpty());
+        AtomicBoolean condition = new AtomicBoolean(false);
+        AtomicReference<List<DocumentEntity>> entities = new AtomicReference<>();
+        entityManagerAsync.select(query, l -> {
+            condition.set(true);
+            entities.set(l);
 
+        });
+        await().untilTrue(condition);
+        assertTrue(entities.get().isEmpty());
     }
 
     @Test
@@ -120,7 +124,7 @@ public void shouldUserSearchBuilder() throws InterruptedException {
             atomicBoolean.set(true);
         }, "person");
 
-        Awaitility.await().untilTrue(atomicBoolean);
+        await().untilTrue(atomicBoolean);
         List<DocumentEntity> account = result.get();
         assertFalse(account.isEmpty());
     }
@@ -138,7 +142,7 @@ public void shouldReturnAll() throws InterruptedException {
             condition.set(true);
             result.set(l);
         });
-        Awaitility.await().untilTrue(condition);
+        await().untilTrue(condition);
         List<DocumentEntity> entities = result.get();
         assertFalse(entities.isEmpty());
 
@@ -157,7 +161,7 @@ public void shouldCount() throws InterruptedException {
             value.set(l);
         };
         entityManagerAsync.count(DocumentEntityGerator.COLLECTION_NAME, callback);
-        Awaitility.await().untilTrue(condition);
+        await().untilTrue(condition);
         assertTrue(value.get() > 0);
 
     }

elasticsearch-driver/src/test/java/org/jnosql/diana/elasticsearch/document/ElasticsearchDocumentCollectionManagerTest.java

@@ -24,12 +24,10 @@
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 
-import java.sql.Time;
 import java.time.Duration;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
-import java.util.concurrent.TimeUnit;
 import java.util.stream.Collectors;
 
 import static java.util.Arrays.asList;
@@ -56,7 +54,7 @@ public class ElasticsearchDocumentCollectionManagerTest {
 
     @BeforeEach
     public void setUp() {
-        ElasticsearchDocumentCollectionManagerFactory managerFactory = DocumentConfigurationUtils.getConfiguration();
+        ElasticsearchDocumentCollectionManagerFactory managerFactory = DocumentConfigurationUtils.getFactory();
         entityManager = managerFactory.get(INDEX);
 
     }