32 lines
935 B
TypeScript
32 lines
935 B
TypeScript
import { expect, test } from "@playwright/test";
|
|
import { canAccessHiddenAiModels } from "@/lib/ai/model-visibility";
|
|
|
|
test.describe("hidden AI model visibility", () => {
|
|
test("uses the current database role when a JWT still contains an old role", () => {
|
|
expect(
|
|
canAccessHiddenAiModels({
|
|
sessionUserType: "regular",
|
|
databaseUserRole: "admin",
|
|
isHighVip: false,
|
|
})
|
|
).toBe(true);
|
|
});
|
|
|
|
test("allows Style Pro administrators and high VIP users", () => {
|
|
expect(
|
|
canAccessHiddenAiModels({ databaseUserRole: "agent_admin" })
|
|
).toBe(true);
|
|
expect(canAccessHiddenAiModels({ isHighVip: true })).toBe(true);
|
|
});
|
|
|
|
test("keeps hidden provider variants out of regular user selectors", () => {
|
|
expect(
|
|
canAccessHiddenAiModels({
|
|
sessionUserType: "regular",
|
|
databaseUserRole: "regular",
|
|
isHighVip: false,
|
|
})
|
|
).toBe(false);
|
|
});
|
|
});
|