tests for withTtl

Author: gam6itko

src/RoadRunnerStore.php

@@ -31,8 +31,12 @@ public function __construct(
         \assert($this->initialWaitTtl >= 0);
     }
 
-    public function withTtl(float $ttl, float $waitTtl = 0): self
+    /**
+     * Clone current instance with another values of ttl.
+     */
+    public function withTtl(float $ttl, ?float $waitTtl = null): self
     {
+        $waitTtl ??= $this->initialWaitTtl;
         return new self($this->lock, $this->tokens, $ttl, $waitTtl);
     }
 

tests/RoadRunnerStoreTest.php

@@ -190,4 +190,45 @@ public function testWaitAndSaveFail(): void
         $store = new RoadRunnerStore($this->rrLock, $this->tokens);
         $store->waitAndSave(new Key('resource-name'));
     }
+
+    /**
+     * @dataProvider dataWithTtl
+     */
+    public function testWithTtl(float $ttl, ?float $waitTtl, float $ttlExp, float $waitTtlExp): void
+    {
+        $this->rrLock->expects($this->once())
+            ->method('lock')
+            ->with('resource-name', 'random-id', $ttlExp, $waitTtlExp)
+            ->willReturn('lock-id');
+
+        $s = new RoadRunnerStore($this->rrLock, $this->tokens);
+        $s->withTtl($ttl, $waitTtl)->save(new Key('resource-name'));
+    }
+
+    /**
+     * @return iterable
+     */
+    public static function dataWithTtl(): iterable
+    {
+        yield [
+            100,
+            null,
+            100,
+            0,
+        ];
+
+        yield [
+            0.1,
+            0.1,
+            0.1,
+            0.1,
+        ];
+
+        yield [
+            0,
+            0,
+            0,
+            0,
+        ];
+    }
 }