get all controllers

faraazahmad · Jul 12, 2023 · 2ec8738 · 2ec8738
1 parent 8db87d5
commit 2ec8738
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/trains/version.rb b/lib/trains/version.rb
@@ -1,3 +1,3 @@
 module Trains
-  VERSION = '0.0.15'.freeze
+  VERSION = '0.0.16'.freeze
 end
diff --git a/lib/trains/visitor/controller.rb b/lib/trains/visitor/controller.rb
@@ -53,7 +53,7 @@ def parse_body(body)
       end
 
       def controller?(parent_class)
-        %w[ActionController::Base ApplicationController].include? parent_class
+        parent_class.include? 'Controller'
       end
 
       def parse_method(node)