From 7cff8faddec105f13d4cf7c7ebc09ae69bd142b2 Mon Sep 17 00:00:00 2001
From: Cory LaViska <cory@abeautifulsite.net>
Date: Tue, 10 Dec 2024 09:26:58 -0500
Subject: [PATCH] update tests

---
 src/components/callout/callout.test.ts | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/components/callout/callout.test.ts b/src/components/callout/callout.test.ts
index bb422fe3b..99489a357 100644
--- a/src/components/callout/callout.test.ts
+++ b/src/components/callout/callout.test.ts
@@ -15,9 +15,7 @@ describe('<wa-callout>', () => {
           await customElements.whenDefined('wa-callout');
           await callout.updateComplete;
 
-          const base = callout.shadowRoot!.querySelector<HTMLElement>('[part="base"]')!;
-
-          expect(base).to.have.class(`callout--${variant}`);
+          expect(callout).to.have.attribute('variant', variant);
 
           // @TODO: For some reason this fails only in CI. I have no clue why. I tested this scenario on the real site, and it works as expected. [Konnor]
           if (fixture.type === 'ssr-client-hydrated') {