retry attempts fixed. Exception throwing for the retry() method unifi…

…ed for all cases

src/retry/src/Aeon/Retry/Retry.php

@@ -84,19 +84,7 @@ public function execute(callable $function)
          */
         $exceptions = [];
 
-        if ($this->retries === 0) {
-            $lastReturn = $function($this->lastExecution = new Execution(0));
-
-            $terminationException = $this->lastExecution->terminationException();
-
-            if ($terminationException) {
-                throw $terminationException;
-            }
-
-            return $lastReturn;
-        }
-
-        for ($retry = 0; $retry < $this->retries; $retry++) {
+        for ($retry = 0; $retry < ($this->retries + 1); $retry++) {
             try {
                 $lastReturn = $function($this->lastExecution = new Execution($retry, $exceptions));
 

src/retry/tests/Aeon/Retry/Tests/Unit/RetryTest.php

@@ -249,4 +249,37 @@ public function test_throws_the_last_exception() : void
         ))->modifyDelay(new ConstantDelay())
             ->execute($callable);
     }
+
+    /**
+     * @dataProvider retryExceptionDataProvider
+     */
+    public function test_throws_the_last_exception_for_given_retries($retries, $expectedException) : void
+    {
+        $this->expectExceptionMessage($expectedException);
+        $queue = [
+            new \RuntimeException('first'),
+            new \RuntimeException('second'),
+            new \RuntimeException('third'),
+        ];
+
+        $callable = function () use (&$queue) : void {
+            throw \array_shift($queue);
+        };
+
+        (new Retry(
+            new DummyProcess(),
+            $retries,
+            TimeUnit::seconds(3)
+        ))->modifyDelay(new ConstantDelay())
+            ->execute($callable);
+    }
+
+    public function retryExceptionDataProvider() : array
+    {
+        return [
+            [0, 'first'],
+            [1, 'second'],
+            [2, 'third'],
+        ];
+    }
 }