Skip to content

Commit b6742d2

Browse files
committed
Clear cooldowns on menu re-enable
1 parent f08847f commit b6742d2

2 files changed

Lines changed: 10 additions & 0 deletions

File tree

index.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3539,6 +3539,8 @@ while (attempted.size < Math.max(1, accountCount)) {
35393539
if (shouldEnable) {
35403540
delete target.enabled;
35413541
delete target.disabledReason;
3542+
delete target.coolingDownUntil;
3543+
delete target.cooldownReason;
35423544
} else {
35433545
target.enabled = false;
35443546
target.disabledReason = "user";

test/index.test.ts

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3970,6 +3970,8 @@ describe("OpenAIOAuthPlugin persistAccountPool", () => {
39703970
accountId: "workspace-managed",
39713971
email: "managed@example.com",
39723972
refreshToken: "refresh-managed",
3973+
coolingDownUntil: 30_000,
3974+
cooldownReason: "network-error",
39733975
addedAt: 10,
39743976
lastUsed: 10,
39753977
},
@@ -3998,17 +4000,23 @@ describe("OpenAIOAuthPlugin persistAccountPool", () => {
39984000
accountId: "workspace-managed",
39994001
enabled: false,
40004002
disabledReason: "user",
4003+
coolingDownUntil: 30_000,
4004+
cooldownReason: "network-error",
40014005
});
40024006
expect(saveAccountsMock.mock.calls[1]?.[0].accounts[0]).toMatchObject({
40034007
accountId: "workspace-managed",
40044008
});
40054009
expect(saveAccountsMock.mock.calls[1]?.[0].accounts[0]).not.toHaveProperty("enabled");
40064010
expect(saveAccountsMock.mock.calls[1]?.[0].accounts[0]).not.toHaveProperty("disabledReason");
4011+
expect(saveAccountsMock.mock.calls[1]?.[0].accounts[0]).not.toHaveProperty("coolingDownUntil");
4012+
expect(saveAccountsMock.mock.calls[1]?.[0].accounts[0]).not.toHaveProperty("cooldownReason");
40074013
expect(mockStorage.accounts[0]).toMatchObject({
40084014
accountId: "workspace-managed",
40094015
});
40104016
expect(mockStorage.accounts[0]).not.toHaveProperty("enabled");
40114017
expect(mockStorage.accounts[0]).not.toHaveProperty("disabledReason");
4018+
expect(mockStorage.accounts[0]).not.toHaveProperty("coolingDownUntil");
4019+
expect(mockStorage.accounts[0]).not.toHaveProperty("cooldownReason");
40124020
});
40134021

40144022
it("keeps auth-failure disables blocked in the auth manage menu until a fresh login", async () => {

0 commit comments

Comments
 (0)