38
38
39
39
import com .arangodb .ArangoDB .Builder ;
40
40
import com .arangodb .entity .ArangoDBVersion .License ;
41
+ import com .arangodb .entity .CollectionPropertiesEntity ;
41
42
import com .arangodb .entity .CollectionType ;
42
43
import com .arangodb .entity .EdgeDefinition ;
43
44
import com .arangodb .entity .GraphEntity ;
45
+ import com .arangodb .entity .ServerRole ;
44
46
import com .arangodb .model .CollectionCreateOptions ;
45
47
import com .arangodb .model .GraphCreateOptions ;
46
48
@@ -59,6 +61,8 @@ public class ArangoGraphTest extends BaseTest {
59
61
private static final String VERTEX_COL_2 = "db_vertex2_collection_test" ;
60
62
private static final String VERTEX_COL_3 = "db_vertex3_collection_test" ;
61
63
private static final String VERTEX_COL_4 = "db_vertex4_collection_test" ;
64
+ private static final Integer REPLICATION_FACTOR = 2 ;
65
+ private static final Integer NUMBER_OF_SHARDS = 2 ;
62
66
63
67
public ArangoGraphTest (final Builder builder ) {
64
68
super (builder );
@@ -88,7 +92,11 @@ public void setup() {
88
92
edgeDefinitions .add (new EdgeDefinition ().collection (EDGE_COL_1 ).from (VERTEX_COL_1 ).to (VERTEX_COL_2 ));
89
93
edgeDefinitions
90
94
.add (new EdgeDefinition ().collection (EDGE_COL_2 ).from (VERTEX_COL_2 ).to (VERTEX_COL_1 , VERTEX_COL_3 ));
91
- db .createGraph (GRAPH_NAME , edgeDefinitions , null );
95
+ final GraphCreateOptions options = new GraphCreateOptions ();
96
+ if (arangoDB .getRole () != ServerRole .SINGLE ) {
97
+ options .replicationFactor (REPLICATION_FACTOR ).numberOfShards (NUMBER_OF_SHARDS );
98
+ }
99
+ db .createGraph (GRAPH_NAME , edgeDefinitions , options );
92
100
}
93
101
94
102
@ After
@@ -190,6 +198,11 @@ public void addEdgeDefinition() {
190
198
assertThat (e .getTo (), hasItem (VERTEX_COL_2 ));
191
199
}
192
200
}
201
+ if (arangoDB .getRole () != ServerRole .SINGLE ) {
202
+ final CollectionPropertiesEntity properties = db .collection (EDGE_COL_3 ).getProperties ();
203
+ assertThat (properties .getReplicationFactor (), is (REPLICATION_FACTOR ));
204
+ assertThat (properties .getNumberOfShards (), is (NUMBER_OF_SHARDS ));
205
+ }
193
206
setup ();
194
207
}
195
208
0 commit comments