Merge pull request #149 from eclipse/test_custom_type

Create test to custom Reader and Writer

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

@@ -0,0 +1,73 @@
+/*
+ *  Copyright (c) 2019 Otávio Santana and others
+ *   All rights reserved. This program and the accompanying materials
+ *   are made available under the terms of the Eclipse Public License v1.0
+ *   and Apache License v2.0 which accompanies this distribution.
+ *   The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
+ *   and the Apache License v2.0 is available at http://www.opensource.org/licenses/apache2.0.php.
+ *
+ *   You may elect to redistribute this code under either of these licenses.
+ *
+ *   Contributors:
+ *
+ *   Otavio Santana
+ */
+
+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 Currency currency;
+
+    private final BigDecimal value;
+
+    private Money(Currency currency, BigDecimal value) {
+        this.currency = currency;
+        this.value = value;
+    }
+
+    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.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(Currency.getInstance(texts[0]),
+                BigDecimal.valueOf(Double.valueOf(texts[1])));
+    }
+}

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

@@ -0,0 +1,30 @@
+/*
+ *  Copyright (c) 2019 Otávio Santana and others
+ *   All rights reserved. This program and the accompanying materials
+ *   are made available under the terms of the Eclipse Public License v1.0
+ *   and Apache License v2.0 which accompanies this distribution.
+ *   The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
+ *   and the Apache License v2.0 is available at http://www.opensource.org/licenses/apache2.0.php.
+ *
+ *   You may elect to redistribute this code under either of these licenses.
+ *
+ *   Contributors:
+ *
+ *   Otavio Santana
+ */
+package org.eclipse.jnosql.diana.mongodb.document.type;
+
+import jakarta.nosql.ValueReader;
+
+public class MoneyValueReader implements ValueReader {
+
+    @Override
+    public boolean isCompatible(Class clazz) {
+        return Money.class.equals(clazz);
+    }
+
+    @Override
+    public <T> T read(Class<T> clazz, Object value) {
+        return (T) Money.parse(value.toString());
+    }
+}

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

@@ -0,0 +1,30 @@
+/*
+ *  Copyright (c) 2019 Otávio Santana and others
+ *   All rights reserved. This program and the accompanying materials
+ *   are made available under the terms of the Eclipse Public License v1.0
+ *   and Apache License v2.0 which accompanies this distribution.
+ *   The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
+ *   and the Apache License v2.0 is available at http://www.opensource.org/licenses/apache2.0.php.
+ *
+ *   You may elect to redistribute this code under either of these licenses.
+ *
+ *   Contributors:
+ *
+ *   Otavio Santana
+ */
+package org.eclipse.jnosql.diana.mongodb.document.type;
+
+import jakarta.nosql.ValueWriter;
+
+public class MoneyValueWriter implements ValueWriter<Money, String> {
+
+    @Override
+    public boolean isCompatible(Class clazz) {
+        return Money.class.equals(clazz);
+    }
+
+    @Override
+    public String write(Money money) {
+        return money.toString();
+    }
+}

mongodb-driver/src/test/resources/META-INF/services/jakarta.nosql.ValueReader

@@ -0,0 +1 @@
+org.eclipse.jnosql.diana.mongodb.document.type.MoneyValueReader

mongodb-driver/src/test/resources/META-INF/services/jakarta.nosql.ValueWriter

@@ -0,0 +1 @@
+org.eclipse.jnosql.diana.mongodb.document.type.MoneyValueWriter