Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
244 changes: 167 additions & 77 deletions packages/gatsby-core-utils/src/__tests__/fetch-remote-file.js
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ import { setupServer } from "msw/node"
import { Writable } from "stream"
import got from "got"
import fs from "fs-extra"
import { slash } from "gatsby-core-utils/path"
import { fetchRemoteFile } from "../fetch-remote-file"
import * as storage from "../utils/get-storage"

Expand Down Expand Up @@ -132,6 +133,7 @@ const server = setupServer(
ctx.body(content)
)
}),

rest.get(`http://external.com/dog`, async (req, res, ctx) => {
const { content, contentLength } = await getFileContent(
path.join(__dirname, `./fixtures/dog-thumbnail.jpg`),
Expand Down Expand Up @@ -175,6 +177,19 @@ const server = setupServer(
ctx.body(content)
)
}),
rest.get(`http://external.com/dog-*.jpg`, async (req, res, ctx) => {
const { content, contentLength } = await getFileContent(
path.join(__dirname, `./fixtures/dog-thumbnail.jpg`),
req
)

return res(
ctx.set(`Content-Type`, `image/jpg`),
ctx.set(`Content-Length`, contentLength),
ctx.status(200),
ctx.body(content)
)
}),
rest.get(`http://external.com/404.jpg`, async (req, res, ctx) => {
const content = `Page not found`

Expand Down Expand Up @@ -456,96 +471,171 @@ Fetch details:
`)
})

it(`should not re-download file if cache is set`, async () => {
const filePath = await fetchRemoteFile({
url: `http://external.com/dog.jpg`,
cache,
cacheKey: `1`,
})
const cachedFilePath = await fetchRemoteFile({
url: `http://external.com/dog.jpg`,
cache,
cacheKey: `1`,
})
let cacheVersion = 0
describe.each([false, true])(`with excludeDigest %s`, excludeDigest => {
function getExternalUrl(cacheVersion) {
return `http://external.com/dog-${cacheVersion}.jpg?v=${cacheVersion}`
}

expect(filePath).toBe(cachedFilePath)
expect(gotStream).toBeCalledTimes(1)
expect(fs.pathExists).toBeCalledTimes(1)
expect(fs.copy).not.toBeCalled()
})
it(`should not re-download file if cache is set`, async () => {
const filePath = await fetchRemoteFile({
url: getExternalUrl(++cacheVersion),
cache,
cacheKey: `${cacheVersion}`,
excludeDigest,
})
const cachedFilePath = await fetchRemoteFile({
url: getExternalUrl(cacheVersion),
cache,
cacheKey: `${cacheVersion}`,
excludeDigest,
})

it(`should not re-download and use same path if ouputDir is not inside public folder`, async () => {
const filePath = await fetchRemoteFile({
url: `http://external.com/dog.jpg`,
directory: cache.directory,
cacheKey: `2`,
})
const cachedFilePath = await fetchRemoteFile({
url: `http://external.com/dog.jpg`,
directory: path.join(cache.directory, `diff`),
cacheKey: `2`,
expect(filePath).toBe(cachedFilePath)
expect(gotStream).toBeCalledTimes(1)
expect(fs.pathExists).toBeCalledTimes(1)
expect(fs.copy).not.toBeCalled()
})

expect(filePath).toBe(cachedFilePath)
expect(gotStream).toBeCalledTimes(1)
expect(fs.pathExists).toBeCalledTimes(1)
expect(fs.copy).not.toBeCalled()
})
it(`should not re-download and use same path if ouputDir is not inside public folder`, async () => {
const filePath = await fetchRemoteFile({
url: getExternalUrl(++cacheVersion),
directory: cache.directory,
cacheKey: `${cacheVersion}`,
excludeDigest,
})
const cachedFilePath = await fetchRemoteFile({
url: getExternalUrl(cacheVersion),
directory: path.join(cache.directory, `diff`),
cacheKey: `${cacheVersion}`,
excludeDigest,
})

it(`should not re-download but copy file to public folder`, async () => {
const currentGlobal = global.__GATSBY
global.__GATSBY = {
root: cache.directory,
}
await fs.ensureDir(path.join(cache.directory, `public`))
const filePath = await fetchRemoteFile({
url: `http://external.com/dog.jpg`,
directory: cache.directory,
cacheKey: `3`,
expect(filePath).toBe(cachedFilePath)
expect(gotStream).toBeCalledTimes(1)
expect(fs.pathExists).toBeCalledTimes(1)
expect(fs.copy).not.toBeCalled()
})
const cachedFilePath = await fetchRemoteFile({
url: `http://external.com/dog.jpg`,
directory: path.join(cache.directory, `public`),
cacheKey: `3`,

it(`should not re-download but copy file to public folder`, async () => {
const currentGlobal = global.__GATSBY
global.__GATSBY = {
root: cache.directory,
}
await fs.ensureDir(path.join(cache.directory, `public`))
const filePath = await fetchRemoteFile({
url: getExternalUrl(++cacheVersion),
directory: cache.directory,
cacheKey: `${cacheVersion}`,
excludeDigest,
})
const cachedFilePath = await fetchRemoteFile({
url: getExternalUrl(cacheVersion),
directory: path.join(cache.directory, `public`),
cacheKey: `${cacheVersion}`,
excludeDigest,
})

expect(filePath).not.toBe(cachedFilePath)
expect(cachedFilePath).toStartWith(path.join(cache.directory, `public`))
expect(gotStream).toBeCalledTimes(1)
expect(fs.pathExists).toBeCalledTimes(1)
expect(fs.copy).toBeCalledTimes(1)
expect(await fs.pathExists(cachedFilePath)).toBe(true)
global.__GATSBY = currentGlobal
})

expect(filePath).not.toBe(cachedFilePath)
expect(cachedFilePath).toStartWith(path.join(cache.directory, `public`))
expect(gotStream).toBeCalledTimes(1)
expect(fs.pathExists).toBeCalledTimes(1)
expect(fs.copy).toBeCalledTimes(1)
expect(await fs.pathExists(cachedFilePath)).toBe(true)
global.__GATSBY = currentGlobal
})
it(`should not re-download but copy file to public folder (with slashes)`, async () => {
const currentGlobal = global.__GATSBY
global.__GATSBY = {
root: cache.directory,
}
await fs.ensureDir(path.join(cache.directory, `public`))
const filePath = await fetchRemoteFile({
url: getExternalUrl(++cacheVersion),
directory: slash(cache.directory),
cacheKey: `${cacheVersion}`,
excludeDigest,
})
const cachedFilePath = await fetchRemoteFile({
url: getExternalUrl(cacheVersion),
directory: slash(path.join(cache.directory, `public`)),
cacheKey: `${cacheVersion}`,
excludeDigest,
})

it(`should not re-download but copy file to public folder when the same url is requested`, async () => {
const currentGlobal = global.__GATSBY
global.__GATSBY = {
root: cache.directory,
}
await fs.ensureDir(path.join(cache.directory, `public`))
const filePathPromise = fetchRemoteFile({
url: `http://external.com/dog.jpg?v=4`,
directory: cache.directory,
cacheKey: `4`,
expect(filePath).not.toBe(cachedFilePath)
expect(cachedFilePath).toStartWith(path.join(cache.directory, `public`))
expect(gotStream).toBeCalledTimes(1)
expect(fs.pathExists).toBeCalledTimes(1)
expect(fs.copy).toBeCalledTimes(1)
expect(await fs.pathExists(cachedFilePath)).toBe(true)
global.__GATSBY = currentGlobal
})
const cachedFilePathPromise = fetchRemoteFile({
url: `http://external.com/dog.jpg?v=4`,
directory: path.join(cache.directory, `public`),
cacheKey: `4`,

it(`should not re-download but copy file to public folder when the same url is requested`, async () => {
const currentGlobal = global.__GATSBY
global.__GATSBY = {
root: cache.directory,
}
await fs.ensureDir(path.join(cache.directory, `public`))
const filePathPromise = fetchRemoteFile({
url: getExternalUrl(++cacheVersion),
directory: cache.directory,
cacheKey: `${cacheVersion}`,
excludeDigest,
})
const cachedFilePathPromise = fetchRemoteFile({
url: getExternalUrl(cacheVersion),
directory: path.join(cache.directory, `public`),
cacheKey: `${cacheVersion}`,
excludeDigest,
})

const [filePath, cachedFilePath] = await Promise.all([
filePathPromise,
cachedFilePathPromise,
])

expect(filePath).not.toBe(cachedFilePath)
expect(cachedFilePath).toStartWith(path.join(cache.directory, `public`))
expect(gotStream).toBeCalledTimes(1)
expect(fs.pathExists).toBeCalledTimes(0)
expect(fs.copy).toBeCalledTimes(1)
global.__GATSBY = currentGlobal
})

const [filePath, cachedFilePath] = await Promise.all([
filePathPromise,
cachedFilePathPromise,
])
it(`should not re-download but copy file to public folder when the same url is requested (with slashes)`, async () => {
const currentGlobal = global.__GATSBY
global.__GATSBY = {
root: cache.directory,
}
await fs.ensureDir(path.join(cache.directory, `public`))
const filePathPromise = fetchRemoteFile({
url: getExternalUrl(++cacheVersion),
directory: slash(cache.directory),
cacheKey: `${cacheVersion}`,
excludeDigest,
})
const cachedFilePathPromise = fetchRemoteFile({
url: getExternalUrl(cacheVersion),
directory: slash(path.join(cache.directory, `public`)),
cacheKey: `${cacheVersion}`,
excludeDigest,
})

expect(filePath).not.toBe(cachedFilePath)
expect(cachedFilePath).toStartWith(path.join(cache.directory, `public`))
expect(gotStream).toBeCalledTimes(1)
expect(fs.pathExists).toBeCalledTimes(0)
expect(fs.copy).toBeCalledTimes(1)
global.__GATSBY = currentGlobal
const [filePath, cachedFilePath] = await Promise.all([
filePathPromise,
cachedFilePathPromise,
])

expect(filePath).not.toBe(cachedFilePath)
expect(cachedFilePath).toStartWith(path.join(cache.directory, `public`))
expect(gotStream).toBeCalledTimes(1)
expect(fs.pathExists).toBeCalledTimes(0)
expect(fs.copy).toBeCalledTimes(1)
global.__GATSBY = currentGlobal
})
})

describe(`retries the download`, () => {
Expand Down
2 changes: 1 addition & 1 deletion packages/gatsby-core-utils/src/fetch-remote-file.ts
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,7 @@ async function fetchFile({
try {
const digest = createContentDigest(url)
const finalDirectory = excludeDigest
? fileDirectory
? path.resolve(fileDirectory)
: path.join(fileDirectory, digest)

if (!name) {
Expand Down
Loading