org.umlg.sqlg.test.json.TestJsonUpdate Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of sqlg-test Show documentation
There is a newer version: 3.1.0
package org.umlg.sqlg.test.json;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import org.apache.tinkerpop.gremlin.structure.T;
import org.apache.tinkerpop.gremlin.structure.Vertex;
import org.junit.Assert;
import org.junit.Assume;
import org.junit.Before;
import org.junit.Test;
import org.umlg.sqlg.test.BaseTest;

import java.io.IOException;

import static org.junit.Assert.assertEquals;

/**
 * Date: 2016/05/23
 * Time: 5:33 PM
 */
public class TestJsonUpdate extends BaseTest {

    @Before
    public void beforeTest() {
        Assume.assumeTrue(this.sqlgGraph.getSqlDialect().supportsJson());
    }

    @Test
    public void testUpdateJson() {
        ObjectMapper objectMapper =  new ObjectMapper();
        ObjectNode json = new ObjectNode(objectMapper.getNodeFactory());
        json.put("username", "john");
        Vertex v1 = this.sqlgGraph.addVertex(T.label, "Person", "doc", json);
        this.sqlgGraph.tx().commit();
        v1 = this.sqlgGraph.traversal().V(v1.id()).next();
        json = new ObjectNode(objectMapper.getNodeFactory());
        json.put("username", "peter");
        v1.property("doc", json);
        this.sqlgGraph.tx().commit();
        JsonNode value = this.sqlgGraph.traversal().V(v1.id()).next().value("doc");
        assertEquals(json, value);
    }

    @Test
    public void testUpdateJson1() throws IOException {
        ObjectMapper objectMapper =  new ObjectMapper();
        String content = "{\"username\":\"robert\",\"posts\":100122,\"emailaddress\":\"[email protected]\"}";
        JsonNode jsonNode = objectMapper.readTree(content);
        Vertex v1 = this.sqlgGraph.addVertex(T.label, "Person", "doc", jsonNode);
        this.sqlgGraph.tx().commit();
        JsonNode value = this.sqlgGraph.traversal().V(v1.id()).next().value("doc");
        assertEquals(jsonNode.get("username"), value.get("username"));
        assertEquals(jsonNode.get("post"), value.get("post"));
        assertEquals(jsonNode.get("emailaddress"), value.get("emailaddress"));

        content = "{\"username\":\"peter\",\"posts\":100133,\"emailaddress\":\"[email protected]\"}";
        jsonNode = objectMapper.readTree(content);
        v1 = this.sqlgGraph.traversal().V(v1.id()).next();
        v1.property("doc", jsonNode);
        this.sqlgGraph.tx().commit();
        value = this.sqlgGraph.traversal().V(v1.id()).next().value("doc");
        assertEquals(jsonNode.get("username"), value.get("username"));
        assertEquals(jsonNode.get("post"), value.get("post"));
        assertEquals(jsonNode.get("emailaddress"), value.get("emailaddress"));
    }

    @Test
    public void testUpdateJsonArray() throws IOException {
        ObjectMapper objectMapper =  new ObjectMapper();
        ArrayNode jsonArray = new ArrayNode(objectMapper.getNodeFactory());
        ObjectNode john = new ObjectNode(objectMapper.getNodeFactory());
        john.put("username", "john");
        ObjectNode pete = new ObjectNode(objectMapper.getNodeFactory());
        pete.put("username", "pete");
        jsonArray.add(john);
        jsonArray.add(pete);
        Vertex v1 = this.sqlgGraph.addVertex(T.label, "Person", "users", jsonArray);
        this.sqlgGraph.tx().commit();

        jsonArray = new ArrayNode(objectMapper.getNodeFactory());
        john = new ObjectNode(objectMapper.getNodeFactory());
        john.put("username", "john1");
        pete = new ObjectNode(objectMapper.getNodeFactory());
        pete.put("username", "pete1");
        jsonArray.add(john);
        jsonArray.add(pete);

        v1 = this.sqlgGraph.traversal().V(v1.id()).next();
        v1.property("users", jsonArray);
        this.sqlgGraph.tx().commit();

        JsonNode value = this.sqlgGraph.traversal().V(v1.id()).next().value("users");
        assertEquals(jsonArray, value);
    }

    @Test
    public void testUpdateJsonArraysForObjectNodes() {
        ObjectMapper objectMapper =  new ObjectMapper();
        ObjectNode json1 = new ObjectNode(objectMapper.getNodeFactory());
        json1.put("username", "john1");
        ObjectNode json2 = new ObjectNode(objectMapper.getNodeFactory());
        json2.put("username", "john2");
        ObjectNode[] objectNodes = new ObjectNode[]{json1, json2};

        Vertex v1 = this.sqlgGraph.addVertex(T.label, "Person", "docs", objectNodes);
        this.sqlgGraph.tx().commit();
        JsonNode[] value = this.sqlgGraph.traversal().V(v1.id()).next().value("docs");
        Assert.assertArrayEquals(objectNodes, value);

        json1 = new ObjectNode(objectMapper.getNodeFactory());
        json1.put("username", "john11");
        json2 = new ObjectNode(objectMapper.getNodeFactory());
        json2.put("username", "john22");
        objectNodes = new ObjectNode[]{json1, json2};

        v1 = this.sqlgGraph.traversal().V(v1).next();
        v1.property("docs", objectNodes);
        this.sqlgGraph.tx().commit();
        value = this.sqlgGraph.traversal().V(v1.id()).next().value("docs");
        Assert.assertArrayEquals(objectNodes, value);

    }

    @Test
    public void testUpdateJsonArraysForArrayNode() {
        ObjectMapper objectMapper =  new ObjectMapper();
        ArrayNode jsonArray1 = new ArrayNode(objectMapper.getNodeFactory());
        ObjectNode john = new ObjectNode(objectMapper.getNodeFactory());
        john.put("username", "john");
        ObjectNode pete = new ObjectNode(objectMapper.getNodeFactory());
        pete.put("username", "pete");
        jsonArray1.add(john);
        jsonArray1.add(pete);

        ArrayNode jsonArray2 = new ArrayNode(objectMapper.getNodeFactory());
        ObjectNode john2 = new ObjectNode(objectMapper.getNodeFactory());
        john2.put("username", "john2");
        ObjectNode pete2 = new ObjectNode(objectMapper.getNodeFactory());
        pete2.put("username", "pete2");
        jsonArray2.add(john2);
        jsonArray2.add(pete2);

        ArrayNode[] arrayNodes = new ArrayNode[]{jsonArray1, jsonArray2};
        Vertex v1 = this.sqlgGraph.addVertex(T.label, "Person", "docs", arrayNodes);
        this.sqlgGraph.tx().commit();
        JsonNode[] value = this.sqlgGraph.traversal().V(v1.id()).next().value("docs");
        Assert.assertArrayEquals(arrayNodes, value);

        jsonArray1 = new ArrayNode(objectMapper.getNodeFactory());
        john = new ObjectNode(objectMapper.getNodeFactory());
        john.put("username", "john");
        pete = new ObjectNode(objectMapper.getNodeFactory());
        pete.put("username", "pete");
        jsonArray1.add(john);
        jsonArray1.add(pete);

        jsonArray2 = new ArrayNode(objectMapper.getNodeFactory());
        john2 = new ObjectNode(objectMapper.getNodeFactory());
        john2.put("username", "john2");
        pete2 = new ObjectNode(objectMapper.getNodeFactory());
        pete2.put("username", "pete2");
        jsonArray2.add(john2);
        jsonArray2.add(pete2);

        arrayNodes = new ArrayNode[]{jsonArray1, jsonArray2};

        v1 = this.sqlgGraph.traversal().V(v1.id()).next();
        v1.property("docs", arrayNodes);
        this.sqlgGraph.tx().commit();

        value = this.sqlgGraph.traversal().V(v1.id()).next().value("docs");
        Assert.assertArrayEquals(arrayNodes, value);
    }
}