implements at mongodb

Author: otaviojava

mongodb-driver/src/test/java/org/jnosql/diana/mongodb/document/MongoDBDocumentCollectionManagerAsyncTest.java

@@ -32,7 +32,9 @@
 import java.util.Map;
 import java.util.Random;
 import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicLong;
 import java.util.concurrent.atomic.AtomicReference;
+import java.util.function.Consumer;
 
 import static org.awaitility.Awaitility.await;
 import static org.hamcrest.Matchers.notNullValue;
@@ -80,15 +82,16 @@ public void shouldThrowExceptionWhenInsertWithTTL() {
 
     @Test
     public void shouldThrowExceptionWhenInsertWithTTLWithCallback() {
-        assertThrows(UnsupportedOperationException.class, () -> entityManager.insert(getEntity(), Duration.ofSeconds(10), r -> {}));
+        assertThrows(UnsupportedOperationException.class, () -> entityManager.insert(getEntity(), Duration.ofSeconds(10), r -> {
+        }));
     }
 
     @Test
     public void shouldUpdateAsync() throws InterruptedException {
 
         Random random = new Random();
         long id = random.nextLong();
-        
+
         AtomicBoolean condition = new AtomicBoolean(false);
         AtomicReference<DocumentEntity> reference = new AtomicReference<>();
         DocumentEntity entity = getEntity();
@@ -192,6 +195,23 @@ public void shouldRemoveEntity() {
         entityManager.delete(deleteQuery);
     }
 
+    @Test
+    public void shouldCount() throws InterruptedException {
+        AtomicBoolean condition = new AtomicBoolean(false);
+        AtomicLong value = new AtomicLong(0);
+        DocumentEntity entity = getEntity();
+        entityManager.insert(entity, c -> condition.set(true));
+        await().untilTrue(condition);
+        condition.set(false);
+        Consumer<Long> callback = l -> {
+            condition.set(true);
+            value.set(l);
+        };
+        entityManager.count(COLLECTION_NAME, callback);
+        await().untilTrue(condition);
+        assertTrue(value.get() > 0);
+    }
+
     private DocumentEntity getEntity() {
         DocumentEntity entity = DocumentEntity.of(COLLECTION_NAME);
         Map<String, Object> map = new HashMap<>();

mongodb-driver/src/test/java/org/jnosql/diana/mongodb/document/MongoDBDocumentCollectionManagerTest.java

@@ -404,6 +404,12 @@ public void shouldRetrieveListSubdocumentList() {
         assertTrue(contacts.stream().allMatch(d -> d.size() == 3));
     }
 
+    @Test
+    public void shouldCount() {
+        DocumentEntity entity = entityManager.insert(getEntity());
+        assertTrue(entityManager.count(COLLECTION_NAME) > 0);
+    }
+
     private DocumentEntity createSubdocumentList() {
         DocumentEntity entity = DocumentEntity.of("AppointmentBook");
         entity.add(Document.of("_id", new Random().nextInt()));