import { expect, describe, it, afterEach, vi } from 'vitest'
import fs from 'node:fs'
import path from 'node:path'
import JSZip from 'jszip'
import { remote } from '../../../src/index.js'
import logger from '@wdio/logger'
import '../../../src/node.js'
vi.mock('node:fs', () => ({
default: {
mkdirSync: vi.fn(),
readFileSync: vi.fn(),
writeFileSync: vi.fn()
}
}))
vi.mock('got')
vi.mock('JSZip', () => ({
default: {
loadAsync: vi.fn()
}
}))
vi.mock('devtools')
vi.mock('@wdio/logger', () => import(path.join(process.cwd(), '__mocks__', '@wdio/logger')))
describe('downloadFile', () => {
it('should throw if browser does not support it', async function () {
const browser = await remote({
baseUrl: 'http://webdriver.io',
capabilities: {
browserName: 'safari'
}
})
await expect(browser.downloadFile('bar.jpg', '/foo/bar')).rejects.toEqual(
new Error('The downloadFile command is not available in mockBrowser and only available when using Selenium Grid'))
})
it('should throw if path is not a string', async function () {
const browser = await remote({
baseUrl: 'http://webdriver.io',
capabilities: {
browserName: 'chrome'
}
})
await expect(browser.downloadFile(123, 456)).rejects.toEqual(
new Error('number or type of arguments don\'t agree with downloadFile command'))
})
it('should unzip the file and use downloadFile command', async () => {
vi.spyOn(JSZip, 'loadAsync').mockReturnValue(Promise.resolve(
{
files: {
'file_1': {
async: vi.fn()
}
}
})
)
const browser = await remote({
baseUrl: 'http://webdriver.io',
capabilities: {
browserName: 'chrome'
}
})
browser.download = vi.fn().mockReturnValue({
fileName: 'test',
contents: 'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=='
})
await browser.downloadFile('toDownload.jpg', __dirname)
const log = logger('test')
expect(log.info).toHaveBeenCalledWith(
expect.stringContaining('File extracted: file_1')
)
expect(fs.writeFileSync).toHaveBeenCalledTimes(2)
})
it('reject on error', async () => {
vi.spyOn(JSZip, 'loadAsync').mockRejectedValue('test'
)
const browser = await remote({
baseUrl: 'http://webdriver.io',
capabilities: {
browserName: 'chrome'
}
})
browser.download = vi.fn().mockReturnValue({
fileName: 'test',
contents: 'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=='
})
await browser.downloadFile('toDownload.jpg', __dirname)
const log = logger('test')
expect(log.error).toHaveBeenCalledWith('Error unzipping file:', 'test')
})
afterEach(() => {
const log = logger('test')
vi.mocked(log.info).mockClear()
vi.mocked(log.error).mockClear()
vi.mocked(fs.mkdirSync).mockClear()
vi.mocked(fs.readFileSync).mockClear()
vi.mocked(fs.writeFileSync).mockClear()
})
})
