mirror of
https://github.com/mudler/LocalAI.git
synced 2026-10-10 15:52:29 -04:00
mudler-agentandEttore Di Giacinto
690f3994b0
A key can now be paused until the owner resumes it, or until a given
time. A paused key is rejected by validation before last_used is
updated, and a pause time that has passed lifts the pause by itself.
Existing keys stay active.
PATCH /api/auth/api-keys/:id takes {"disabled": bool, "paused_until":
RFC 3339 string or null}. Only the key owner can change it, and a
paused_until in the past is rejected. The key list returns the pause
fields. The Account page gets a Pause and Resume button for each key
and a Paused badge that shows the resume time.
Assisted-by: Claude Code:claude-sonnet-5-5
Co-authored-by: Ettore Di Giacinto <mudler@localai.io>
83 lines
3.5 KiB
JavaScript
83 lines
3.5 KiB
JavaScript
import { test, expect } from '@playwright/test'
|
|
|
|
// Account > API Keys: pause and resume a key without deleting it.
|
|
|
|
const soon = new Date(Date.now() + 3 * 3600 * 1000).toISOString()
|
|
|
|
function json(body) {
|
|
return { contentType: 'application/json', body: JSON.stringify(body) }
|
|
}
|
|
|
|
test.describe('Account API keys pause', () => {
|
|
let keys
|
|
let patches
|
|
|
|
test.beforeEach(async ({ page }) => {
|
|
patches = []
|
|
keys = [
|
|
{ id: 'k1', name: 'active-key', keyPrefix: 'lai-aaaaaaaa', role: 'user', createdAt: new Date().toISOString(), disabled: false },
|
|
{ id: 'k2', name: 'paused-key', keyPrefix: 'lai-bbbbbbbb', role: 'user', createdAt: new Date().toISOString(), disabled: true },
|
|
{ id: 'k3', name: 'timed-key', keyPrefix: 'lai-cccccccc', role: 'user', createdAt: new Date().toISOString(), disabled: false, pausedUntil: soon },
|
|
]
|
|
|
|
await page.route('**/api/auth/status', (route) =>
|
|
route.fulfill(json({
|
|
authEnabled: true,
|
|
providers: ['local'],
|
|
hasUsers: true,
|
|
user: { id: 'u1', email: 'u@example.com', name: 'U', role: 'user' },
|
|
}))
|
|
)
|
|
await page.route('**/api/auth/api-keys', (route) => route.fulfill(json({ keys })))
|
|
await page.route('**/api/auth/api-keys/*', async (route) => {
|
|
const req = route.request()
|
|
if (req.method() === 'PATCH') {
|
|
const id = req.url().split('/').pop()
|
|
const body = req.postDataJSON()
|
|
patches.push({ id, body })
|
|
const key = keys.find((k) => k.id === id)
|
|
key.disabled = body.disabled
|
|
key.pausedUntil = body.paused_until || undefined
|
|
return route.fulfill(json({ message: 'API key updated' }))
|
|
}
|
|
return route.continue()
|
|
})
|
|
|
|
await page.goto('/app/account')
|
|
await page.getByRole('button', { name: /API Keys/ }).click()
|
|
})
|
|
|
|
test('shows paused badges and resume buttons', async ({ page }) => {
|
|
await expect(page.locator('.apikey-item')).toHaveCount(3)
|
|
await expect(page.locator('.apikey-item').nth(0).locator('.apikey-paused-badge')).toHaveCount(0)
|
|
await expect(page.locator('.apikey-item').nth(1).locator('.apikey-paused-badge')).toHaveText('Paused')
|
|
await expect(page.locator('.apikey-item').nth(2).locator('.apikey-paused-badge')).toContainText('Paused until')
|
|
})
|
|
|
|
test('pauses a key indefinitely', async ({ page }) => {
|
|
const item = page.locator('.apikey-item').nth(0)
|
|
await item.getByRole('button', { name: 'Pause' }).click()
|
|
await item.getByRole('button', { name: 'Pause key' }).click()
|
|
await expect(item.locator('.apikey-paused-badge')).toHaveText('Paused')
|
|
expect(patches).toEqual([{ id: 'k1', body: { disabled: true, paused_until: null } }])
|
|
})
|
|
|
|
test('pauses a key until a chosen time', async ({ page }) => {
|
|
const item = page.locator('.apikey-item').nth(0)
|
|
await item.getByRole('button', { name: 'Pause' }).click()
|
|
await item.getByLabel('Until', { exact: true }).first().check()
|
|
await item.locator('input[type="datetime-local"]').fill('2099-01-02T03:04')
|
|
await item.getByRole('button', { name: 'Pause key' }).click()
|
|
await expect(item.locator('.apikey-paused-badge')).toContainText('Paused until')
|
|
expect(patches[0].body.disabled).toBe(false)
|
|
expect(patches[0].body.paused_until).toMatch(/^2099-01-0[12]T/)
|
|
})
|
|
|
|
test('resumes a paused key', async ({ page }) => {
|
|
const item = page.locator('.apikey-item').nth(1)
|
|
await item.getByRole('button', { name: 'Resume' }).click()
|
|
await expect(item.locator('.apikey-paused-badge')).toHaveCount(0)
|
|
expect(patches).toEqual([{ id: 'k2', body: { disabled: false, paused_until: null } }])
|
|
})
|
|
})
|