creates document entity test

jakartaee · Mar 9, 2021 · a7dab0d · a7dab0d
1 parent 203f0a7
commit a7dab0d
Showing 1 changed file with 50 additions and 1 deletion.
diff --git a/...k-document/src/main/java/jakarta/nosql/communication/tck/document/DocumentEntityTest.java b/...k-document/src/main/java/jakarta/nosql/communication/tck/document/DocumentEntityTest.java
@@ -16,6 +16,8 @@
 
 package jakarta.nosql.communication.tck.document;
 
+import jakarta.nosql.TypeReference;
+import jakarta.nosql.TypeSupplier;
 import jakarta.nosql.Value;
 import jakarta.nosql.document.Document;
 import jakarta.nosql.document.DocumentEntity;
@@ -71,6 +73,15 @@ public void shouldDoCopy() {
 
     }
 
+    @Test
+    public void shouldReturnErrorWhenFindHasNullParameter() {
+        Document document = Document.of("name", "name");
+        DocumentEntity entity = DocumentEntity.of("entity", singletonList(document));
+        Assertions.assertThrows(NullPointerException.class, () -> entity.find(null));
+        Assertions.assertThrows(NullPointerException.class, () -> entity.find("name", (Class<Object>) null));
+        Assertions.assertThrows(NullPointerException.class, () -> entity.find("name", (TypeSupplier<Object>) null));
+    }
+
     @Test
     public void shouldFindDocument() {
         Document document = Document.of("name", "name");
@@ -82,15 +93,53 @@ public void shouldFindDocument() {
         assertEquals(document, name.get());
     }
 
+
     @Test
-    public void shouldReturnErroWhenFindDocumentIsNull() {
+    public void shouldReturnErrorWhenFindDocumentIsNull() {
         Assertions.assertThrows(NullPointerException.class, () -> {
             Document document = Document.of("name", "name");
             DocumentEntity entity = DocumentEntity.of("entity", singletonList(document));
             entity.find(null);
         });
     }
 
+    @Test
+    public void shouldFindValue() {
+        Document column = Document.of("name", "name");
+        DocumentEntity entity = DocumentEntity.of("entity", singletonList(column));
+        Optional<String> name = entity.find("name", String.class);
+        Assertions.assertNotNull(name);
+        Assertions.assertTrue(name.isPresent());
+        Assertions.assertEquals("name", name.orElse(""));
+    }
+
+    @Test
+    public void shouldNotFindValue() {
+        Document column = Document.of("name", "name");
+        DocumentEntity entity = DocumentEntity.of("entity", singletonList(column));
+        Optional<String> notFound = entity.find("not_found", String.class);
+        Assertions.assertNotNull(notFound);
+        Assertions.assertFalse(notFound.isPresent());
+    }
+
+    @Test
+    public void shouldFindTypeSupplier() {
+        Document column = Document.of("name", "name");
+        DocumentEntity entity = DocumentEntity.of("entity", singletonList(column));
+        List<String> names = entity.find("name", new TypeReference<List<String>>() {});
+        Assertions.assertNotNull(names);
+        Assertions.assertFalse(names.isEmpty());
+    }
+
+    @Test
+    public void shouldNotFindTypeSupplier() {
+        Document column = Document.of("name", "name");
+        DocumentEntity entity = DocumentEntity.of("entity", singletonList(column));
+        List<String> names = entity.find("not_found", new TypeReference<List<String>>() {});
+        Assertions.assertNotNull(names);
+        Assertions.assertTrue(names.isEmpty());
+    }
+
     @Test
     public void shouldRemoveDocumentByName() {
         Document document = Document.of("name", "name");