[10.x] Adds testing helpers for Precognition

src/Illuminate/Foundation/Testing/Concerns/MakesHttpRequests.php

@@ -309,6 +309,16 @@ public function from(string $url)
         return $this->withHeader('referer', $url);
     }
 
+    /**
+     * Set the precognition header.
+     *
+     * @return $this
+     */
+    public function withPrecognition()
+    {
+        return $this->withHeader('Precognition', 'true');
+    }
+
     /**
      * Visit the given URI with a GET request.
      *

src/Illuminate/Testing/TestResponse.php

@@ -1417,6 +1417,27 @@ public function assertSessionMissing($key)
         return $this;
     }
 
+    /**
+     * Assert that the Precognition request was successful.
+     *
+     * @return $this
+     */
+    public function assertPrecognitionSuccess()
+    {
+        $this->assertNoContent();
+
+        PHPUnit::assertTrue(
+            $this->headers->has('Precognition-Success'), 'Precognition-Success Header not present on response.'
+        );
+
+        PHPUnit::assertSame(
+            'true', $this->headers->get('Precognition-Success'),
+            'Precognition-Success Header was found, but the value is not `true`.'
+        );
+
+        return $this;
+    }
+
     /**
      * Get the current session store.
      *

tests/Foundation/Testing/Concerns/MakesHttpRequestsTest.php

@@ -4,6 +4,7 @@
 
 use Illuminate\Contracts\Routing\Registrar;
 use Illuminate\Contracts\Routing\UrlGenerator;
+use Illuminate\Foundation\Http\Middleware\HandlePrecognitiveRequests;
 use Illuminate\Http\RedirectResponse;
 use Orchestra\Testbench\TestCase;
 
@@ -186,6 +187,19 @@ public function testFollowingRedirectsTerminatesInExpectedOrder()
 
         $this->assertEquals(['from', 'to'], $callOrder);
     }
+
+    public function testWithPrecognition()
+    {
+        $this->withPrecognition();
+        $this->assertSame('true', $this->defaultHeaders['Precognition']);
+
+        $this->app->make(Registrar::class)
+            ->get('test-route', fn () => 'ok')->middleware(HandlePrecognitiveRequests::class);
+        $this->get('test-route')
+            ->assertStatus(204)
+            ->assertHeader('Precognition', 'true')
+            ->assertHeader('Precognition-Success', 'true');
+    }
 }
 
 class MyMiddleware

tests/Testing/TestResponseTest.php

@@ -1035,6 +1035,32 @@ public function testAssertHeaderMissing()
         $response->assertHeaderMissing('Location');
     }
 
+    public function testAssertPrecognitionSuccessfulWithMissingHeader()
+    {
+        $this->expectException(AssertionFailedError::class);
+        $this->expectExceptionMessage('Precognition-Success Header not present on response.');
+
+        $baseResponse = new Response('', 204);
+
+        $response = TestResponse::fromBaseResponse($baseResponse);
+
+        $response->assertPrecognitionSuccess();
+    }
+
+    public function testAssertPrecognitionSuccessfulWithIncorrectValue()
+    {
+        $this->expectException(AssertionFailedError::class);
+        $this->expectExceptionMessage('Precognition-Success Header was found, but the value is not `true`.');
+
+        $baseResponse = tap(new Response('', 204), function ($response) {
+            $response->header('Precognition-Success', '');
+        });
+
+        $response = TestResponse::fromBaseResponse($baseResponse);
+
+        $response->assertPrecognitionSuccess();
+    }
+
     public function testAssertJsonWithArray()
     {
         $response = TestResponse::fromBaseResponse(new Response(new JsonSerializableSingleResourceStub));