tests: add test for wait

patrickhulce · patrickhulce · commit 2d525b339144 · 2019-05-23T12:09:21.000-05:00
diff --git a/test/fixtures/late-page.html b/test/fixtures/late-page.html
@@ -0,0 +1,13 @@
+<!DOCTYPE html>
+<html>
+  <body>
+    <span>Loading...</span>
+    <script>
+      setTimeout(() => {
+        const loaded = document.createElement('span')
+        loaded.textContent = 'Loaded!'
+        document.body.appendChild(loaded)
+      }, 5000)
+    </script>
+  </body>
+</html>
diff --git a/test/index.test.ts b/test/index.test.ts
@@ -1,6 +1,6 @@
 import * as path from 'path'
 import * as puppeteer from 'puppeteer'
-import {getDocument, queries, getQueriesForElement} from '../lib'
+import {getDocument, queries, getQueriesForElement, wait} from '../lib'
 
 describe('lib/index.ts', () => {
   let browser: puppeteer.Browser
@@ -24,6 +24,16 @@ describe('lib/index.ts', () => {
     expect(await queries.getNodeText(element)).toEqual('Hello h1')
   })
 
+  it('should load the defered page', async () => {
+    await page.goto(`file://${path.join(__dirname, 'fixtures/late-page.html')}`)
+  })
+
+  it('should use `wait` properly', async () => {
+    const {getByText} = getQueriesForElement(await getDocument(page))
+    await wait(() => getByText('Loaded!'))
+    expect(await getByText('Loaded!')).toBeTruthy()
+  }, 7000)
+
   afterAll(async () => {
     await browser.close()
   })
