isDisabled: Throw errors for unexpected element types

mainmatter · Apr 19, 2018 · 9c2cb8c · 9c2cb8c
1 parent 615fd23
commit 9c2cb8c
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/lib/assertions/is-disabled.js b/lib/assertions/is-disabled.js
@@ -2,15 +2,24 @@ export default function isDisabled(message) {
   let element = this.findTargetElement();
   if (!element) return;
 
+  if (!(
+      element instanceof HTMLInputElement ||
+      element instanceof HTMLTextAreaElement ||
+      element instanceof HTMLSelectElement ||
+      element instanceof HTMLButtonElement ||
+      element instanceof HTMLOptGroupElement ||
+      element instanceof HTMLOptionElement ||
+      element instanceof HTMLFieldSetElement
+    )) {
+    throw new TypeError(`Unexpected Element Type: ${element.toString()}`);
+  }
+
   let expected = `Element ${this.targetDescription} is disabled`;
   let actual = `Element ${this.targetDescription} is disabled`;
   let result = element.disabled;
 
   if (result === false) {
     actual = `Element ${this.targetDescription} is not disabled`;
-  } else if(result === undefined) {
-    actual = `Element ${this.targetDescription} does not support disabled`;
-    result = false;
   }
 
   if (!message) {