test: enhance count query tests for CassandraColumnManager

Improved the count query tests by adding assertions to check
for NullPointerExceptions when passing null arguments, ensuring
robustness and correctness of the count method implementation.

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

Author: dearrudam

jnosql-cassandra/src/test/java/org/eclipse/jnosql/databases/cassandra/communication/CassandraColumnManagerTest.java

@@ -48,6 +48,7 @@
 import static java.util.Collections.singletonMap;
 import static java.util.stream.Collectors.toList;
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.SoftAssertions.assertSoftly;
 import static org.eclipse.jnosql.communication.driver.IntegrationTest.MATCHES;
 import static org.eclipse.jnosql.communication.driver.IntegrationTest.NAMED;
 import static org.eclipse.jnosql.communication.semistructured.DeleteQuery.delete;
@@ -455,11 +456,29 @@ void shouldCount() {
     void shouldCountBySelectQuery() {
         entityManager.insert(getColumnFamily());
         var query = select().from(Constants.COLUMN_FAMILY).where("id").eq(10L).build();
-        assertThat(entityManager.count(query))
-                .isEqualTo(1);
+        assertSoftly(softly -> {
+
+            softly.assertThat(entityManager.count(query))
+                    .as("the implementation is returning an non expected value")
+                    .isEqualTo(1);
+
+            softly.assertThatThrownBy(() -> entityManager.count(query, null))
+                    .as("must throw NPE when call the count method passing null arguments")
+                    .isInstanceOf(NullPointerException.class);
+
+            softly.assertThatThrownBy(() -> entityManager.count(null, null))
+                    .as("must throw NPE when call the count method passing null arguments")
+                    .isInstanceOf(NullPointerException.class);
+
+            softly.assertThatThrownBy(() -> entityManager.count((SelectQuery) null))
+                    .as("must throw NPE when call the count method passing null arguments")
+                    .isInstanceOf(NullPointerException.class);
+
+        });
+
     }
 
-   @Test
+    @Test
     void shouldPagingState() {
         for (long index = 1; index <= 10; index++) {
             var columnFamily = getColumnFamily();
@@ -509,7 +528,7 @@ void shouldInsertNullValues(){
         family.add(Element.of("name", null));
         var columnEntity = entityManager.insert(family);
         var column = columnEntity.find("name");
-        SoftAssertions.assertSoftly(soft -> {
+        assertSoftly(soft -> {
             soft.assertThat(column).get().extracting(Element::name).isEqualTo("name");
             soft.assertThat(column).get().extracting(Element::get).isNull();
         });
@@ -522,7 +541,7 @@ void shouldUpdateNullValues(){
         family.addNull("name");
         var columnEntity = entityManager.update(family);
         var column = columnEntity.find("name");
-        SoftAssertions.assertSoftly(soft -> {
+        assertSoftly(soft -> {
             soft.assertThat(column).get().extracting(Element::name).isEqualTo("name");
             soft.assertThat(column).get().extracting(Element::get).isNull();
         });