Fix detecting nested EE constants in RuboCop

The InjectEnterpriseEditionModule cop would not detect certain nested EE
constants such as `EE::Foo::Bar::Baz`. This could result in it not
enforcing `prepend` being placed on the last line. This commit fixes
this by just performing a string match on the line, instead of relying
on AST matching.

1 files changed, 7 insertions, 3 deletions

diff --git a/rubocop/cop/inject_enterprise_edition_module.rb b/rubocop/cop/inject_enterprise_edition_module.rb
index c8b8aca51ab..1d37b1bd12d 100644
--- a/rubocop/cop/inject_enterprise_edition_module.rb
+++ b/rubocop/cop/inject_enterprise_edition_module.rb
@@ -11,9 +11,13 @@ module RuboCop
 
       METHODS = Set.new(%i[include extend prepend]).freeze
 
-      def_node_matcher :ee_const?, <<~PATTERN
-        (const (const _ :EE) _)
-      PATTERN
+      def ee_const?(node)
+        line = node.location.expression.source_line
+
+        # We use `match?` here instead of RuboCop's AST matching, as this makes
+        # it far easier to handle nested constants such as `EE::Foo::Bar::Baz`.
+        line.match?(/(\s|\()(::)?EE::/)
+      end
 
       def on_send(node)
         return unless METHODS.include?(node.children[1])