fix failing tests

gatsbyjs · Apr 17, 2019 · dfbc209 · dfbc209
1 parent 5964ff9
commit dfbc209
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/packages/gatsby/src/redux/machines/__tests__/page-component.js b/packages/gatsby/src/redux/machines/__tests__/page-component.js
@@ -1,8 +1,8 @@
 const { interpret } = require(`xstate`)
 const machine = require(`../page-component`)
 
-jest.mock(`../../../query/query-watcher`)
-const { runQueryForPage } = require(`../../../query/query-watcher`)
+jest.mock(`../../../query`)
+const { enqueueExtractedQueryId, runQueuedQueries } = require(`../../../query`)
 
 const getService = (args = {}) =>
   interpret(
@@ -19,7 +19,8 @@ const sleep = (delay = 50) => new Promise(resolve => setTimeout(resolve, delay))
 
 describe(`bootstrap`, () => {
   beforeEach(() => {
-    runQueryForPage.mockClear()
+    enqueueExtractedQueryId.mockClear()
+    runQueuedQueries.mockClear()
   })
 
   it(`handles not running queries during bootstrap`, () => {
@@ -56,7 +57,7 @@ describe(`bootstrap`, () => {
     service.send({ type: `NEW_PAGE_CREATED`, path: `/test` })
     // there is setTimeout in action handler for `NEW_PAGE_CREATED`
     await sleep()
-    expect(runQueryForPage).not.toBeCalled()
+    expect(runQueuedQueries).not.toBeCalled()
   })
 
   it(`will queue query when page if new page is created after bootstrap`, async () => {
@@ -65,6 +66,6 @@ describe(`bootstrap`, () => {
     service.send({ type: `NEW_PAGE_CREATED`, path })
     // there is setTimeout in action handler for `NEW_PAGE_CREATED`
     await sleep()
-    expect(runQueryForPage).toBeCalledWith(path)
+    expect(runQueuedQueries).toBeCalledWith(path)
   })
 })