Coverage Summary for Class: TestStreamingFromCsvImport (org.umlg.sqlg.test.batch)
Class |
Method, %
|
Line, %
|
TestStreamingFromCsvImport |
100%
(4/4)
|
100%
(17/17)
|
TestStreamingFromCsvImport$1 |
100%
(1/1)
|
100%
(5/5)
|
TestStreamingFromCsvImport$2 |
100%
(1/1)
|
100%
(5/5)
|
TestStreamingFromCsvImport$3 |
100%
(1/1)
|
100%
(4/4)
|
TestStreamingFromCsvImport$4 |
100%
(1/1)
|
100%
(4/4)
|
TestStreamingFromCsvImport$5 |
100%
(1/1)
|
100%
(4/4)
|
Total |
100%
(9/9)
|
100%
(39/39)
|
| package org.umlg.sqlg.test.batch; |
| |
| import org.apache.tinkerpop.gremlin.structure.T; |
| import org.junit.Assert; |
| import org.junit.Assume; |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.umlg.sqlg.structure.PropertyDefinition; |
| import org.umlg.sqlg.structure.PropertyType; |
| import org.umlg.sqlg.structure.SqlgVertex; |
| import org.umlg.sqlg.test.BaseTest; |
| |
| import java.time.LocalDateTime; |
| import java.util.LinkedHashMap; |
| |
| |
| |
| |
| public class TestStreamingFromCsvImport extends BaseTest { |
| |
| @Before |
| public void beforeTest() { |
| Assume.assumeTrue(this.sqlgGraph.getSqlDialect().supportsStreamingBatchMode()); |
| } |
| |
| @Test |
| public void testStreamCsvEdge() { |
| SqlgVertex v1 = (SqlgVertex) this.sqlgGraph.addVertex(T.label, "A"); |
| SqlgVertex v2 = (SqlgVertex) this.sqlgGraph.addVertex(T.label, "A"); |
| this.sqlgGraph.tx().commit(); |
| this.sqlgGraph.tx().streamingBatchModeOn(); |
| v1.streamEdge("a", v2, new LinkedHashMap<>() {{ |
| put("a", "a1"); |
| put("b", "1"); |
| put("c", 1.1); |
| put("d", LocalDateTime.now()); |
| }}); |
| v1.streamEdge("a", v2, new LinkedHashMap<>() {{ |
| put("a", "a1"); |
| put("b", "1"); |
| put("c", "2.2"); |
| put("d", LocalDateTime.now().toString()); |
| }}); |
| this.sqlgGraph.tx().commit(); |
| Assert.assertEquals(2, this.sqlgGraph.traversal().V(v1).out().count().next(), 0); |
| } |
| |
| @Test |
| public void testStreamVertexLabelCsv() { |
| this.sqlgGraph.getTopology().getPublicSchema().ensureVertexLabelExist( |
| "Test", |
| new LinkedHashMap<>() {{ |
| put("a", PropertyDefinition.of(PropertyType.STRING)); |
| put("b", PropertyDefinition.of(PropertyType.INTEGER)); |
| put("c", PropertyDefinition.of(PropertyType.DOUBLE)); |
| }} |
| ); |
| this.sqlgGraph.tx().commit(); |
| this.sqlgGraph.tx().streamingBatchModeOn(); |
| this.sqlgGraph.streamVertex("Test", new LinkedHashMap<>() {{ |
| put("a", "a1"); |
| put("b", "1"); |
| put("c", 1.1); |
| }}); |
| this.sqlgGraph.streamVertex("Test", new LinkedHashMap<>() {{ |
| put("a", "a1"); |
| put("b", "1"); |
| put("c", "2.2"); |
| }}); |
| this.sqlgGraph.tx().commit(); |
| Assert.assertEquals(2, this.sqlgGraph.traversal().V().count().next(), 0); |
| } |
| } |