VertexWrapper { public Vertex getVertex() { return vertex; } VertexWrapper(Vertex v, ResourceDefinition resourceDefinition);  VertexWrapper(Vertex v,
                         PropertyMapper mapper,
                         Map<String, PropertyValueFormatter> formatters); Vertex getVertex(); T getProperty(String name); void setProperty(String name, Object value); Collection<String> getPropertyKeys(); Map<String, Object> getPropertyMap(); void removeProperty(String name); boolean isPropertyRemoved(String name); String toString();  }
@Test public void testGetVertex() { Vertex v = createStrictMock(Vertex.class); ResourceDefinition resourceDefinition = createStrictMock(ResourceDefinition.class); expect(resourceDefinition.getPropertyMapper()).andReturn(null); expect(resourceDefinition.getPropertyValueFormatters()).andReturn(null); expect(v.<String>getProperty(Constants.ENTITY_TYPE_PROPERTY_KEY)).andReturn("testType"); replay(v, resourceDefinition); VertexWrapper vWrapper = new VertexWrapper(v, resourceDefinition); assertEquals(vWrapper.getVertex(), v); verify(v, resourceDefinition); }