expected = predicate.test(2L) ?
+ Collections.singletonList(2L) :
+ Collections.emptyList();
+ Assert.assertEquals(expected,
+ control.inject(1, 2).count().is(predicate.clone()).toList());
+ Assert.assertEquals(expected,
+ g.inject(1, 2).count().is(predicate.clone()).toList());
+ GraphTraversal, Long> traversal = __.count().is(predicate.clone());
+ HugeCountStrategy.instance().apply(traversal.asAdmin());
+ Assert.assertInstanceOf(CountGlobalStep.class,
+ traversal.asAdmin().getStartStep());
+ }
+ }
+ }
+
+ private void assertCountPredicatesKeepFinalResults(List> predicates) {
+ this.initSchema();
+ Vertex source = graph().addVertex(T.label, "person", "name", "source");
+ Vertex first = graph().addVertex(T.label, "person", "name", "first");
+ Vertex second = graph().addVertex(T.label, "person", "name", "second");
+ source.addEdge("knows", first);
+ source.addEdge("knows", second);
+ commitTx();
+
+ GraphTraversalSource g = graph().traversal();
+ GraphTraversalSource control = g.withoutStrategies(HugeCountStrategy.class);
+ for (P predicate : predicates) {
+ List