test(cli): refactor tests for async render utilities (#23252)

This commit is contained in:
Tommaso Sciortino
2026-03-20 20:08:29 +00:00
committed by GitHub
parent 86a3a913b5
commit 6c78eb7a39
198 changed files with 3592 additions and 4802 deletions

View File

@@ -29,8 +29,8 @@ describe('ProQuotaDialog', () => {
});
describe('for flash model failures', () => {
it('should render "Keep trying" and "Stop" options', () => {
const { unmount } = render(
it('should render "Keep trying" and "Stop" options', async () => {
const { unmount } = await render(
<ProQuotaDialog
failedModel={DEFAULT_GEMINI_FLASH_MODEL}
fallbackModel={DEFAULT_GEMINI_FLASH_MODEL}
@@ -63,8 +63,8 @@ describe('ProQuotaDialog', () => {
describe('for non-flash model failures', () => {
describe('when it is a terminal quota error', () => {
it('should render switch, upgrade, and stop options for LOGIN_WITH_GOOGLE', () => {
const { unmount } = render(
it('should render switch, upgrade, and stop options for LOGIN_WITH_GOOGLE', async () => {
const { unmount } = await render(
<ProQuotaDialog
failedModel="gemini-2.5-pro"
fallbackModel="gemini-2.5-flash"
@@ -101,8 +101,8 @@ describe('ProQuotaDialog', () => {
unmount();
});
it('should NOT render upgrade option for USE_GEMINI', () => {
const { unmount } = render(
it('should NOT render upgrade option for USE_GEMINI', async () => {
const { unmount } = await render(
<ProQuotaDialog
failedModel="gemini-2.5-pro"
fallbackModel="gemini-2.5-flash"
@@ -134,8 +134,8 @@ describe('ProQuotaDialog', () => {
unmount();
});
it('should render "Keep trying" and "Stop" options when failed model and fallback model are the same', () => {
const { unmount } = render(
it('should render "Keep trying" and "Stop" options when failed model and fallback model are the same', async () => {
const { unmount } = await render(
<ProQuotaDialog
failedModel={PREVIEW_GEMINI_MODEL}
fallbackModel={PREVIEW_GEMINI_MODEL}
@@ -165,8 +165,8 @@ describe('ProQuotaDialog', () => {
unmount();
});
it('should render switch, upgrade, and stop options for LOGIN_WITH_GOOGLE (free tier)', () => {
const { unmount } = render(
it('should render switch, upgrade, and stop options for LOGIN_WITH_GOOGLE (free tier)', async () => {
const { unmount } = await render(
<ProQuotaDialog
failedModel="gemini-2.5-pro"
fallbackModel="gemini-2.5-flash"
@@ -203,8 +203,8 @@ describe('ProQuotaDialog', () => {
unmount();
});
it('should NOT render upgrade option for LOGIN_WITH_GOOGLE if tier is Ultra', () => {
const { unmount } = render(
it('should NOT render upgrade option for LOGIN_WITH_GOOGLE if tier is Ultra', async () => {
const { unmount } = await render(
<ProQuotaDialog
failedModel="gemini-2.5-pro"
fallbackModel="gemini-2.5-flash"
@@ -239,8 +239,8 @@ describe('ProQuotaDialog', () => {
});
describe('when it is a capacity error', () => {
it('should render keep trying, switch, and stop options', () => {
const { unmount } = render(
it('should render keep trying, switch, and stop options', async () => {
const { unmount } = await render(
<ProQuotaDialog
failedModel="gemini-2.5-pro"
fallbackModel="gemini-2.5-flash"
@@ -274,8 +274,8 @@ describe('ProQuotaDialog', () => {
});
describe('when it is a model not found error', () => {
it('should render switch, upgrade, and stop options for LOGIN_WITH_GOOGLE', () => {
const { unmount } = render(
it('should render switch, upgrade, and stop options for LOGIN_WITH_GOOGLE', async () => {
const { unmount } = await render(
<ProQuotaDialog
failedModel="gemini-3-pro-preview"
fallbackModel="gemini-2.5-pro"
@@ -312,8 +312,8 @@ describe('ProQuotaDialog', () => {
unmount();
});
it('should NOT render upgrade option for USE_GEMINI', () => {
const { unmount } = render(
it('should NOT render upgrade option for USE_GEMINI', async () => {
const { unmount } = await render(
<ProQuotaDialog
failedModel="gemini-3-pro-preview"
fallbackModel="gemini-2.5-pro"
@@ -348,8 +348,8 @@ describe('ProQuotaDialog', () => {
});
describe('onChoice handling', () => {
it('should call onChoice with the selected value', () => {
const { unmount } = render(
it('should call onChoice with the selected value', async () => {
const { unmount } = await render(
<ProQuotaDialog
failedModel="gemini-2.5-pro"
fallbackModel="gemini-2.5-flash"