fix test types

saucelabs · Sep 25, 2024 · e048195 · e048195
1 parent 445c9c4
commit e048195
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/visual-js/visual/src/utils.spec.ts b/visual-js/visual/src/utils.spec.ts
@@ -3,11 +3,13 @@ import { getFullPageConfig, parseRegionsForAPI } from './utils';
 import { RegionIn } from './graphql/__generated__/graphql';
 import { FullPageScreenshotOptions } from './types';
 
-const configDelay: FullPageScreenshotOptions = {
+type MockElement = { elementId: string };
+
+const configDelay: FullPageScreenshotOptions<MockElement> = {
   delayAfterScrollMs: 1500,
 };
 
-const configDelayBig: FullPageScreenshotOptions = {
+const configDelayBig: FullPageScreenshotOptions<MockElement> = {
   delayAfterScrollMs: 5000,
 };
 
@@ -94,15 +96,19 @@ describe('utils', () => {
         const local = {
           scrollElement: Promise.resolve({ elementId: elementId }),
         };
-        expect(await getFullPageConfig(main, local)).toEqual({
+        expect(
+          await getFullPageConfig(main, local, (el) => el.elementId),
+        ).toEqual({
           scrollElement: elementId,
         });
       });
       test('with scrollElement when scrollElement is an object', async () => {
         const elementId = 'elementId';
         const main = { scrollElement: { elementId: elementId } };
         const local = {};
-        expect(await getFullPageConfig(main, local)).toEqual({
+        expect(
+          await getFullPageConfig(main, local, (el) => el.elementId),
+        ).toEqual({
           scrollElement: elementId,
         });
       });