From c277c198d2e7ef3f5eaf23127071a1ed70743de3 Mon Sep 17 00:00:00 2001
From: George Kalpakas <kalpakas.g@gmail.com>
Date: Mon, 29 Oct 2018 11:18:23 +0200
Subject: [PATCH] test(docs-infra): make redirection tests more robust (#26649)

PR Close #26649
---
 aio/tests/deployment/e2e/redirection.e2e-spec.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/aio/tests/deployment/e2e/redirection.e2e-spec.ts b/aio/tests/deployment/e2e/redirection.e2e-spec.ts
index 481b92419a..d6229e0fd6 100644
--- a/aio/tests/deployment/e2e/redirection.e2e-spec.ts
+++ b/aio/tests/deployment/e2e/redirection.e2e-spec.ts
@@ -59,7 +59,9 @@ describe(browser.baseUrl, () => {
     it('should serve `index.html` for unknown pages', async () => {
       const aioShell = element(by.css('aio-shell'));
       const heading = aioShell.element(by.css('h1'));
+
       await page.goTo(unknownPagePath);
+      await browser.wait(() => page.getDocViewerText(), 5000);  // Wait for the document to be loaded.
 
       expect(aioShell.isPresent()).toBe(true);
       expect(heading.getText()).toMatch(/page not found/i);