introduce counter test

Author: khiga8

lib/erblint-github/linters/custom_helpers.rb

@@ -27,7 +27,7 @@ def rule_disabled?(processed_source)
       def counter_correct?(processed_source)
         comment_node = nil
         expected_count = 0
-        rule_name = self.class.name.match(/:?:?(\w+)\Z/)[1]
+        rule_name = simple_class_name
         offenses_count = @offenses.length
 
         processed_source.parser.ast.descendants(:erb).each do |node|

test/custom_helpers_test.rb

@@ -11,7 +11,7 @@ class FakeLinter < ERBLint::Linter
 
     MESSAGE = "Please fix your code."
   end
-
+  
   def linter_class
     CustomHelpersTest::FakeLinter
   end
@@ -38,8 +38,36 @@ def test_rule_disabled_adds_offense_if_disable_comment_is_present_but_no_offense
     assert_empty @linter.offenses
 
     extended_linter.rule_disabled?(processed_source)
+    assert_equal "Unused erblint:disable comment for CustomHelpersTest::FakeLinter", @linter.offenses.first.message
+  end
 
-    assert_equal @linter.offenses.length, 1
+  def test_counter_correct_does_not_add_offense_if_counter_matches_offense_count
+    @file = <<~HTML
+      <%# erblint:count CustomHelpersTest::FakeLinter 1 %>
+    HTML
+    @linter.offenses = ["fake offense"]
+    
+    extended_linter.counter_correct?(processed_source)
+    assert_empty @linter.offenses
+  end
+
+  def test_counter_correct_add_offense_if_counter_comment_is_unused
+    @file = <<~HTML
+      <%# erblint:count CustomHelpersTest::FakeLinter 1 %>
+    HTML
+
+    extended_linter.counter_correct?(processed_source)
+    assert_equal "Unused erblint:count comment for CustomHelpersTest::FakeLinter", @linter.offenses.first.message
+  end
+
+  def test_counter_correct_add_offense_if_counter_comment_count_is_incorrect
+    @file = <<~HTML
+      <%# erblint:count CustomHelpersTest::FakeLinter 2 %>
+    HTML
+    @linter.offenses = ["fake offense"]
+
+    extended_linter.counter_correct?(processed_source)
+    assert_equal "Incorrect erblint:counter number for CustomHelpersTest::FakeLinter. Expected: 2, actual: 1.", @linter.offenses.first.message
   end
 
   def test_generate_offense_with_message_defined_in_linter_class