updated test case

Author: rohithandique

src/App.test.tsx

@@ -1,11 +1,28 @@
-import { render, screen } from '@testing-library/react';
+import { render, screen, fireEvent } from '@testing-library/react';
 import { describe, it, expect } from 'vitest';
 import App from './App';
 
 describe('App', () => {
   it('renders headline', () => {
     render(<App />);
-    const headline = screen.getByText(/Vite \+ React/i);
+    const headline = screen.getByText(/count is/i);
     expect(headline).toBeInTheDocument();
   });
+
+  it('increments count on button click', () => {
+    render(<App />);
+    const button = screen.getByText(/count is/i);
+    fireEvent.click(button);
+    const updatedButton = screen.getByText(/count is 1/i);
+    expect(updatedButton).toBeInTheDocument();
+  });
+
+  it('increments count on multiple button clicks', () => {
+    render(<App />);
+    const button = screen.getByText(/count is/i);
+    fireEvent.click(button);
+    fireEvent.click(button);
+    const updatedButton = screen.getByText(/count is 2/i);
+    expect(updatedButton).toBeInTheDocument();
+  });
 });

src/App.tsx

@@ -1,33 +1,16 @@
 import { useState } from 'react'
-import reactLogo from './assets/react.svg'
-import viteLogo from './assets/vite.svg'
 import './App.css'
 
 function App() {
   const [count, setCount] = useState(0)
 
   return (
     <>
-      <div>
-        <a href="https://vite.dev" target="_blank">
-          <img src={viteLogo} className="logo" alt="Vite logo" />
-        </a>
-        <a href="https://react.dev" target="_blank">
-          <img src={reactLogo} className="logo react" alt="React logo" />
-        </a>
-      </div>
-      <h1>Vite + React</h1>
       <div className="card">
         <button onClick={() => setCount((count) => count + 1)}>
           count is {count}
         </button>
-        <p>
-          Edit <code>src/App.tsx</code> and save to test HMR
-        </p>
       </div>
-      <p className="read-the-docs">
-        Click on the Vite and React logos to learn more
-      </p>
     </>
   )
 }