lib/pelusa/lint/case_statements.rb in pelusa-0.2.2 vs lib/pelusa/lint/case_statements.rb in pelusa-0.2.3

@@ -4,11 +4,10 @@
       def initialize
         @violations = Set.new
       end
 
       def check(klass)
-        initialize
         iterate_lines!(klass)
 
         return SuccessfulAnalysis.new(name) if @violations.empty?
 
         FailedAnalysis.new(name, @violations) do |violations|
@@ -21,15 +20,14 @@
       def name
         "Doesn't use case statements"
       end
 
       def iterate_lines!(klass)
-        iterator = Iterator.new do |node|
+        ClassAnalyzer.walk(klass) do |node|
           if node.is_a?(Rubinius::AST::Case)
             @violations << node.line
           end
         end
-        Array(klass).each(&iterator)
       end
     end
   end
 end