Reconnect

tests/ExponentialStrategyTest.php

@@ -0,0 +1,90 @@
+<?php
+
+declare(strict_types=1);
+
+use Nsq\Exception\ConnectionFail;
+use Nsq\Reconnect\ExponentialStrategy;
+use Nsq\Reconnect\TimeProvider;
+use PHPUnit\Framework\TestCase;
+
+final class ExponentialStrategyTest extends TestCase
+{
+    public function testTimeNotYetCome(): void
+    {
+        $timeProvider = new FakeTimeProvider();
+        $strategy = new ExponentialStrategy(
+            minDelay: 8,
+            maxDelay: 32,
+            timeProvider: $timeProvider,
+        );
+
+        $successConnect = static function (int $time = null) use ($strategy, $timeProvider): void {
+            $timeProvider($time);
+
+            $strategy->connect(static function (): void {
+            });
+        };
+        $failConnect = static function (int $time = null) use ($strategy, $timeProvider): void {
+            $timeProvider($time);
+
+            try {
+                $strategy->connect(function (): void {
+                    throw new ConnectionFail('Time come but failed');
+                });
+            } catch (ConnectionFail $e) {
+                self::assertSame('Time come but failed', $e->getMessage());
+
+                return;
+            }
+
+            self::fail('Expecting exception with message "Time come but failed"');
+        };
+        $timeNotCome = static function (int $time = null) use ($strategy, $timeProvider): void {
+            $timeProvider($time);
+
+            try {
+                $strategy->connect(function (): void {
+                    throw new ConnectionFail('');
+                });
+            } catch (ConnectionFail $e) {
+                self::assertSame('Time to reconnect has not yet come', $e->getMessage());
+
+                return;
+            }
+
+            self::fail('Was expecting exception with message "Time to reconnect has not yet come"');
+        };
+
+        $failConnect(0);
+        $timeNotCome(7);
+        $failConnect(8);
+        $timeNotCome(22);
+        $timeNotCome(13);
+        $failConnect(24);
+        $successConnect(56);
+        $failConnect();
+        $timeNotCome();
+        $timeNotCome(63);
+        $failConnect(64);
+
+        $this->expectException(ConnectionFail::class);
+        $this->expectExceptionMessage('Time to reconnect has not yet come');
+
+        $successConnect();
+    }
+}
+
+class FakeTimeProvider implements TimeProvider
+{
+    public int $time = 0;
+
+    public function time(): int
+    {
+        return $this->time;
+    }
+
+    public function __invoke(int $time = null): void
+    {
+        $this->time = $time ?? $this->time;
+    }
+}