creates custom type

Author: otaviojava

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

@@ -22,20 +22,15 @@
 import jakarta.nosql.document.DocumentEntity;
 import jakarta.nosql.document.DocumentQuery;
 import org.eclipse.jnosql.diana.document.Documents;
+import org.eclipse.jnosql.diana.mongodb.document.type.Money;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
 
 import java.io.IOException;
+import java.math.BigDecimal;
 import java.time.Duration;
 import java.time.LocalDate;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Date;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
-import java.util.Optional;
-import java.util.Random;
+import java.util.*;
 import java.util.concurrent.ThreadLocalRandom;
 import java.util.stream.Collectors;
 import java.util.stream.StreamSupport;
@@ -76,7 +71,7 @@ public void shouldThrowExceptionWhenInsertWithTTL() {
     }
 
     @Test
-    public void shouldUpdateSave() {
+    public void shouldUpdate() {
         DocumentEntity entity = getEntity();
         DocumentEntity documentEntity = entityManager.insert(entity);
         Document newField = Documents.of("newField", "10");
@@ -482,6 +477,22 @@ public void shouldCount() {
         assertTrue(entityManager.count(COLLECTION_NAME) > 0);
     }
 
+    @Test
+    public void shouldCustomTypeWork() {
+        DocumentEntity entity = getEntity();
+        Currency currency = Currency.getInstance("USD");
+        Money money = Money.of(currency, BigDecimal.valueOf(10D));
+        entity.add("money", money);
+        DocumentEntity documentEntity = entityManager.insert(entity);
+        Document id = documentEntity.find("_id").get();
+        DocumentQuery query = DocumentQuery.select().from(documentEntity.getName())
+                .where(id.getName()).eq(id.get()).build();
+
+        DocumentEntity result = entityManager.singleResult(query).get();
+        assertEquals(money, result.find("money").get().get(Money.class));
+
+    }
+
     private DocumentEntity createSubdocumentList() {
         DocumentEntity entity = DocumentEntity.of("AppointmentBook");
         entity.add(Document.of("_id", new Random().nextInt()));

mongodb-driver/src/test/java/org/eclipse/jnosql/diana/mongodb/document/type/Money.java

@@ -16,33 +16,58 @@
 package org.eclipse.jnosql.diana.mongodb.document.type;
 
 import java.math.BigDecimal;
+import java.util.Currency;
+import java.util.Objects;
 
 public class Money {
 
-    private final String currency;
+    private final Currency currency;
 
     private final BigDecimal value;
 
-    Money(String currency, BigDecimal value) {
+    private Money(Currency currency, BigDecimal value) {
         this.currency = currency;
         this.value = value;
     }
 
-    public String getCurrency() {
+    public Currency getCurrency() {
         return currency;
     }
 
     public BigDecimal getValue() {
         return value;
     }
 
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) {
+            return true;
+        }
+        if (o == null || getClass() != o.getClass()) {
+            return false;
+        }
+        Money money = (Money) o;
+        return Objects.equals(currency, money.currency) &&
+                Objects.equals(value, money.value);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(currency, value);
+    }
+
     @Override
     public String toString() {
-        return currency + ' ' + value;
+        return currency.getCurrencyCode() + ' ' + value;
+    }
+
+    public static Money of(Currency currency, BigDecimal value) {
+        return new Money(currency, value);
     }
 
     public static Money parse(String text) {
         String[] texts = text.split(" ");
-        return new Money(texts[0], BigDecimal.valueOf(Double.valueOf(texts[1])));
+        return new Money(Currency.getInstance(texts[0]),
+                BigDecimal.valueOf(Double.valueOf(texts[1])));
     }
 }