fix: add argument node checks

Closes #29
jest-community · Dec 5, 2017 · 775e75d · 775e75d
1 parent e7e9e62
commit 775e75d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/rules/__tests__/prefer_to_be_null.test.js b/rules/__tests__/prefer_to_be_null.test.js
@@ -6,7 +6,7 @@ const rules = require('../../').rules;
 const ruleTester = new RuleTester();
 
 ruleTester.run('prefer_to_be_null', rules['prefer-to-be-null'], {
-  valid: ['expect(null).toBeNull();'],
+  valid: ['expect(null).toBeNull();', 'expect(null).toEqual();'],
 
   invalid: [
     {

diff --git a/rules/util.js b/rules/util.js
@@ -26,22 +26,26 @@ const expectToBeCase = (node, arg) =>
   !(expectNotCase(node) || expectResolveCase(node) || expectRejectCase(node)) &&
   expectCase(node) &&
   methodName(node) === 'toBe' &&
+  argument(node) &&
   argument(node).value === arg;
 
 const expectNotToBeCase = (node, arg) =>
   expectNotCase(node) &&
   methodName2(node) === 'toBe' &&
+  argument2(node) &&
   argument2(node).value === arg;
 
 const expectToEqualCase = (node, arg) =>
   !(expectNotCase(node) || expectResolveCase(node) || expectRejectCase(node)) &&
   expectCase(node) &&
   methodName(node) === 'toEqual' &&
+  argument(node) &&
   argument(node).value === arg;
 
 const expectNotToEqualCase = (node, arg) =>
   expectNotCase(node) &&
   methodName2(node) === 'toEqual' &&
+  argument2(node) &&
   argument2(node).value === arg;
 
 const expectToBeUndefinedCase = node =>