From 6236010693e5350019cc8d21316d444a57fc7767 Mon Sep 17 00:00:00 2001
From: JJ Kasper <jj@jjsweb.site>
Date: Mon, 11 Jan 2021 13:14:42 -0600
Subject: [PATCH] Update next/image test

---
 .../image-component/custom-resolver/pages/client-side.js     | 2 +-
 .../image-component/custom-resolver/pages/index.js           | 2 +-
 .../image-component/custom-resolver/test/index.test.js       | 5 ++++-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/test/integration/image-component/custom-resolver/pages/client-side.js b/test/integration/image-component/custom-resolver/pages/client-side.js
index d702f8c0d71c3..47d6a70d5466f 100644
--- a/test/integration/image-component/custom-resolver/pages/client-side.js
+++ b/test/integration/image-component/custom-resolver/pages/client-side.js
@@ -12,7 +12,7 @@ const MyImage = (props) => {
 const Page = () => {
   return (
     <div>
-      <p>Image Client Side Test</p>
+      <p id="client-side">Image Client Side Test</p>
       <MyImage
         id="basic-image"
         src="foo.jpg"
diff --git a/test/integration/image-component/custom-resolver/pages/index.js b/test/integration/image-component/custom-resolver/pages/index.js
index fd8db6f3e72d0..2b534c9f9b628 100644
--- a/test/integration/image-component/custom-resolver/pages/index.js
+++ b/test/integration/image-component/custom-resolver/pages/index.js
@@ -13,7 +13,7 @@ const MyImage = (props) => {
 const Page = () => {
   return (
     <div>
-      <p>Image SSR Test</p>
+      <p id="ssr">Image SSR Test</p>
       <MyImage
         id="basic-image"
         src="foo.jpg"
diff --git a/test/integration/image-component/custom-resolver/test/index.test.js b/test/integration/image-component/custom-resolver/test/index.test.js
index c7a9155dbdabb..ad576ff978b92 100644
--- a/test/integration/image-component/custom-resolver/test/index.test.js
+++ b/test/integration/image-component/custom-resolver/test/index.test.js
@@ -50,7 +50,10 @@ describe('Custom Resolver Tests', () => {
   describe('Client-side Custom Loader Tests', () => {
     beforeAll(async () => {
       browser = await webdriver(appPort, '/')
-      await browser.waitForElementByCss('#clientlink').click()
+      await browser
+        .elementByCss('#clientlink')
+        .click()
+        .waitForElementByCss('#client-side')
     })
     afterAll(async () => {
       browser = null