New rule: Performance/FirstLastAfterFilter (#76)

spec/ameba/rule/performance/first_last_after_filter_spec.cr

@@ -0,0 +1,125 @@
+require "../../../spec_helper"
+
+module Ameba::Rule::Performance
+  subject = FirstLastAfterFilter.new
+
+  describe FirstLastAfterFilter do
+    it "passes if there is no potential performance improvements" do
+      source = Source.new %(
+        [1, 2, 3].select { |e| e > 1 }
+        [1, 2, 3].reverse.select { |e| e > 1 }
+        [1, 2, 3].reverse.last
+        [1, 2, 3].reverse.first
+        [1, 2, 3].reverse.first
+      )
+      subject.catch(source).should be_valid
+    end
+
+    it "reports if there is select followed by last" do
+      source = Source.new %(
+        [1, 2, 3].select { |e| e > 2 }.last
+      )
+      subject.catch(source).should_not be_valid
+    end
+
+    it "reports if there is select followed by last?" do
+      source = Source.new %(
+        [1, 2, 3].select { |e| e > 2 }.last?
+      )
+      subject.catch(source).should_not be_valid
+    end
+
+    it "reports if there is select followed by first" do
+      source = Source.new %(
+        [1, 2, 3].select { |e| e > 2 }.first
+      )
+      subject.catch(source).should_not be_valid
+    end
+
+    it "reports if there is select followed by first?" do
+      source = Source.new %(
+        [1, 2, 3].select { |e| e > 2 }.first?
+      )
+      subject.catch(source).should_not be_valid
+    end
+
+    it "does not report if there is select followed by any other call" do
+      source = Source.new %(
+        [1, 2, 3].select { |e| e > 2 }.size
+        [1, 2, 3].select { |e| e > 2 }.any?
+      )
+      subject.catch(source).should be_valid
+    end
+
+    context "properties" do
+      it "allows to configure object_call_names" do
+        source = Source.new %(
+          [1, 2, 3].select { |e| e > 2 }.first
+        )
+        rule = Rule::Performance::FirstLastAfterFilter.new
+        rule.filter_names = %w(reject)
+        rule.catch(source).should be_valid
+      end
+    end
+
+    it "reports rule, pos and message" do
+      s = Source.new %(
+        [1, 2, 3].select { |e| e > 2 }.first
+      ), "source.cr"
+      subject.catch(s).should_not be_valid
+      s.issues.size.should eq 1
+
+      issue = s.issues.first
+      issue.rule.should_not be_nil
+      issue.location.to_s.should eq "source.cr:2:19"
+      issue.end_location.to_s.should eq "source.cr:2:45"
+
+      issue.message.should eq "Use `find {...}` instead of `select {...}.first`"
+    end
+
+    it "reports a correct message for first?" do
+      s = Source.new %(
+        [1, 2, 3].select { |e| e > 2 }.first?
+      ), "source.cr"
+      subject.catch(s).should_not be_valid
+      s.issues.size.should eq 1
+
+      issue = s.issues.first
+      issue.rule.should_not be_nil
+      issue.location.to_s.should eq "source.cr:2:19"
+      issue.end_location.to_s.should eq "source.cr:2:46"
+
+      issue.message.should eq "Use `find {...}` instead of `select {...}.first?`"
+    end
+
+    it "reports rule, pos and reverse message" do
+      s = Source.new %(
+        [1, 2, 3].select { |e| e > 2 }.last
+      ), "source.cr"
+      subject.catch(s).should_not be_valid
+      s.issues.size.should eq 1
+
+      issue = s.issues.first
+      issue.rule.should_not be_nil
+      issue.location.to_s.should eq "source.cr:2:19"
+      issue.end_location.to_s.should eq "source.cr:2:44"
+
+      issue.message.should eq "Use `reverse_each.find {...}` instead of `select {...}.last`"
+    end
+
+    it "reports a correct message for last?" do
+      s = Source.new %(
+        [1, 2, 3].select { |e| e > 2 }.last?
+      ), "source.cr"
+      subject.catch(s).should_not be_valid
+      s.issues.size.should eq 1
+
+      issue = s.issues.first
+      issue.rule.should_not be_nil
+      issue.location.to_s.should eq "source.cr:2:19"
+      issue.end_location.to_s.should eq "source.cr:2:45"
+
+      issue.message.should eq "Use `reverse_each.find {...}` instead of `select {...}.last?`"
+    end
+  end
+end