1
0
Fork 0
mirror of https://github.com/terribleplan/next.js.git synced 2024-01-19 02:48:18 +00:00
next.js/test/unit/router.test.js

69 lines
1.9 KiB
JavaScript
Raw Normal View History

/* eslint-env jest */
2018-10-01 22:55:31 +00:00
import Router from 'next-server/dist/lib/router/router'
class PageLoader {
constructor (options = {}) {
this.options = options
this.loaded = {}
this.prefetched = {}
}
loadPage (route) {
this.loaded[route] = true
if (this.options.delay) {
return new Promise((resolve) => setTimeout(resolve, this.options.delay))
}
}
prefetch (route) {
this.prefetched[route] = true
}
}
describe('Router', () => {
const request = { clone: () => null }
describe('.prefetch()', () => {
it('should prefetch a given page', async () => {
global.__NEXT_DATA__ = {}
const pageLoader = new PageLoader()
const router = new Router('/', {}, '/', { pageLoader })
const route = '/routex'
await router.prefetch(route)
expect(pageLoader.prefetched['/routex']).toBeTruthy()
})
it('should call prefetch correctly', async () => {
global.__NEXT_DATA__ = {}
// delay loading pages for an hour
const pageLoader = new PageLoader({ delay: 1000 * 3600 })
const router = new Router('/', {}, '/', { pageLoader })
router.prefetch('route1')
router.prefetch('route2')
router.prefetch('route3')
router.prefetch('route4')
expect(Object.keys(pageLoader.prefetched).length).toBe(4)
expect(Object.keys(pageLoader.prefetched)).toEqual(['route1', 'route2', 'route3', 'route4'])
})
it('should run all the jobs', async () => {
global.__NEXT_DATA__ = {}
const pageLoader = new PageLoader()
const router = new Router('/', {}, '/', { pageLoader })
const routes = ['route1', 'route2', 'route3', 'route4']
router.doFetchRoute = () => Promise.resolve(request)
await router.prefetch(routes[0])
await router.prefetch(routes[1])
await router.prefetch(routes[2])
await router.prefetch(routes[3])
expect(Object.keys(pageLoader.prefetched)).toEqual(routes)
})
})
})