feat: enhance count tests in CouchbaseDocumentManager

Refactored count tests to use assertSoftly for better readability
and added additional assertions to verify count behavior with
null values and specific queries.

Signed-off-by: Maximillian Arruda <[email protected]>

(cherry picked from commit d6889ea)
Signed-off-by: Maximillian Arruda <[email protected]>

Author: dearrudam

jnosql-couchbase/src/test/java/org/eclipse/jnosql/databases/couchbase/communication/CouchbaseDocumentManagerTest.java

@@ -17,7 +17,6 @@
 import com.couchbase.client.core.error.DocumentNotFoundException;
 import com.couchbase.client.java.json.JsonObject;
 import org.assertj.core.api.Assertions;
-import org.assertj.core.api.SoftAssertions;
 import org.eclipse.jnosql.communication.Settings;
 import org.eclipse.jnosql.communication.TypeReference;
 import org.eclipse.jnosql.communication.keyvalue.BucketManager;
@@ -44,6 +43,7 @@
 
 import static java.util.Arrays.asList;
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.SoftAssertions.assertSoftly;
 import static org.awaitility.Awaitility.await;
 import static org.eclipse.jnosql.communication.driver.IntegrationTest.MATCHES;
 import static org.eclipse.jnosql.communication.driver.IntegrationTest.NAMED;
@@ -206,11 +206,29 @@ void shouldRetrieveListDocumentList() {
     }
 
     @Test
-   void shouldCount() {
-        CommunicationEntity entity = getEntity();
-        entityManager.insert(entity);
-        long counted = entityManager.count(COLLECTION_PERSON_NAME);
-        assertTrue(counted > 0);
+    void shouldCount() {
+        CommunicationEntity entity = entityManager.insert(createSubdocumentList());
+        Element key = entity.find("_id").get();
+        var query = select().from(COLLECTION_APP_NAME).where(key.name()).eq(key.get()).build();
+
+        assertSoftly(softly -> {
+            softly.assertThat(entityManager.count(entity.name()))
+                    .as("must return more than zero for person collection")
+                    .isEqualTo(1L);
+
+            softly.assertThat(entityManager.count(query))
+                    .isEqualTo(1L);
+
+            softly.assertThatThrownBy(()->
+                    entityManager.count((String) null))
+                    .isInstanceOf(NullPointerException.class);
+
+            softly.assertThatThrownBy(()->
+                    entityManager.count((SelectQuery) null))
+                    .isInstanceOf(NullPointerException.class);
+
+        });
+
     }
 
     @Test
@@ -222,7 +240,7 @@ void shouldFindContains() {
                 "lia")), COLLECTION_PERSON_NAME, Collections.emptyList());
 
         var result = entityManager.select(query).toList();
-        SoftAssertions.assertSoftly(softly -> {
+        assertSoftly(softly -> {
             softly.assertThat(result).hasSize(1);
             softly.assertThat(result.get(0).find("name").orElseThrow().get(String.class)).isEqualTo("Poliana");
         });
@@ -237,7 +255,7 @@ void shouldStartsWith() {
                 "Pol")), COLLECTION_PERSON_NAME, Collections.emptyList());
 
         var result = entityManager.select(query).toList();
-        SoftAssertions.assertSoftly(softly -> {
+        assertSoftly(softly -> {
             softly.assertThat(result).hasSize(1);
             softly.assertThat(result.get(0).find("name").orElseThrow().get(String.class)).isEqualTo("Poliana");
         });
@@ -252,7 +270,7 @@ void shouldEndsWith() {
                 "ana")), COLLECTION_PERSON_NAME, Collections.emptyList());
 
         var result = entityManager.select(query).toList();
-        SoftAssertions.assertSoftly(softly -> {
+        assertSoftly(softly -> {
             softly.assertThat(result).hasSize(1);
             softly.assertThat(result.get(0).find("name").orElseThrow().get(String.class)).isEqualTo("Poliana");
         });
@@ -322,7 +340,7 @@ void shouldInsertNull() {
         entity.add(Element.of("name", null));
         CommunicationEntity documentEntity = entityManager.insert(entity);
         Optional<Element> name = documentEntity.find("name");
-        SoftAssertions.assertSoftly(soft -> {
+        assertSoftly(soft -> {
             soft.assertThat(name).isPresent();
             soft.assertThat(name).get().extracting(Element::name).isEqualTo("name");
             soft.assertThat(name).get().extracting(Element::get).isNull();
@@ -335,7 +353,7 @@ void shouldUpdateNull(){
         entity.add(Element.of("name", null));
         var documentEntity = entityManager.update(entity);
         Optional<Element> name = documentEntity.find("name");
-        SoftAssertions.assertSoftly(soft -> {
+        assertSoftly(soft -> {
             soft.assertThat(name).isPresent();
             soft.assertThat(name).get().extracting(Element::name).isEqualTo("name");
             soft.assertThat(name).get().extracting(Element::get).isNull();