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;
 
 /**
  * This test checks that streaming mode works with string literals
  */
 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);
  }
 }