@@ -137,6 +137,24 @@ else if(testMethodName.startsWith("shouldReadWriteVertexWithOUTEdges")) {
137137 else if (testMethodName .startsWith ("shouldReadWriteDetachedVertexNoEdges" )) {
138138 builder .withEdgeCollection ("friends" );
139139 }
140+ else if (testMethodName .startsWith ("shouldReadWriteDetachedEdge" )) {
141+ builder .withVertexCollection ("person" );
142+ builder .withEdgeCollection ("friend" );
143+ }
144+ else if (testMethodName .startsWith ("shouldReadWriteDetachedEdgeAsReference" )) {
145+ builder .withVertexCollection ("person" );
146+ builder .withEdgeCollection ("friend" );
147+ }
148+ else if (testMethodName .startsWith ("shouldReadWriteEdge" )) {
149+ builder .withVertexCollection ("person" );
150+ builder .withEdgeCollection ("friend" );
151+ }
152+ else if (testMethodName .startsWith ("shouldThrowOnGraphEdgeSetPropertyStandard" )) {
153+ builder .withEdgeCollection ("self" );
154+ }
155+ else if (testMethodName .startsWith ("shouldThrowOnGraphAddEdge" )) {
156+ builder .withEdgeCollection ("self" );
157+ }
140158 else {
141159 // Perhaps change for startsWith, but then it would be more verbose. Perhaps a set?
142160 switch (testMethodName ) {
@@ -203,6 +221,30 @@ else if(testMethodName.startsWith("shouldReadWriteDetachedVertexNoEdges")) {
203221 break ;
204222 case "shouldAddVertexWithLabel" :
205223 builder .withVertexCollection ("person" );
224+ break ;
225+ case "shouldNotAllowSetProperty" :
226+ case "shouldHashAndEqualCorrectly" :
227+ case "shouldNotAllowRemove" :
228+ case "shouldNotConstructNewWithSomethingAlreadyDetached" :
229+ case "shouldNotConstructNewWithSomethingAlreadyReferenced" :
230+ builder .withEdgeCollection ("test" );
231+ break ;
232+ case "shouldHaveExceptionConsistencyWhenUsingNullVertex" :
233+ builder .withEdgeCollection ("tonothing" );
234+ break ;
235+ case "shouldHandleSelfLoops" :
236+ builder .withVertexCollection ("person" );
237+ builder .withEdgeCollection ("self" );
238+ break ;
239+ case "shouldAttachWithCreateMethod" :
240+ builder .withVertexCollection ("person" );
241+ builder .withVertexCollection ("project" );
242+ builder .withEdgeCollection ("knows" );
243+ builder .withEdgeCollection ("developedBy" );
244+ break ;
245+ case "shouldCreateVertex" :
246+ builder .withVertexCollection ("dog" );
247+ break ;
206248 default :
207249 System .out .println ("case \" " + testMethodName + "\" :" );
208250 }
0 commit comments