test(ts/components/diversionPage): add helpers for route interativity (…

…#2663)
mbta · Jun 17, 2024 · c8a22ca · c8a22ca
1 parent 5e270be
commit c8a22ca
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 6 deletions.
diff --git a/assets/tests/components/detours/diversionPage.test.tsx b/assets/tests/components/detours/diversionPage.test.tsx
@@ -1733,14 +1733,12 @@ describe("DiversionPage", () => {
       )
 
       // Non-Interactive route shape
-      expect(
-        container.querySelectorAll(".c-detour_map--original-route-shape-core")
-      ).toHaveLength(1)
+      expect(originalRouteShape.not.interactive.getAll(container)).toHaveLength(
+        1
+      )
 
       // Interactive route shape
-      expect(
-        container.querySelectorAll(".c-detour_map--original-route-shape")
-      ).toHaveLength(0)
+      expect(originalRouteShape.interactive.getAll(container)).toHaveLength(0)
     })
   })
 })
diff --git a/assets/tests/testHelpers/selectors/components/detours/diversionPage.tsx b/assets/tests/testHelpers/selectors/components/detours/diversionPage.tsx
@@ -4,6 +4,17 @@ import { byRole } from "testing-library-selector"
 export const finishDetourButton = byRole("button", { name: "Finish Detour" })
 
 export const originalRouteShape = {
+  interactive: {
+    getAll: (container: HTMLElement) =>
+      container.querySelectorAll(".c-detour_map--original-route-shape"),
+  },
+  not: {
+    interactive: {
+      getAll: (container: HTMLElement) =>
+        container.querySelectorAll(".c-detour_map--original-route-shape-core"),
+    },
+  },
+
   get(container: HTMLElement): Element {
     const maybeShape = container.querySelector(
       ".c-detour_map--original-route-shape"