repeat until/whilst are combinable with times (#272)

* repeat until/whilst are combinable with `times` * toSetMutable for scala3
ShiftLeftSecurity · Jan 17, 2022 · d145bba · d145bba
1 parent 316765a
commit d145bba
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/traversal/src/test/scala/overflowdb/traversal/RepeatTraversalTests.scala b/traversal/src/test/scala/overflowdb/traversal/RepeatTraversalTests.scala
@@ -113,6 +113,11 @@ class RepeatTraversalTests extends AnyWordSpec {
       centerTrav.repeat(_.out)(_.until(_.hasLabel(Thing.Label))).property(Name).toSetMutable shouldBe expectedResults
       centerTrav.repeat(_.out)(_.until(_.hasLabel(Thing.Label)).breadthFirstSearch).property(Name).toSetMutable shouldBe expectedResults
     }
+
+    "be combinable with `.times`" in {
+      centerTrav.repeat(_.followedBy)(_.until(_.name("R2")).times(3)).name.toSetMutable shouldBe Set("L3", "R2")
+    }
+
   }
 
   "support repeat/while behaviour" should {
@@ -149,6 +154,10 @@ class RepeatTraversalTests extends AnyWordSpec {
         Seq(center, l1),
       )
     }
+
+    "be combinable with `.times`" in {
+      centerTrav.repeat(_.followedBy)(_.whilst(_.nameNot("R2")).times(3)).name.toSetMutable shouldBe Set("L3", "R2")
+    }
   }
 
   ".dedup should apply to all repeat iterations" when {