Remove quota disabling, it didn't have much of an effect.

This commit is contained in:
Sam Rose 2025-03-04 17:42:10 +00:00
parent 84542dd808
commit e38d4c1e48
No known key found for this signature in database
4 changed files with 31 additions and 37 deletions

@ -1 +1 @@
Subproject commit bec35b4c27ec10fe31dc07f9d42d8bda426b8535
Subproject commit b28dbd549284cf450be7f25ad85aadf614d08f0b

View File

@ -177,28 +177,26 @@ if (descriptions.length) {
}
async function expectRowUsage(expected: number, f: () => Promise<void>) {
return await quotas.withEnabled(async () => {
const before = await getRowUsage()
await f()
const after = await getRowUsage()
const usage = after - before
const before = await getRowUsage()
await f()
const after = await getRowUsage()
const usage = after - before
// Because our quota tracking is not perfect, we allow a 10% margin of
// error. This is to account for the fact that parallel writes can
// result in some quota updates getting lost. We don't have any need
// to solve this right now, so we just allow for some error.
if (expected === 0) {
expect(usage).toEqual(0)
return
}
if (usage < 0) {
expect(usage).toBeGreaterThan(expected * 1.1)
expect(usage).toBeLessThan(expected * 0.9)
} else {
expect(usage).toBeGreaterThan(expected * 0.9)
expect(usage).toBeLessThan(expected * 1.1)
}
})
// Because our quota tracking is not perfect, we allow a 10% margin of
// error. This is to account for the fact that parallel writes can
// result in some quota updates getting lost. We don't have any need
// to solve this right now, so we just allow for some error.
if (expected === 0) {
expect(usage).toEqual(0)
return
}
if (usage < 0) {
expect(usage).toBeGreaterThan(expected * 1.1)
expect(usage).toBeLessThan(expected * 0.9)
} else {
expect(usage).toBeGreaterThan(expected * 0.9)
expect(usage).toBeLessThan(expected * 1.1)
}
}
const defaultRowFields = isInternal

View File

@ -2830,14 +2830,12 @@ if (descriptions.length) {
expected: number,
f: () => Promise<T>
): Promise<T> {
return await quotas.withEnabled(async () => {
const before = await getRowUsage()
const result = await f()
const after = await getRowUsage()
const usage = after - before
expect(usage).toBe(expected)
return result
})
const before = await getRowUsage()
const result = await f()
const after = await getRowUsage()
const usage = after - before
expect(usage).toBe(expected)
return result
}
it("should be able to delete a row", async () => {

View File

@ -44,13 +44,11 @@ describe("test the openai action", () => {
}
const expectAIUsage = async <T>(expected: number, f: () => Promise<T>) => {
return await quotas.withEnabled(async () => {
const before = await getAIUsage()
const result = await f()
const after = await getAIUsage()
expect(after - before).toEqual(expected)
return result
})
const before = await getAIUsage()
const result = await f()
const after = await getAIUsage()
expect(after - before).toEqual(expected)
return result
}
it("should be able to receive a response from ChatGPT given a prompt", async () => {