fix(click): add an option to pass native element and debug element

ngneat · Apr 2, 2018 · fafe43d · fafe43d
1 parent d94dbd9
commit fafe43d
Showing 1 changed file with 19 additions and 7 deletions.
diff --git a/src/lib/src/internals.ts b/src/lib/src/internals.ts
@@ -88,16 +88,28 @@ export class Spectator<C> {
 
   /**
    *
-   * @param {string} selector
+   * @param selector string | Element | DebugElement
    */
-  click(selector: string) {
-    const element = this.debugElement.query(By.css(selector));
-    if (element) {
-      element.nativeElement.click();
-      this.detectChanges();
+  click(selector: string | Element | DebugElement) {
+    let element;
+
+    if (typeof selector === 'string') {
+      const checkExists = this.debugElement.query(By.css(selector));
+      if (checkExists) {
+        element = checkExists.nativeElement;
+      } else {
+        throw new Error(`${selector} does not exists`);
+      }
     } else {
-      throw new Error(`${selector} does not exists`);
+      if (selector instanceof DebugElement) {
+        element = selector.nativeElement;
+      } else {
+        element = selector;
+      }
     }
+
+    element.click();
+    this.detectChanges();
   }
 
   /**