import path from 'node:path'
import { ELEMENT_KEY } from 'webdriver'
import { expect, describe, it, vi } from 'vitest'

import { remote } from '../../../src/index.js'

vi.mock('fetch')
vi.mock('@wdio/logger', () => import(path.join(process.cwd(), '__mocks__', '@wdio/logger')))

vi.mock('../../../src/commands/element/$$', () => ({
    __esModule: true,
    default: vi.fn().mockImplementation(() => { })
}))

/**
 * Todo(Christian): make unit test work
 */
describe('shadow$$', () => {
    it('should find elements within a shadow root', async () => {
        const browser = await remote({
            baseUrl: 'http://foobar.com',
            capabilities: {
                browserName: 'foobar'
            }
        })

        const elem = await browser.$('#foo')
        const elems = await elem.shadow$$('#subfoo')

        expect(vi.mocked(fetch).mock.calls[2][0]!.pathname)
            .toBe('/session/foobar-123/element/some-elem-123/shadow')
        expect(vi.mocked(fetch).mock.calls[3][0]!.pathname)
            .toBe('/session/foobar-123/shadow/some-shadow-elem-123/elements')

        expect(elems[0].elementId).toBe('some-shadow-sub-elem-321')
        expect(elems[0][ELEMENT_KEY]).toBe('some-shadow-sub-elem-321')
        expect(elems[0].ELEMENT).toBe(undefined)
        expect(elems[0].selector).toBe('#subfoo')
        expect(elems[0].index).toBe(0)
        expect(elems[1].elementId).toBe('some-sub-shadow-elem-456')
        expect(elems[1][ELEMENT_KEY]).toBe('some-sub-shadow-elem-456')
        expect(elems[1].ELEMENT).toBe(undefined)
        expect(elems[1].selector).toBe('#subfoo')
        expect(elems[1].index).toBe(1)
        expect(elems[2].elementId).toBe('some-sub-shadow-elem-789')
        expect(elems[2][ELEMENT_KEY]).toBe('some-sub-shadow-elem-789')
        expect(elems[2].ELEMENT).toBe(undefined)
        expect(elems[2].selector).toBe('#subfoo')
        expect(elems[2].index).toBe(2)
    })

    it('fails back to JS for browser that dont have shadow support in WebDriver', async () => {
        const browser = await remote({
            baseUrl: 'http://foobar.com',
            capabilities: {
                browserName: 'foobar'
            }
        })
        const errorResponse = { error: 'ups' }
        const el = await browser.$('#foo')
        // @ts-expect-error mock feature
        fetch.setMockResponse([errorResponse, errorResponse, errorResponse, errorResponse])
        const queryResult = [{ elem: 123 }]
        // @ts-expect-error
        queryResult.getElements = vi.fn().mockReturnValue(queryResult)
        const mock: any = {
            $$: vi.fn().mockReturnValue([{ elem: 123 }]),
            options: {},
            selector: 'foo',
        }
        mock.parent = { $: vi.fn().mockReturnValue({ getElement: () => ({}) }) }
        mock.waitForExist = vi.fn().mockResolvedValue(mock)
        const elem = await el.shadow$$.call(mock, '#shadowfoo').getElements()
        expect(elem).toEqual([{ elem: 123 }])

        expect(vi.mocked(fetch).mock.calls[1][0]!.pathname)
            .toBe('/session/foobar-123/element')
    })
})