Tengo un componente que muestra texto basado en el estado ('éxito' o 'cargando')

 const Button = (state) => <button>{state === 'success' ? 'Sucess!' : state === 'loading' ? 'Please wait...' : ''}</button>

y tengo que escribir prueba así:

 test('Button approved state', () => { render(<Button state="success" />) expect(screen.getByText('Sucess!')) }) test('Button pending state', () => { render(<Button state="loading" />) expect(screen.getByText('Please wait...')) })

¿Es posible para ellos en una sola prueba? Lo intenté

 test('Button approved and pending state', () => { render(<Button state="success" />) expect(screen.getByText('Sucess!')) //need to reset here? render(<Button state="loading" />) expect(screen.getByText('Please wait...')) })

la prueba falló. ¿Cómo hago algún tipo de reinicio en el medio?