From 50d28b2a22fe8d1d320595317f06dc1ceb7f33df Mon Sep 17 00:00:00 2001
From: Joe Haddad <joe.haddad@zeit.co>
Date: Fri, 22 Jan 2021 12:34:55 -0500
Subject: [PATCH] new tests

---
 .../pages/nav/shallow-routing.js              | 17 +++++++++++
 .../client-navigation/test/index.test.js      | 28 ++++++++++++++++---
 2 files changed, 41 insertions(+), 4 deletions(-)

diff --git a/test/integration/client-navigation/pages/nav/shallow-routing.js b/test/integration/client-navigation/pages/nav/shallow-routing.js
index 7d41faa1df8bf54..728467f9afde42b 100644
--- a/test/integration/client-navigation/pages/nav/shallow-routing.js
+++ b/test/integration/client-navigation/pages/nav/shallow-routing.js
@@ -27,6 +27,17 @@ export default class extends Component {
     Router.push(href, href, { shallow: true })
   }
 
+  increaseNonShallow() {
+    const counter = this.getCurrentCounter()
+    const href = `/nav/shallow-routing?counter=${counter + 1}`
+    Router.push(href, href, {})
+  }
+
+  gotoNavShallow() {
+    const href = `/nav`
+    Router.push(href, href, { shallow: true })
+  }
+
   render() {
     return (
       <div className="shallow-routing">
@@ -44,6 +55,12 @@ export default class extends Component {
         <button id="increase" onClick={() => this.increase()}>
           Increase
         </button>
+        <button id="increase2" onClick={() => this.increaseNonShallow()}>
+          Increase Non-Shallow
+        </button>
+        <button id="invalidShallow" onClick={() => this.gotoNavShallow()}>
+          Invalid Shallow Nav
+        </button>
       </div>
     )
   }
diff --git a/test/integration/client-navigation/test/index.test.js b/test/integration/client-navigation/test/index.test.js
index e3b951a3927e7b9..0d87cdeda148056 100644
--- a/test/integration/client-navigation/test/index.test.js
+++ b/test/integration/client-navigation/test/index.test.js
@@ -862,12 +862,32 @@ describe('Client Navigation', () => {
       )
       const scrollPosition = await browser.eval('window.pageYOffset')
 
-      const newScrollPosition = await browser
-        .elementByCss('#increase')
-        .click()
-        .eval('window.pageYOffset')
+      expect(scrollPosition).toBeGreaterThan(3000)
+
+      await browser.elementByCss('#increase').click()
+      await waitFor(500)
+      const newScrollPosition = await browser.eval('window.pageYOffset')
 
       expect(newScrollPosition).toBe(scrollPosition)
+
+      await browser.elementByCss('#increase2').click()
+      await waitFor(500)
+      const newScrollPosition2 = await browser.eval('window.pageYOffset')
+
+      expect(newScrollPosition2).toBe(0)
+
+      await browser.eval(() =>
+        document.querySelector('#invalidShallow').scrollIntoView()
+      )
+      const scrollPositionDown = await browser.eval('window.pageYOffset')
+
+      expect(scrollPositionDown).toBeGreaterThan(3000)
+
+      await browser.elementByCss('#invalidShallow').click()
+      await waitFor(500)
+      const newScrollPosition3 = await browser.eval('window.pageYOffset')
+
+      expect(newScrollPosition3).toBe(0)
     })
   })