regenerate the value of Set\Call when shrinking

Innmind · Jul 29, 2023 · 3a13c2b · 3a13c2b
1 parent 4c4c93a
commit 3a13c2b
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 8 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,11 @@
 # Changelog
 
+## 5.1.1 - 2023-07-29
+
+### Changed
+
+- `Innmind\BlackBox\Set\Call` now regenerate the value when shrinking
+
 ## 5.1.0 - 2023-07-14
 
 ### Added

diff --git a/proofs/set/call.php b/proofs/set/call.php
@@ -10,12 +10,26 @@
     yield test(
         'Set\Call always return a new value',
         static function($assert) {
-            $set = Set\Call::of(static fn() => new \stdClass);
+            $set = Set\Call::of(static fn() => new \stdClass)->values(Random::default);
+            $current = $set->current()->unwrap();
+            $set->next();
 
             $assert
-                ->expected($set->values(Random::default)->current()->unwrap())
+                ->expected($current)
                 ->not()
-                ->same($set->values(Random::default)->current()->unwrap());
+                ->same($set->current()->unwrap());
+        },
+    );
+    yield test(
+        'Set\Call regenerate the value when shrinking',
+        static function($assert) {
+            $set = Set\Call::of(static fn() => new \stdClass)->values(Random::default);
+            $current = $set->current();
+
+            $assert
+                ->expected($current->unwrap())
+                ->not()
+                ->same($current->shrink()->a()->unwrap());
         },
     );
 };
diff --git a/src/Set/Call.php b/src/Set/Call.php
@@ -16,10 +16,6 @@ final class Call
      */
     public static function of(callable $call): Set
     {
-        return FromGenerator::of(static function() use ($call) {
-            while (true) {
-                yield $call();
-            }
-        });
+        return Integers::any()->map(static fn() => $call());
     }
 }