first commit

tests/e2e/features/model-limits/model-limits.spec.ts

@@ -0,0 +1,81 @@
+import { expect, test } from '../../core/fixtures/base.fixture'
+import { createModelLimitData } from './model-limits.data'
+
+const createdLimits: { modelName: string; provider: string }[] = []
+
+test.describe('Model Limits', () => {
+  test.beforeEach(async ({ modelLimitsPage }) => {
+    await modelLimitsPage.goto()
+  })
+
+  test.afterEach(async ({ modelLimitsPage }) => {
+    await modelLimitsPage.closeSheet()
+    for (const { modelName, provider } of [...createdLimits]) {
+      try {
+        const exists = await modelLimitsPage.modelLimitExists(modelName, provider)
+        if (exists) {
+          await modelLimitsPage.deleteModelLimit(modelName, provider)
+        }
+      } catch (e) {
+        console.error(`[CLEANUP] Failed to delete model limit ${modelName}:`, e)
+      }
+    }
+    createdLimits.length = 0
+  })
+
+  test('should display create button or empty state', async ({ modelLimitsPage }) => {
+    const createVisible = await modelLimitsPage.createBtn.isVisible().catch(() => false)
+    expect(createVisible).toBe(true)
+  })
+
+  test('should create a model limit with budget and rate limit', async ({ modelLimitsPage }) => {
+    const limitData = createModelLimitData({
+      provider: 'openai',
+      budget: { maxLimit: 5, resetDuration: '1M' },
+      rateLimit: { tokenMaxLimit: 500, requestMaxLimit: 20 },
+    })
+
+    const modelName = await modelLimitsPage.createModelLimit(limitData)
+    createdLimits.push({ modelName, provider: limitData.provider })
+
+    const exists = await modelLimitsPage.modelLimitExists(modelName, limitData.provider)
+    expect(exists).toBe(true)
+  })
+
+  test('should edit a model limit budget and rate limit', async ({ modelLimitsPage }) => {
+    const limitData = createModelLimitData({ provider: 'openai' })
+
+    const modelName = await modelLimitsPage.createModelLimit(limitData)
+    createdLimits.push({ modelName, provider: limitData.provider })
+
+    await modelLimitsPage.editModelLimit(modelName, limitData.provider, {
+      budget: { maxLimit: 20 },
+      rateLimit: { tokenMaxLimit: 2000, requestMaxLimit: 100 },
+    })
+
+    const exists = await modelLimitsPage.modelLimitExists(modelName, limitData.provider)
+    expect(exists).toBe(true)
+  })
+
+  test('should delete a model limit', async ({ modelLimitsPage }) => {
+    const limitData = createModelLimitData({
+      provider: 'openai',
+      budget: { maxLimit: 5 },
+    })
+
+    const modelName = await modelLimitsPage.createModelLimit(limitData)
+    createdLimits.push({ modelName, provider: limitData.provider })
+
+    let exists = await modelLimitsPage.modelLimitExists(modelName, limitData.provider)
+    expect(exists).toBe(true)
+
+    await modelLimitsPage.deleteModelLimit(modelName, limitData.provider)
+    const idx = createdLimits.findIndex(
+      (limit) => limit.modelName === modelName && limit.provider === limitData.provider
+    )
+    if (idx >= 0) createdLimits.splice(idx, 1)
+
+    exists = await modelLimitsPage.modelLimitExists(modelName, limitData.provider)
+    expect(exists).toBe(false)
+  })
+})