Search in sources :

Example 56 with HugeConfig

use of com.baidu.hugegraph.config.HugeConfig in project incubator-hugegraph by apache.

the class BinarySerializerTest method testEdge.

@Test
public void testEdge() {
    HugeConfig config = FakeObjects.newConfig();
    BinarySerializer ser = new BinarySerializer(config);
    FakeObjects objects = new FakeObjects();
    HugeEdge edge1 = objects.newEdge(123, 456);
    HugeEdge edge2 = objects.newEdge(147, 789);
    BackendEntry entry1 = ser.writeEdge(edge1);
    HugeVertex vertex1 = ser.readVertex(edge1.graph(), entry1);
    Assert.assertEquals(1, vertex1.getEdges().size());
    HugeEdge edge = vertex1.getEdges().iterator().next();
    Assert.assertEquals(edge1, edge);
    assertCollectionEquals(edge1.getProperties(), edge.getProperties());
    BackendEntry entry2 = ser.writeEdge(edge2);
    HugeVertex vertex2 = ser.readVertex(edge1.graph(), entry2);
    Assert.assertEquals(1, vertex2.getEdges().size());
    edge = vertex2.getEdges().iterator().next();
    Assert.assertEquals(edge2, edge);
    assertCollectionEquals(edge2.getProperties(), edge.getProperties());
}
Also used : BackendEntry(com.baidu.hugegraph.backend.store.BackendEntry) FakeObjects(com.baidu.hugegraph.unit.FakeObjects) HugeEdge(com.baidu.hugegraph.structure.HugeEdge) HugeConfig(com.baidu.hugegraph.config.HugeConfig) HugeVertex(com.baidu.hugegraph.structure.HugeVertex) BinarySerializer(com.baidu.hugegraph.backend.serializer.BinarySerializer) BaseUnitTest(com.baidu.hugegraph.unit.BaseUnitTest) Test(org.junit.Test)

Example 57 with HugeConfig

use of com.baidu.hugegraph.config.HugeConfig in project incubator-hugegraph by apache.

the class BinarySerializerTest method testVertex.

@Test
public void testVertex() {
    HugeConfig config = FakeObjects.newConfig();
    BinarySerializer ser = new BinarySerializer(config);
    HugeEdge edge = new FakeObjects().newEdge(123, 456);
    BackendEntry entry1 = ser.writeVertex(edge.sourceVertex());
    HugeVertex vertex1 = ser.readVertex(edge.graph(), entry1);
    Assert.assertEquals(edge.sourceVertex(), vertex1);
    assertCollectionEquals(edge.sourceVertex().getProperties(), vertex1.getProperties());
    BackendEntry entry2 = ser.writeVertex(edge.targetVertex());
    HugeVertex vertex2 = ser.readVertex(edge.graph(), entry2);
    Assert.assertEquals(edge.targetVertex(), vertex2);
    assertCollectionEquals(edge.targetVertex().getProperties(), vertex2.getProperties());
    Whitebox.setInternalState(vertex2, "removed", true);
    Assert.assertTrue(vertex2.removed());
    BackendEntry entry3 = ser.writeVertex(vertex2);
    Assert.assertEquals(0, entry3.columnsSize());
    Assert.assertNull(ser.readVertex(edge.graph(), null));
}
Also used : BackendEntry(com.baidu.hugegraph.backend.store.BackendEntry) FakeObjects(com.baidu.hugegraph.unit.FakeObjects) HugeEdge(com.baidu.hugegraph.structure.HugeEdge) HugeConfig(com.baidu.hugegraph.config.HugeConfig) HugeVertex(com.baidu.hugegraph.structure.HugeVertex) BinarySerializer(com.baidu.hugegraph.backend.serializer.BinarySerializer) BaseUnitTest(com.baidu.hugegraph.unit.BaseUnitTest) Test(org.junit.Test)

Example 58 with HugeConfig

use of com.baidu.hugegraph.config.HugeConfig in project incubator-hugegraph by apache.

the class SerializerFactoryTest method testRegister.

@Test
public void testRegister() {
    HugeConfig config = FakeObjects.newConfig();
    SerializerFactory.register("fake", FakeSerializer.class.getName());
    Assert.assertEquals(FakeSerializer.class, SerializerFactory.serializer(config, "fake").getClass());
    Assert.assertThrows(BackendException.class, () -> {
        // exist
        SerializerFactory.register("fake", FakeSerializer.class.getName());
    }, e -> {
        Assert.assertContains("Exists serializer:", e.getMessage());
    });
    Assert.assertThrows(BackendException.class, () -> {
        // invalid class
        SerializerFactory.register("fake", "com.baidu.hugegraph.Invalid");
    }, e -> {
        Assert.assertContains("Invalid class:", e.getMessage());
    });
    Assert.assertThrows(BackendException.class, () -> {
        // subclass
        SerializerFactory.register("fake", "com.baidu.hugegraph.HugeGraph");
    }, e -> {
        Assert.assertContains("Class is not a subclass of class", e.getMessage());
    });
}
Also used : HugeConfig(com.baidu.hugegraph.config.HugeConfig) BaseUnitTest(com.baidu.hugegraph.unit.BaseUnitTest) Test(org.junit.Test)

Example 59 with HugeConfig

use of com.baidu.hugegraph.config.HugeConfig in project incubator-hugegraph by apache.

the class SecurityManagerTest method loadGraph.

private static HugeGraph loadGraph(boolean needClear) {
    HugeConfig config = FakeObjects.newConfig();
    HugeGraph graph = HugeFactory.open(config);
    if (needClear) {
        graph.clearBackend();
    }
    graph.initBackend();
    graph.serverStarted(IdGenerator.of("server1"), NodeRole.MASTER);
    return graph;
}
Also used : HugeGraph(com.baidu.hugegraph.HugeGraph) HugeConfig(com.baidu.hugegraph.config.HugeConfig)

Example 60 with HugeConfig

use of com.baidu.hugegraph.config.HugeConfig in project incubator-hugegraph by apache.

the class BaseRocksDBUnitTest method open.

private static RocksDBSessions open(String table) throws RocksDBException {
    HugeConfig config = FakeObjects.newConfig();
    RocksDBSessions rocks = new RocksDBStdSessions(config, "db", "store", DB_PATH, DB_PATH);
    rocks.createTable(table);
    return rocks;
}
Also used : RocksDBSessions(com.baidu.hugegraph.backend.store.rocksdb.RocksDBSessions) RocksDBStdSessions(com.baidu.hugegraph.backend.store.rocksdb.RocksDBStdSessions) HugeConfig(com.baidu.hugegraph.config.HugeConfig)

Aggregations

HugeConfig (com.baidu.hugegraph.config.HugeConfig)63 Test (org.junit.Test)33 PropertiesConfiguration (org.apache.commons.configuration2.PropertiesConfiguration)15 BaseUnitTest (com.baidu.hugegraph.unit.BaseUnitTest)12 Configuration (org.apache.commons.configuration2.Configuration)11 RpcClientProvider (com.baidu.hugegraph.rpc.RpcClientProvider)7 RpcServer (com.baidu.hugegraph.rpc.RpcServer)7 HugeGraph (com.baidu.hugegraph.HugeGraph)5 File (java.io.File)5 BackendEntry (com.baidu.hugegraph.backend.store.BackendEntry)4 HashMap (java.util.HashMap)4 RocksDBSessions (com.baidu.hugegraph.backend.store.rocksdb.RocksDBSessions)3 HugeEdge (com.baidu.hugegraph.structure.HugeEdge)3 HugeVertex (com.baidu.hugegraph.structure.HugeVertex)3 FakeObjects (com.baidu.hugegraph.unit.FakeObjects)3 MapConfiguration (org.apache.commons.configuration.MapConfiguration)3 HugeException (com.baidu.hugegraph.HugeException)2 BackendException (com.baidu.hugegraph.backend.BackendException)2 BinaryBackendEntry (com.baidu.hugegraph.backend.serializer.BinaryBackendEntry)2 BinaryScatterSerializer (com.baidu.hugegraph.backend.serializer.BinaryScatterSerializer)2