import path from 'node:path'
import { expect, describe, beforeEach, beforeAll, afterEach, it, vi, type MockInstance } from 'vitest'
import { remote } from '../../../src/index.js'
vi.mock('fetch')
vi.mock('@wdio/logger', () => import(path.join(process.cwd(), '__mocks__', '@wdio/logger')))
describe('setCookies', () => {
const cookie1 = { name: 'cookie1', value: 'dummy-value-1' }
const cookie2 = { name: 'cookie2', value: 'dummy-value-2' }
const cookie3 = { name: 'cookie2', value: 'dummy-value-3' }
let browser: WebdriverIO.Browser
describe('classic', () => {
beforeAll(async () => {
browser = await remote({
baseUrl: 'http://foobar.com',
capabilities: {
browserName: 'foobar'
}
})
})
it('should output the expected format', async () => {
await browser.setCookies([cookie1])
expect(vi.mocked(fetch).mock.calls).toHaveLength(2!)
expect(vi.mocked(fetch).mock.calls[1][1]!.method).toBe('POST')
expect(vi.mocked(fetch).mock.calls[1][0]!.pathname)
.toBe('/session/foobar-123/cookie')
expect(vi.mocked(fetch).mock.calls[1][1]!.body)
.toEqual(JSON.stringify({ 'cookie': cookie1 }))
})
it('should support passing an object', async () => {
await browser.setCookies(cookie1)
expect(vi.mocked(fetch).mock.calls).toHaveLength(1!)
expect(vi.mocked(fetch).mock.calls[0][1]!.method).toBe('POST')
expect(vi.mocked(fetch).mock.calls[0][0]!.pathname)
.toBe('/session/foobar-123/cookie')
expect(vi.mocked(fetch).mock.calls[0][1]!.body)
.toEqual(JSON.stringify({ 'cookie': cookie1 }))
})
it('can be called multiple times', async () => {
await browser.setCookies(cookie1)
await browser.setCookies(cookie2)
expect(vi.mocked(fetch).mock.calls).toHaveLength(2!)
expect(vi.mocked(fetch).mock.calls[0][1]!.method).toBe('POST')
expect(vi.mocked(fetch).mock.calls[0][0]!.pathname)
.toBe('/session/foobar-123/cookie')
expect(vi.mocked(fetch).mock.calls[0][1]!.body)
.toEqual(JSON.stringify({ 'cookie': cookie1 }))
expect(vi.mocked(fetch).mock.calls[1][1]!.method).toBe('POST')
expect(vi.mocked(fetch).mock.calls[1][0]!.pathname)
.toBe('/session/foobar-123/cookie')
expect(vi.mocked(fetch).mock.calls[1][1]!.body)
.toEqual(JSON.stringify({ 'cookie': cookie2 }))
})
it('should work with multiple objects passed', async () => {
const cookies = [cookie1, cookie2, cookie3]
await browser.setCookies(cookies)
cookies.forEach((cookie, i) => {
expect(vi.mocked(fetch).mock.calls[i][1]!.method).toBe('POST')
expect(vi.mocked(fetch).mock.calls[i][0]!.pathname)
.toBe('/session/foobar-123/cookie')
expect(vi.mocked(fetch).mock.calls[i][1]!.body)
.toEqual(JSON.stringify({ 'cookie': cookie }))
})
})
it('should throw error if invalid arguments are passed', async () => {
await expect(browser.setCookies([2]))
.rejects
.toEqual(new Error('Invalid input (see https://webdriver.io/docs/api/browser/setCookies for documentation)'))
})
afterEach(() => {
vi.mocked(fetch).mockClear()
})
})
describe('bidi', () => {
let storageSetCookie: MockInstance
beforeAll(async () => {
browser = await remote({
baseUrl: 'http://foobar.com',
capabilities: {
browserName: 'bidi'
}
})
storageSetCookie = vi.spyOn(browser, 'storageSetCookie')
storageSetCookie.mockImplementation((() => {}) as any)
})
beforeEach(() => {
storageSetCookie.mockClear()
})
it('should support passing an array', async () => {
await browser.setCookies([cookie1])
expect(storageSetCookie).toBeCalledTimes(1)
expect(storageSetCookie).toBeCalledWith({
cookie: {
domain: 'webdriver.io',
name: 'cookie1',
value: {
type: 'string',
value: cookie1.value
}
}
})
})
it('should support passing an object', async () => {
await browser.setCookies(cookie1)
expect(storageSetCookie).toBeCalledTimes(1)
expect(storageSetCookie).toBeCalledWith({
cookie: {
domain: 'webdriver.io',
name: 'cookie1',
value: {
type: 'string',
value: cookie1.value
}
}
})
})
it('can be called multiple times', async () => {
await browser.setCookies([cookie1, cookie2, cookie3])
expect(storageSetCookie).toBeCalledTimes(3)
})
})
})
