refactor(testing): improve flexibility of TuiNativeInputPO (#7574)

taiga-family · May 30, 2024 · 582625f · 582625f
1 parent 06fa5eb
commit 582625f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/projects/legacy/components/input-range/test/input-range.component.spec.ts b/projects/legacy/components/input-range/test/input-range.component.spec.ts
@@ -440,13 +440,17 @@ describe('InputRange', () => {
 
         inputPOLeft = new TuiNativeInputPO(
             fixture,
-            testContext.nativeInputAutoId,
-            leftInputWrapper,
+            pageObject.getByAutomationId(
+                testContext.nativeInputAutoId,
+                leftInputWrapper,
+            )!,
         );
         inputPORight = new TuiNativeInputPO(
             fixture,
-            testContext.nativeInputAutoId,
-            rightInputWrapper,
+            pageObject.getByAutomationId(
+                testContext.nativeInputAutoId,
+                rightInputWrapper,
+            )!,
         );
     }
 });
diff --git a/projects/testing/utils/native-input.page-object.ts b/projects/testing/utils/native-input.page-object.ts
@@ -10,17 +10,15 @@ export class TuiNativeInputPO {
 
     constructor(
         private readonly fixture: ComponentFixture<unknown>,
-        private readonly automationId: string,
-        private readonly hostDebugElement?: DebugElement,
+        private readonly automationIdOrElement: DebugElement | string,
     ) {
         this.pageObject = new TuiPageObject(fixture);
     }
 
     public get nativeElement(): HTMLInputElement | HTMLTextAreaElement | null {
-        return (
-            this.pageObject.getByAutomationId(this.automationId, this.hostDebugElement)
-                ?.nativeElement ?? null
-        );
+        return typeof this.automationIdOrElement === 'string'
+            ? this.pageObject.getByAutomationId(this.automationIdOrElement)?.nativeElement
+            : this.automationIdOrElement.nativeElement;
     }
 
     public get value(): string {