From 9bf1ccf1b8031276170603de0ff947e6fa1337d3 Mon Sep 17 00:00:00 2001
From: yoonie-jang <yoonie.jang@mail.utoronto.ca>
Date: Wed, 6 Nov 2024 13:59:26 -0500
Subject: [PATCH] Update tests to reflect changes

---
 webstepper/src/__tests__/App.spec.tsx | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

diff --git a/webstepper/src/__tests__/App.spec.tsx b/webstepper/src/__tests__/App.spec.tsx
index 7820608..054f920 100644
--- a/webstepper/src/__tests__/App.spec.tsx
+++ b/webstepper/src/__tests__/App.spec.tsx
@@ -82,18 +82,12 @@ describe("App", () => {
             "code > span:nth-child(2)"
         );
 
-        expect(firstLineElement).toHaveStyle({
-            backgroundColor: "rgb(255, 255, 0)",
-        });
+        expect(firstLineElement).toHaveClass("code-box__line--highlighted");
 
         const nextButton = screen.getByText("Next");
         fireEvent.click(nextButton);
 
-        expect(firstLineElement).not.toHaveStyle({
-            backgroundColor: "rgb(255, 255, 0)",
-        });
-        expect(secondLineElement).toHaveStyle({
-            backgroundColor: "rgb(255, 255, 0)",
-        });
+        expect(firstLineElement).not.toHaveClass("code-box__line--highlighted");
+        expect(secondLineElement).toHaveClass("code-box__line--highlighted");
     });
 });