Neo4j2VertexTest.java

package com.tinkerpop.blueprints.impls.neo4j2;

import org.apache.commons.io.FileUtils;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.io.File;
import java.util.Arrays;

import static org.junit.Assert.assertEquals;

/**
 * @author mh
 * @since 08.01.14
 */
public class Neo4j2VertexTest {

    private Neo4j2Graph graph;

    @Before
    public void setUp() throws Exception {
        FileUtils.deleteDirectory(new File("target/test.db"));
        graph = new Neo4j2Graph("target/test.db");
    }

    @Test
    public void testVertexLabels() throws Exception {
        Neo4j2Vertex vertex = graph.addVertex(null);
        vertex.addLabel("Label");
        assertEquals(Arrays.asList("Label"), vertex.getLabels());
        vertex.addLabel("Label2");
        assertEquals(Arrays.asList("Label","Label2"), vertex.getLabels());
        vertex.removeLabel("Label2");
        assertEquals(Arrays.asList("Label"), vertex.getLabels());
        vertex.removeLabel("Label");
        assertEquals(Arrays.<String>asList(), vertex.getLabels());
    }

    @After
    public void tearDown() throws Exception {
        graph.shutdown();
    }
}