43 lines
1.8 KiB
TypeScript
43 lines
1.8 KiB
TypeScript
import { expect, test } from '@playwright/test'
|
|
import { authenticate, mockHermesApi, TEST_ACCESS_KEY } from './fixtures'
|
|
|
|
test('fetches custom provider models through the backend proxy', async ({ page }) => {
|
|
await authenticate(page, TEST_ACCESS_KEY)
|
|
const api = await mockHermesApi(page)
|
|
|
|
const thirdPartyRequests: string[] = []
|
|
page.on('request', (request) => {
|
|
const url = request.url()
|
|
if (url.startsWith('https://provider.example.test')) {
|
|
thirdPartyRequests.push(url)
|
|
}
|
|
})
|
|
|
|
await page.goto('/#/hermes/models')
|
|
|
|
await page.getByRole('button', { name: 'Add Provider' }).click()
|
|
await page.getByRole('button', { name: 'Custom' }).click()
|
|
await page.getByPlaceholder('e.g. https://api.example.com/v1').fill('https://provider.example.test/v1')
|
|
await page.getByPlaceholder('sk-...').fill('test-provider-key')
|
|
await page.getByRole('button', { name: 'Fetch', exact: true }).click()
|
|
|
|
await expect(page.getByText('Found 2 models')).toBeVisible()
|
|
// Scope to the dropdown options: Arco renders both the select's current value
|
|
// and the option list, so a bare text match resolves to two nodes. The option
|
|
// list is collapsed by this point, so assert presence rather than visibility.
|
|
await expect(
|
|
page.locator('.arco-select-option-content', { hasText: 'proxy-model-a' }),
|
|
).toHaveCount(1)
|
|
|
|
const proxyRequest = api.requests.find((request) => request.pathname === '/api/hermes/provider-models')
|
|
expect(proxyRequest).toBeTruthy()
|
|
expect(proxyRequest?.method).toBe('POST')
|
|
expect(proxyRequest?.headers.authorization).toBe(`Bearer ${TEST_ACCESS_KEY}`)
|
|
expect(JSON.parse(proxyRequest?.postData || '{}')).toMatchObject({
|
|
base_url: 'https://provider.example.test/v1',
|
|
api_key: 'test-provider-key',
|
|
})
|
|
expect(thirdPartyRequests).toEqual([])
|
|
expect(api.unexpectedRequests).toEqual([])
|
|
})
|