From e42c9e3de40b7fa64f572b61e0d9f67b57c913f2 Mon Sep 17 00:00:00 2001
From: Paul Armstrong <paularmstrong@users.noreply.github.com>
Date: Thu, 9 Aug 2018 15:49:43 -0700
Subject: [PATCH] fix(rule): no-large-snapshots to allow maxSize of 0 (#132)

---
 .../no-large-snapshots.test.js.snap           | 22 ++++++++++++++++++
 rules/__tests__/no-large-snapshots.test.js    | 23 +++++++++++++++++++
 rules/no-large-snapshots.js                   |  8 +++++--
 3 files changed, 51 insertions(+), 2 deletions(-)

diff --git a/rules/__tests__/__snapshots__/no-large-snapshots.test.js.snap b/rules/__tests__/__snapshots__/no-large-snapshots.test.js.snap
index 16df709e7..88e33d6b6 100644
--- a/rules/__tests__/__snapshots__/no-large-snapshots.test.js.snap
+++ b/rules/__tests__/__snapshots__/no-large-snapshots.test.js.snap
@@ -1,5 +1,27 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
+exports[`no-large-snapshots ExpressionStatement function should report if maxSize is zero 1`] = `
+Array [
+  Object {
+    "data": Object {
+      "lineCount": 1,
+      "lineLimit": 0,
+    },
+    "message": "Expected to not encounter a Jest snapshot but was found with {{ lineCount }} lines long",
+    "node": Object {
+      "loc": Object {
+        "end": Object {
+          "line": 2,
+        },
+        "start": Object {
+          "line": 1,
+        },
+      },
+    },
+  },
+]
+`;
+
 exports[`no-large-snapshots ExpressionStatement function should report if node has more lines of code than number given in sizeThreshold option 1`] = `
 Array [
   Object {
diff --git a/rules/__tests__/no-large-snapshots.test.js b/rules/__tests__/no-large-snapshots.test.js
index 8c2942a34..51ba4fa32 100644
--- a/rules/__tests__/no-large-snapshots.test.js
+++ b/rules/__tests__/no-large-snapshots.test.js
@@ -74,6 +74,29 @@ describe('no-large-snapshots', () => {
       expect(mockReport.mock.calls[0]).toMatchSnapshot();
     });
 
+    it('should report if maxSize is zero', () => {
+      const mockReport = jest.fn();
+      const mockContext = {
+        getFilename: () => 'mock-component.jsx.snap',
+        options: [{ maxSize: 0 }],
+        report: mockReport,
+      };
+      const mockNode = {
+        loc: {
+          start: {
+            line: 1,
+          },
+          end: {
+            line: 2,
+          },
+        },
+      };
+      noLargeSnapshots(mockContext).ExpressionStatement(mockNode);
+
+      expect(mockReport).toHaveBeenCalledTimes(1);
+      expect(mockReport.mock.calls[0]).toMatchSnapshot();
+    });
+
     it('should not report if node has fewer lines of code than limit', () => {
       const mockReport = jest.fn();
       const mockContext = {
diff --git a/rules/no-large-snapshots.js b/rules/no-large-snapshots.js
index 61f49bb6b..9956fbf1b 100644
--- a/rules/no-large-snapshots.js
+++ b/rules/no-large-snapshots.js
@@ -11,7 +11,9 @@ module.exports = {
   create(context) {
     if (context.getFilename().endsWith('.snap')) {
       const lineLimit =
-        (context.options[0] && context.options[0].maxSize) || 50;
+        context.options[0] && Number.isFinite(context.options[0].maxSize)
+          ? context.options[0].maxSize
+          : 50;
 
       return {
         ExpressionStatement(node) {
@@ -22,7 +24,9 @@ module.exports = {
           if (lineCount > lineLimit) {
             context.report({
               message:
-                'Expected Jest snapshot to be smaller than {{ lineLimit }} lines but was {{ lineCount }} lines long',
+                lineLimit === 0
+                  ? 'Expected to not encounter a Jest snapshot but was found with {{ lineCount }} lines long'
+                  : 'Expected Jest snapshot to be smaller than {{ lineLimit }} lines but was {{ lineCount }} lines long',
               data: { lineLimit, lineCount },
               node,
             });