1 package org.openecomp.sdc.be.dao.jsongraph;
6 import org.apache.tinkerpop.gremlin.structure.Edge;
7 import org.apache.tinkerpop.gremlin.structure.Vertex;
9 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
10 import org.openecomp.sdc.be.dao.jsongraph.types.EdgePropertyEnum;
11 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
12 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
13 import org.openecomp.sdc.be.dao.titan.TitanGraphClient;
14 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
15 import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
17 import com.thinkaurelius.titan.core.TitanGraph;
19 import fj.data.Either;
22 public class TitanDaoTest {
24 private TitanDao createTestSubject() {
25 return new TitanDao(new TitanGraphClient());
30 public void testCommit() throws Exception {
32 TitanOperationStatus result;
35 testSubject = createTestSubject();
36 result = testSubject.commit();
41 public void testRollback() throws Exception {
43 TitanOperationStatus result;
46 testSubject = createTestSubject();
47 result = testSubject.rollback();
52 public void testGetGraph() throws Exception {
54 Either<TitanGraph, TitanOperationStatus> result;
57 testSubject = createTestSubject();
58 result = testSubject.getGraph();
66 public void testGetVertexByPropertyAndLabel() throws Exception {
68 GraphPropertyEnum name = null;
70 VertexTypeEnum label = null;
71 Either<GraphVertex, TitanOperationStatus> result;
74 testSubject = createTestSubject();
75 result = testSubject.getVertexByPropertyAndLabel(name, value, label);
80 public void testGetVertexByPropertyAndLabel_1() throws Exception {
82 GraphPropertyEnum name = null;
84 VertexTypeEnum label = null;
85 JsonParseFlagEnum parseFlag = null;
86 Either<GraphVertex, TitanOperationStatus> result;
89 testSubject = createTestSubject();
90 result = testSubject.getVertexByPropertyAndLabel(name, value, label, parseFlag);
95 public void testGetVertexById() throws Exception {
98 Either<GraphVertex, TitanOperationStatus> result;
101 testSubject = createTestSubject();
102 result = testSubject.getVertexById(id);
122 public void testGetByCriteria() throws Exception {
123 TitanDao testSubject;
124 VertexTypeEnum type = null;
125 Map<GraphPropertyEnum, Object> props = null;
126 Either<List<GraphVertex>, TitanOperationStatus> result;
129 testSubject = createTestSubject();
130 result = testSubject.getByCriteria(type, props);
135 public void testGetByCriteria_1() throws Exception {
136 TitanDao testSubject;
137 VertexTypeEnum type = null;
138 Map<GraphPropertyEnum, Object> props = null;
139 JsonParseFlagEnum parseFlag = null;
140 Either<List<GraphVertex>, TitanOperationStatus> result;
143 testSubject = createTestSubject();
144 result = testSubject.getByCriteria(type, props, parseFlag);
149 public void testGetByCriteria_2() throws Exception {
150 TitanDao testSubject;
151 VertexTypeEnum type = null;
152 Map<GraphPropertyEnum, Object> props = null;
153 Map<GraphPropertyEnum, Object> hasNotProps = null;
154 JsonParseFlagEnum parseFlag = null;
155 Either<List<GraphVertex>, TitanOperationStatus> result;
158 testSubject = createTestSubject();
159 result = testSubject.getByCriteria(type, props, hasNotProps, parseFlag);
167 public void testGetChildVertex() throws Exception {
168 TitanDao testSubject;
169 GraphVertex parentVertex = null;
170 EdgeLabelEnum edgeLabel = null;
171 JsonParseFlagEnum parseFlag = null;
172 Either<GraphVertex, TitanOperationStatus> result;
175 testSubject = createTestSubject();
176 result = testSubject.getChildVertex(parentVertex, edgeLabel, parseFlag);
181 public void testGetParentVertex() throws Exception {
182 TitanDao testSubject;
183 GraphVertex parentVertex = null;
184 EdgeLabelEnum edgeLabel = null;
185 JsonParseFlagEnum parseFlag = null;
186 Either<GraphVertex, TitanOperationStatus> result;
189 testSubject = createTestSubject();
190 result = testSubject.getParentVertex(parentVertex, edgeLabel, parseFlag);
195 public void testGetChildrenVertecies() throws Exception {
196 TitanDao testSubject;
197 GraphVertex parentVertex = null;
198 EdgeLabelEnum edgeLabel = null;
199 JsonParseFlagEnum parseFlag = null;
200 Either<List<GraphVertex>, TitanOperationStatus> result;
203 testSubject = createTestSubject();
204 result = testSubject.getChildrenVertecies(parentVertex, edgeLabel, parseFlag);
209 public void testGetParentVertecies() throws Exception {
210 TitanDao testSubject;
211 GraphVertex parentVertex = null;
212 EdgeLabelEnum edgeLabel = null;
213 JsonParseFlagEnum parseFlag = null;
214 Either<List<GraphVertex>, TitanOperationStatus> result;
217 testSubject = createTestSubject();
218 result = testSubject.getParentVertecies(parentVertex, edgeLabel, parseFlag);
247 public void testUpdateVertexMetadataPropertiesWithJson() throws Exception {
248 TitanDao testSubject;
249 Vertex vertex = null;
250 Map<GraphPropertyEnum, Object> properties = null;
251 TitanOperationStatus result;
254 testSubject = createTestSubject();
255 result = testSubject.updateVertexMetadataPropertiesWithJson(vertex, properties);
265 public void testGetProperty_1() throws Exception {
266 TitanDao testSubject;
268 EdgePropertyEnum key = null;
272 testSubject = createTestSubject();
273 result = testSubject.getProperty(edge, key);
280 public void testGetBelongingEdgeByCriteria_1() throws Exception {
281 TitanDao testSubject;
282 String parentId = "";
283 EdgeLabelEnum label = null;
284 Map<GraphPropertyEnum, Object> properties = null;
285 Either<Edge, TitanOperationStatus> result;
288 testSubject = createTestSubject();
289 result = testSubject.getBelongingEdgeByCriteria(parentId, label, properties);