org.umlg.sqlg.test.batch.TestBatchNormalNullUpdate Maven / Gradle / Ivy

Go to download
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.1
package org.umlg.sqlg.test.batch;

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.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.umlg.sqlg.test.BaseTest;

import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.time.ZonedDateTime;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;

/**
 * @author Pieter Martin (https://github.com/pietermartin)
 * Date: 2017/07/25
 */
@RunWith(Parameterized.class)
public class TestBatchNormalNullUpdate extends BaseTest {

    @Parameterized.Parameter
    public Object value;

    @Parameterized.Parameters
    public static Collection data() {
        return Arrays.asList(new Object[][]{{true}, {(byte)1}, {(short)1}, {1}, {1L}, {1F}, {1.111D}, {"haloThere"},
                {LocalDate.now()}, {LocalDateTime.now()}, {LocalTime.now().withNano(0)}, {ZonedDateTime.now()}});
    }

    @Test
    public void testBatchNormalNullUpdate() {
        if (value instanceof Float) {
            Assume.assumeTrue(this.sqlgGraph.getSqlDialect().supportsFloatValues());
        }
        Vertex a1 = this.sqlgGraph.addVertex(T.label, "A", "test", "test1", "name1", this.value);
        Vertex a2 = this.sqlgGraph.addVertex(T.label, "A", "test", "test2", "name2", this.value);
        Vertex a3 = this.sqlgGraph.addVertex(T.label, "A", "test", "test3", "name3", this.value);
        this.sqlgGraph.tx().commit();
        this.sqlgGraph.tx().normalBatchModeOn();
        a1.property("name1", this.value);
        a2.property("name2", this.value);
        a3.property("name3", this.value);
        this.sqlgGraph.tx().commit();

        List test1Vertices =  this.sqlgGraph.traversal().V().hasLabel("A").has("test", "test1").toList();
        Assert.assertEquals(1, test1Vertices.size());
        Vertex test1 = test1Vertices.get(0);
        Assert.assertEquals(this.value, test1.value("name1"));
        Assert.assertFalse(test1.property("name2").isPresent());
        Assert.assertFalse(test1.property("name3").isPresent());

        List test2Vertices =  this.sqlgGraph.traversal().V().hasLabel("A").has("test", "test2").toList();
        Assert.assertEquals(1, test2Vertices.size());
        Vertex test2 = test2Vertices.get(0);
        Assert.assertFalse(test2.property("name1").isPresent());
        Assert.assertEquals(this.value, test2.value("name2"));
        Assert.assertFalse(test2.property("name3").isPresent());

        List test3Vertices =  this.sqlgGraph.traversal().V().hasLabel("A").has("test", "test3").toList();
        Assert.assertEquals(1, test3Vertices.size());
        Vertex test3 = test3Vertices.get(0);
        Assert.assertFalse(test3.property("name1").isPresent());
        Assert.assertFalse(test3.property("name2").isPresent());
        Assert.assertEquals(this.value, test3.value("name3"));

    }

}