Merge branch 'ee-fix-rubocop-ee-module-injection-detection' into 'master'

EE: Fix detecting nested EE constants in RuboCop

See merge request gitlab-org/gitlab-ee!9189

app/controllers/admin/application_controller.rb

@@ -4,7 +4,7 @@
 #
 # Automatically sets the layout and ensures an administrator is logged in
 class Admin::ApplicationController < ApplicationController
-  prepend EE::Admin::ApplicationController
+  prepend EE::Admin::ApplicationController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   before_action :authenticate_admin!
   layout 'admin'

app/controllers/admin/dashboard_controller.rb

 # frozen_string_literal: true
 
 class Admin::DashboardController < Admin::ApplicationController
-  prepend ::EE::Admin::DashboardController
+  prepend ::EE::Admin::DashboardController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   include CountHelper
 

app/controllers/admin/projects_controller.rb

 # frozen_string_literal: true
 
 class Admin::ProjectsController < Admin::ApplicationController
-  prepend EE::Admin::ProjectsController
+  prepend EE::Admin::ProjectsController # rubocop: disable Cop/InjectEnterpriseEditionModule
   include MembersPresentation
 
   before_action :project, only: [:show, :transfer, :repository_check]

app/controllers/boards/lists_controller.rb

@@ -2,7 +2,7 @@
 
 module Boards
   class ListsController < Boards::ApplicationController
-    prepend ::EE::Boards::ListsController
+    prepend ::EE::Boards::ListsController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     include BoardsResponses
 

app/controllers/import/github_controller.rb

 # frozen_string_literal: true
 
 class Import::GithubController < Import::BaseController
-  prepend ::EE::Import::GithubController
+  prepend ::EE::Import::GithubController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   before_action :verify_import_enabled
   before_action :provider_auth, only: [:status, :jobs, :create]

app/controllers/projects/autocomplete_sources_controller.rb

 # frozen_string_literal: true
 
 class Projects::AutocompleteSourcesController < Projects::ApplicationController
-  prepend EE::Projects::AutocompleteSourcesController
+  prepend EE::Projects::AutocompleteSourcesController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   def members
     render json: ::Projects::ParticipantsService.new(@project, current_user).execute(target)

app/controllers/projects/environments_controller.rb

@@ -15,7 +15,7 @@ class Projects::EnvironmentsController < Projects::ApplicationController
     push_frontend_feature_flag(:area_chart, project)
   end
 
-  prepend ::EE::Projects::EnvironmentsController
+  prepend ::EE::Projects::EnvironmentsController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   def index
     @environments = project.environments

app/controllers/projects/git_http_client_controller.rb

@@ -5,7 +5,7 @@
 class Projects::GitHttpClientController < Projects::ApplicationController
   include ActionController::HttpAuthentication::Basic
   include KerberosSpnegoHelper
-  prepend ::EE::Projects::GitHttpClientController
+  prepend ::EE::Projects::GitHttpClientController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   attr_reader :authentication_result, :redirected_path
 

app/controllers/projects/git_http_controller.rb

@@ -2,7 +2,7 @@
 
 class Projects::GitHttpController < Projects::GitHttpClientController
   include WorkhorseRequest
-  prepend ::EE::Projects::GitHttpController
+  prepend ::EE::Projects::GitHttpController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   before_action :access_check
 

app/controllers/projects/issues_controller.rb

@@ -9,7 +9,7 @@ class Projects::IssuesController < Projects::ApplicationController
   include IssuesCalendar
   include SpammableActions
 
-  prepend ::EE::Projects::IssuesController
+  prepend ::EE::Projects::IssuesController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   def self.issue_except_actions
     %i[index calendar new create bulk_update import_csv]

app/controllers/projects/merge_requests/application_controller.rb

 # frozen_string_literal: true
 
 class Projects::MergeRequests::ApplicationController < Projects::ApplicationController
-  prepend ::EE::Projects::MergeRequests::ApplicationController
+  prepend ::EE::Projects::MergeRequests::ApplicationController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   before_action :check_merge_requests_available!
   before_action :merge_request

app/controllers/projects/merge_requests/diffs_controller.rb

 # frozen_string_literal: true
 
 class Projects::MergeRequests::DiffsController < Projects::MergeRequests::ApplicationController
-  prepend ::EE::Projects::MergeRequests::DiffsController
+  prepend ::EE::Projects::MergeRequests::DiffsController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   include DiffForPath
   include DiffHelper

app/controllers/projects/merge_requests_controller.rb

@@ -8,7 +8,7 @@ class Projects::MergeRequestsController < Projects::MergeRequests::ApplicationCo
   include ToggleAwardEmoji
   include IssuableCollections
 
-  prepend ::EE::Projects::MergeRequestsController
+  prepend ::EE::Projects::MergeRequestsController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   skip_before_action :merge_request, only: [:index, :bulk_update]
   before_action :whitelist_query_limiting, only: [:assign_related_issues, :update]

app/controllers/projects/pipelines_controller.rb

 # frozen_string_literal: true
 
 class Projects::PipelinesController < Projects::ApplicationController
-  prepend ::EE::Projects::PipelinesController
+  prepend ::EE::Projects::PipelinesController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   before_action :whitelist_query_limiting, only: [:create, :retry]
   before_action :pipeline, except: [:index, :new, :create, :charts]

app/controllers/projects/settings/ci_cd_controller.rb

@@ -6,7 +6,7 @@ module Projects
       before_action :authorize_admin_pipeline!
       before_action :define_variables
 
-      prepend ::EE::Projects::Settings::CiCdController
+      prepend ::EE::Projects::Settings::CiCdController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def show
       end

app/controllers/projects/settings/repository_controller.rb

@@ -6,7 +6,7 @@ module Projects
       before_action :authorize_admin_project!
       before_action :remote_mirror, only: [:show]
 
-      prepend ::EE::Projects::Settings::RepositoryController
+      prepend ::EE::Projects::Settings::RepositoryController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def show
         render_show

app/controllers/projects/variables_controller.rb

 # frozen_string_literal: true
 
 class Projects::VariablesController < Projects::ApplicationController
-  prepend ::EE::Projects::VariablesController
+  prepend ::EE::Projects::VariablesController # rubocop: disable Cop/InjectEnterpriseEditionModule
 
   before_action :authorize_admin_build!
 

app/models/concerns/protected_ref_access.rb

@@ -51,7 +51,7 @@ module ProtectedRefAccess
   end
 end
 
-ProtectedRefAccess.include(EE::ProtectedRefAccess::Scopes)
+ProtectedRefAccess.include(EE::ProtectedRefAccess::Scopes) # rubocop: disable Cop/InjectEnterpriseEditionModule
 ProtectedRefAccess.prepend(EE::ProtectedRefAccess) # rubocop: disable Cop/InjectEnterpriseEditionModule
 
 # When using `prepend` (or `include` for that matter), the `ClassMethods`

app/models/project_services/chat_message/merge_message.rb

@@ -2,7 +2,7 @@
 
 module ChatMessage
   class MergeMessage < BaseMessage
-    prepend ::EE::ChatMessage::MergeMessage
+    prepend ::EE::ChatMessage::MergeMessage # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     attr_reader :merge_request_iid
     attr_reader :source_branch

app/services/applications/create_service.rb

@@ -2,7 +2,7 @@
 
 module Applications
   class CreateService
-    prepend ::EE::Applications::CreateService
+    prepend ::EE::Applications::CreateService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     # rubocop: disable CodeReuse/ActiveRecord
     def initialize(current_user, params)

app/services/boards/issues/create_service.rb

@@ -3,7 +3,7 @@
 module Boards
   module Issues
     class CreateService < Boards::BaseService
-      prepend ::EE::Boards::Issues::CreateService
+      prepend ::EE::Boards::Issues::CreateService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       attr_accessor :project
 

app/services/boards/lists/list_service.rb

@@ -3,7 +3,7 @@
 module Boards
   module Lists
     class ListService < Boards::BaseService
-      prepend ::EE::Boards::Lists::ListService
+      prepend ::EE::Boards::Lists::ListService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def execute(board)
         board.lists.create(list_type: :backlog) unless board.lists.backlog.exists?

app/services/emails/create_service.rb

@@ -2,7 +2,7 @@
 
 module Emails
   class CreateService < ::Emails::BaseService
-    prepend ::EE::Emails::CreateService
+    prepend ::EE::Emails::CreateService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def execute(extra_params = {})
       skip_confirmation = @params.delete(:skip_confirmation)

app/services/emails/destroy_service.rb

@@ -2,7 +2,7 @@
 
 module Emails
   class DestroyService < ::Emails::BaseService
-    prepend ::EE::Emails::DestroyService
+    prepend ::EE::Emails::DestroyService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def execute(email)
       email.destroy && update_secondary_emails!

app/services/groups/create_service.rb

@@ -2,7 +2,7 @@
 
 module Groups
   class CreateService < Groups::BaseService
-    prepend ::EE::Groups::CreateService
+    prepend ::EE::Groups::CreateService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def initialize(user, params = {})
       @current_user, @params = user, params.dup

app/services/groups/destroy_service.rb

@@ -2,7 +2,7 @@
 
 module Groups
   class DestroyService < Groups::BaseService
-    prepend ::EE::Groups::DestroyService
+    prepend ::EE::Groups::DestroyService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     DestroyError = Class.new(StandardError)
 

app/services/groups/update_service.rb

@@ -3,7 +3,7 @@
 module Groups
   class UpdateService < Groups::BaseService
     include UpdateVisibilityLevel
-    prepend ::EE::Groups::UpdateService
+    prepend ::EE::Groups::UpdateService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def execute
       reject_parent_id!

app/services/issues/build_service.rb

@@ -3,7 +3,7 @@
 module Issues
   class BuildService < Issues::BaseService
     include ResolveDiscussions
-    prepend ::EE::Issues::BuildService
+    prepend ::EE::Issues::BuildService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def execute
       filter_resolve_discussion_params

app/services/issues/move_service.rb

@@ -2,7 +2,7 @@
 
 module Issues
   class MoveService < Issuable::Clone::BaseService
-    prepend ::EE::Issues::MoveService
+    prepend ::EE::Issues::MoveService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     MoveError = Class.new(StandardError)
 

app/services/merge_requests/update_service.rb

@@ -2,7 +2,7 @@
 
 module MergeRequests
   class UpdateService < MergeRequests::BaseService
-    prepend ::EE::MergeRequests::UpdateService
+    prepend ::EE::MergeRequests::UpdateService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def execute(merge_request)
       # We don't allow change of source/target projects and source branch

app/services/notes/build_service.rb

@@ -2,7 +2,7 @@
 
 module Notes
   class BuildService < ::BaseService
-    prepend ::EE::Notes::BuildService
+    prepend ::EE::Notes::BuildService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def execute
       should_resolve = false

app/services/notification_recipient_service.rb

@@ -242,7 +242,7 @@ module NotificationRecipientService
     end
 
     class Default < Base
-      prepend ::EE::NotificationRecipientBuilders::Default
+      prepend ::EE::NotificationRecipientBuilders::Default # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       MENTION_TYPE_ACTIONS = [:new_issue, :new_merge_request].freeze
 

app/services/projects/after_rename_service.rb

@@ -12,7 +12,7 @@ module Projects
   #
   #     Projects::AfterRenameService.new(project).execute
   class AfterRenameService
-    prepend ::EE::Projects::AfterRenameService
+    prepend ::EE::Projects::AfterRenameService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     attr_reader :project, :full_path_before, :full_path_after, :path_before
 

app/services/projects/autocomplete_service.rb

@@ -2,7 +2,7 @@
 
 module Projects
   class AutocompleteService < BaseService
-    prepend EE::Projects::AutocompleteService
+    prepend EE::Projects::AutocompleteService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     include LabelsAsHash
     def issues

app/services/projects/create_from_template_service.rb

@@ -2,7 +2,7 @@
 
 module Projects
   class CreateFromTemplateService < BaseService
-    prepend ::EE::Projects::CreateFromTemplateService
+    prepend ::EE::Projects::CreateFromTemplateService # rubocop: disable Cop/InjectEnterpriseEditionModule
     include Gitlab::Utils::StrongMemoize
 
     def initialize(user, params)

app/services/projects/create_service.rb

@@ -2,7 +2,7 @@
 
 module Projects
   class CreateService < BaseService
-    prepend ::EE::Projects::CreateService
+    prepend ::EE::Projects::CreateService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def initialize(user, params)
       @current_user, @params = user, params.dup

app/services/projects/destroy_service.rb

@@ -3,7 +3,7 @@
 module Projects
   class DestroyService < BaseService
     include Gitlab::ShellAdapter
-    prepend ::EE::Projects::DestroyService
+    prepend ::EE::Projects::DestroyService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     DestroyError = Class.new(StandardError)
 

app/services/projects/gitlab_projects_import_service.rb

@@ -5,7 +5,7 @@
 # The latter will under the hood just import an archive supplied by GitLab.
 module Projects
   class GitlabProjectsImportService
-    prepend ::EE::Projects::GitlabProjectsImportService
+    prepend ::EE::Projects::GitlabProjectsImportService # rubocop: disable Cop/InjectEnterpriseEditionModule
     include Gitlab::Utils::StrongMemoize
     include Gitlab::TemplateHelper
 

app/services/projects/group_links/create_service.rb

@@ -3,7 +3,7 @@
 module Projects
   module GroupLinks
     class CreateService < BaseService
-      prepend ::EE::Projects::GroupLinks::CreateService
+      prepend ::EE::Projects::GroupLinks::CreateService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def execute(group)
         return false unless group

app/services/projects/group_links/destroy_service.rb

@@ -3,7 +3,7 @@
 module Projects
   module GroupLinks
     class DestroyService < BaseService
-      prepend ::EE::Projects::GroupLinks::DestroyService
+      prepend ::EE::Projects::GroupLinks::DestroyService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def execute(group_link)
         return false unless group_link

app/services/projects/hashed_storage/migrate_attachments_service.rb

@@ -7,7 +7,7 @@ module Projects
     class MigrateAttachmentsService < BaseService
       attr_reader :logger, :old_disk_path, :new_disk_path
 
-      prepend ::EE::Projects::HashedStorage::MigrateAttachmentsService
+      prepend ::EE::Projects::HashedStorage::MigrateAttachmentsService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def initialize(project, old_disk_path, logger: nil)
         @project = project

app/services/projects/hashed_storage/migrate_repository_service.rb

@@ -7,7 +7,7 @@ module Projects
     class MigrateRepositoryService < BaseService
       include Gitlab::ShellAdapter
 
-      prepend ::EE::Projects::HashedStorage::MigrateRepositoryService
+      prepend ::EE::Projects::HashedStorage::MigrateRepositoryService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       attr_reader :old_disk_path, :new_disk_path, :old_wiki_disk_path, :old_storage_version, :logger, :move_wiki
 

app/services/projects/transfer_service.rb

@@ -13,7 +13,7 @@ module Projects
     include Gitlab::ShellAdapter
     TransferError = Class.new(StandardError)
 
-    prepend ::EE::Projects::TransferService
+    prepend ::EE::Projects::TransferService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def execute(new_namespace)
       @new_namespace = new_namespace

app/services/projects/update_service.rb

@@ -4,7 +4,7 @@ module Projects
   class UpdateService < BaseService
     include UpdateVisibilityLevel
 
-    prepend ::EE::Projects::UpdateService
+    prepend ::EE::Projects::UpdateService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     ValidationError = Class.new(StandardError)
 

app/services/users/build_service.rb

@@ -2,7 +2,7 @@
 
 module Users
   class BuildService < BaseService
-    prepend ::EE::Users::BuildService
+    prepend ::EE::Users::BuildService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     delegate :user_default_internal_regex_enabled?,
              :user_default_internal_regex_instance,

app/services/users/destroy_service.rb

@@ -2,7 +2,7 @@
 
 module Users
   class DestroyService
-    prepend ::EE::Users::DestroyService
+    prepend ::EE::Users::DestroyService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     DestroyError = Class.new(StandardError)
 

app/uploaders/object_storage.rb

@@ -137,7 +137,7 @@ module ObjectStorage
     included do |base|
       base.include(ObjectStorage)
 
-      include ::EE::ObjectStorage::Concern
+      include ::EE::ObjectStorage::Concern # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       after :migrate, :delete_migrated_file
     end

app/workers/repository_check/batch_worker.rb

@@ -2,7 +2,7 @@
 
 module RepositoryCheck
   class BatchWorker
-    prepend ::EE::RepositoryCheck::BatchWorker
+    prepend ::EE::RepositoryCheck::BatchWorker # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     include ApplicationWorker
     include RepositoryCheckQueue

app/workers/repository_check/single_repository_worker.rb

@@ -5,7 +5,7 @@ module RepositoryCheck
     include ApplicationWorker
     include RepositoryCheckQueue
 
-    prepend ::EE::RepositoryCheck::SingleRepositoryWorker
+    prepend ::EE::RepositoryCheck::SingleRepositoryWorker # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def perform(project_id)
       project = Project.find(project_id)

config/initializers/rack_attack_global.rb

@@ -43,7 +43,7 @@ class Rack::Attack
   end
 
   class Request
-    prepend ::EE::Gitlab::Rack::Attack::Request
+    prepend ::EE::Gitlab::Rack::Attack::Request # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def unauthenticated?
       !authenticated_user_id([:api, :rss, :ics])

ee/app/controllers/ee/confirmations_controller.rb

@@ -2,7 +2,7 @@
 
 module EE
   module ConfirmationsController
-    include EE::Audit::Changes
+    include EE::Audit::Changes # rubocop: disable Cop/InjectEnterpriseEditionModule
     extend ::Gitlab::Utils::Override
 
     protected

ee/app/services/ee/emails/create_service.rb

@@ -3,7 +3,7 @@
 module EE
   module Emails
     module CreateService
-      include ::EE::Emails::BaseService
+      include ::EE::Emails::BaseService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def execute(*args, &blk)
         super.tap do |email|

ee/app/services/ee/emails/destroy_service.rb

@@ -3,7 +3,7 @@
 module EE
   module Emails
     module DestroyService
-      include ::EE::Emails::BaseService
+      include ::EE::Emails::BaseService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def execute(*args, &blk)
         super.tap do

ee/app/services/ee/users/update_service.rb

@@ -3,7 +3,7 @@
 module EE
   module Users
     module UpdateService
-      include EE::Audit::Changes
+      include EE::Audit::Changes # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       private
 

ee/lib/ee/api/boards.rb

@@ -6,7 +6,7 @@ module EE
       include ::API::PaginationParams
       include ::API::BoardsResponses
 
-      prepend EE::API::BoardsResponses
+      prepend EE::API::BoardsResponses # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       before { authenticate! }
 

ee/lib/ee/api/group_boards.rb

@@ -6,7 +6,7 @@ module EE
       include ::API::PaginationParams
       include ::API::BoardsResponses
 
-      prepend EE::API::BoardsResponses
+      prepend EE::API::BoardsResponses # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       before do
         authenticate!

lib/api/api_guard.rb

@@ -41,7 +41,7 @@ module API
 
     # Helper Methods for Grape Endpoint
     module HelperMethods
-      prepend EE::API::APIGuard::HelperMethods
+      prepend EE::API::APIGuard::HelperMethods # rubocop: disable Cop/InjectEnterpriseEditionModule
       include Gitlab::Auth::UserAuthFinders
 
       def find_current_user!

lib/api/boards.rb

@@ -5,7 +5,7 @@ module API
     include BoardsResponses
     include PaginationParams
 
-    prepend EE::API::BoardsResponses
+    prepend EE::API::BoardsResponses # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     before { authenticate! }
 

lib/api/group_boards.rb

@@ -5,7 +5,7 @@ module API
     include BoardsResponses
     include PaginationParams
 
-    prepend EE::API::BoardsResponses
+    prepend EE::API::BoardsResponses # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     before do
       authenticate!

lib/api/helpers.rb

@@ -2,7 +2,7 @@
 
 module API
   module Helpers
-    prepend EE::API::Helpers
+    prepend EE::API::Helpers # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     include Gitlab::Utils
     include Gitlab::Utils::StrongMemoize

lib/api/helpers/notes_helpers.rb

@@ -3,7 +3,7 @@
 module API
   module Helpers
     module NotesHelpers
-      prepend EE::API::Helpers::NotesHelpers
+      prepend EE::API::Helpers::NotesHelpers # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def update_note(noteable, note_id)
         note = noteable.notes.find(params[:note_id])

lib/api/helpers/project_snapshots_helpers.rb

@@ -3,7 +3,7 @@
 module API
   module Helpers
     module ProjectSnapshotsHelpers
-      prepend ::EE::API::Helpers::ProjectSnapshotsHelpers
+      prepend ::EE::API::Helpers::ProjectSnapshotsHelpers # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def authorize_read_git_snapshot!
         authenticated_with_full_private_access!

lib/api/helpers/runner.rb

@@ -3,7 +3,7 @@
 module API
   module Helpers
     module Runner
-      prepend EE::API::Helpers::Runner
+      prepend EE::API::Helpers::Runner # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       JOB_TOKEN_HEADER = 'HTTP_JOB_TOKEN'.freeze
       JOB_TOKEN_PARAM = :token

lib/api/issues.rb

@@ -17,7 +17,7 @@ module API
       end
     end
 
-    prepend EE::API::Issues
+    prepend EE::API::Issues # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     helpers do
       # rubocop: disable CodeReuse/ActiveRecord

lib/api/job_artifacts.rb

@@ -11,7 +11,7 @@ module API
       end
     end
 
-    prepend EE::API::JobArtifacts
+    prepend EE::API::JobArtifacts # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     params do
       requires :id, type: String, desc: 'The ID of a project'

lib/api/merge_requests.rb

@@ -29,7 +29,7 @@ module API
       ]
     end
 
-    prepend EE::API::MergeRequests
+    prepend EE::API::MergeRequests # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     helpers do
       # rubocop: disable CodeReuse/ActiveRecord

lib/api/namespaces.rb

@@ -17,7 +17,7 @@ module API
       end
     end
 
-    prepend EE::API::Namespaces
+    prepend EE::API::Namespaces # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     resource :namespaces do
       desc 'Get a namespaces list' do

lib/api/project_clusters.rb

@@ -16,7 +16,7 @@ module API
       end
     end
 
-    prepend EE::API::ProjectClusters
+    prepend EE::API::ProjectClusters # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     params do
       requires :id, type: String, desc: 'The ID of the project'

lib/api/projects.rb

@@ -32,7 +32,7 @@ module API
       end
     end
 
-    prepend EE::API::Projects
+    prepend EE::API::Projects # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def self.update_params_at_least_one_of
       [

lib/api/todos.rb

@@ -3,7 +3,7 @@
 module API
   class Todos < Grape::API
     include PaginationParams
-    prepend EE::API::Todos
+    prepend EE::API::Todos # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     before { authenticate! }
 

lib/banzai/filter/epic_reference_filter.rb

@@ -4,7 +4,7 @@ module Banzai
   module Filter
     # The actual filter is implemented in the EE mixin
     class EpicReferenceFilter < IssuableReferenceFilter
-      prepend EE::Banzai::Filter::EpicReferenceFilter
+      prepend EE::Banzai::Filter::EpicReferenceFilter # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       self.reference_type = :epic
 

lib/banzai/issuable_extractor.rb

@@ -9,7 +9,7 @@ module Banzai
   # so we can avoid N+1 queries problem
 
   class IssuableExtractor
-    prepend EE::Banzai::IssuableExtractor
+    prepend EE::Banzai::IssuableExtractor # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     attr_reader :context
 

lib/banzai/pipeline/gfm_pipeline.rb

@@ -3,7 +3,7 @@
 module Banzai
   module Pipeline
     class GfmPipeline < BasePipeline
-      prepend EE::Banzai::Pipeline::GfmPipeline
+      prepend EE::Banzai::Pipeline::GfmPipeline # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       # These filters convert GitLab Flavored Markdown (GFM) to HTML.
       # The handlers defined in app/assets/javascripts/behaviors/markdown/copy_as_gfm.js

lib/banzai/pipeline/post_process_pipeline.rb

@@ -3,7 +3,7 @@
 module Banzai
   module Pipeline
     class PostProcessPipeline < BasePipeline
-      prepend EE::Banzai::Pipeline::PostProcessPipeline
+      prepend EE::Banzai::Pipeline::PostProcessPipeline # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def self.filters
         @filters ||= FilterArray[

lib/banzai/pipeline/single_line_pipeline.rb

@@ -3,7 +3,7 @@
 module Banzai
   module Pipeline
     class SingleLinePipeline < GfmPipeline
-      prepend EE::Banzai::Pipeline::SingleLinePipeline
+      prepend EE::Banzai::Pipeline::SingleLinePipeline # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def self.filters
         @filters ||= FilterArray[

lib/banzai/reference_parser/epic_parser.rb

@@ -4,7 +4,7 @@ module Banzai
   module ReferenceParser
     # The actual parser is implemented in the EE mixin
     class EpicParser < IssuableParser
-      prepend ::EE::Banzai::ReferenceParser::EpicParser
+      prepend ::EE::Banzai::ReferenceParser::EpicParser # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       self.reference_type = :epic
 

lib/gitlab/access.rb

@@ -7,7 +7,7 @@
 #
 module Gitlab
   module Access
-    extend ::EE::Gitlab::Access
+    extend ::EE::Gitlab::Access # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     AccessDeniedError = Class.new(StandardError)
 

lib/gitlab/auth.rb

@@ -16,7 +16,7 @@ module Gitlab
     DEFAULT_SCOPES = [:api].freeze
 
     class << self
-      prepend EE::Gitlab::Auth
+      prepend EE::Gitlab::Auth # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def omniauth_enabled?
         Gitlab.config.omniauth.enabled

lib/gitlab/auth/ldap/access.rb

@@ -8,7 +8,7 @@ module Gitlab
   module Auth
     module LDAP
       class Access
-        prepend ::EE::Gitlab::Auth::LDAP::Access
+        prepend ::EE::Gitlab::Auth::LDAP::Access # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         attr_reader :provider, :user, :ldap_identity
 

lib/gitlab/auth/ldap/adapter.rb

@@ -4,7 +4,7 @@ module Gitlab
   module Auth
     module LDAP
       class Adapter
-        prepend ::EE::Gitlab::Auth::LDAP::Adapter
+        prepend ::EE::Gitlab::Auth::LDAP::Adapter # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         SEARCH_RETRY_FACTOR = [1, 1, 2, 3].freeze
         MAX_SEARCH_RETRIES = Rails.env.test? ? 1 : SEARCH_RETRY_FACTOR.size.freeze

lib/gitlab/auth/ldap/config.rb

@@ -5,7 +5,7 @@ module Gitlab
   module Auth
     module LDAP
       class Config
-        prepend ::EE::Gitlab::Auth::LDAP::Config
+        prepend ::EE::Gitlab::Auth::LDAP::Config # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         NET_LDAP_ENCRYPTION_METHOD = {
           simple_tls: :simple_tls,

lib/gitlab/auth/ldap/person.rb

@@ -4,7 +4,7 @@ module Gitlab
   module Auth
     module LDAP
       class Person
-        prepend ::EE::Gitlab::Auth::LDAP::Person
+        prepend ::EE::Gitlab::Auth::LDAP::Person # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         # Active Directory-specific LDAP filter that checks if bit 2 of the
         # userAccountControl attribute is set.

lib/gitlab/auth/ldap/user.rb

@@ -11,7 +11,7 @@ module Gitlab
     module LDAP
       class User < Gitlab::Auth::OAuth::User
         extend ::Gitlab::Utils::Override
-        prepend ::EE::Gitlab::Auth::LDAP::User
+        prepend ::EE::Gitlab::Auth::LDAP::User # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         class << self
           # rubocop: disable CodeReuse/ActiveRecord

lib/gitlab/auth/o_auth/auth_hash.rb

@@ -6,7 +6,7 @@ module Gitlab
   module Auth
     module OAuth
       class AuthHash
-        prepend ::EE::Gitlab::Auth::OAuth::AuthHash
+        prepend ::EE::Gitlab::Auth::OAuth::AuthHash # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         attr_reader :auth_hash
         def initialize(auth_hash)

lib/gitlab/auth/o_auth/user.rb

@@ -9,7 +9,7 @@ module Gitlab
   module Auth
     module OAuth
       class User
-        prepend ::EE::Gitlab::Auth::OAuth::User
+        prepend ::EE::Gitlab::Auth::OAuth::User # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         SignupDisabledError = Class.new(StandardError)
         SigninDisabledForProviderError = Class.new(StandardError)

lib/gitlab/auth/result.rb

@@ -4,7 +4,7 @@
 module Gitlab
   module Auth
     Result = Struct.new(:actor, :project, :type, :authentication_abilities) do
-      prepend ::EE::Gitlab::Auth::Result
+      prepend ::EE::Gitlab::Auth::Result # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def ci?(for_project)
         type == :ci &&

lib/gitlab/auth/saml/config.rb

@@ -4,7 +4,7 @@ module Gitlab
   module Auth
     module Saml
       class Config
-        prepend ::EE::Gitlab::Auth::Saml::Config
+        prepend ::EE::Gitlab::Auth::Saml::Config # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         class << self
           def options

lib/gitlab/auth/saml/user.rb

@@ -9,7 +9,7 @@ module Gitlab
   module Auth
     module Saml
       class User < Gitlab::Auth::OAuth::User
-        prepend ::EE::Gitlab::Auth::Saml::User
+        prepend ::EE::Gitlab::Auth::Saml::User # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         extend ::Gitlab::Utils::Override
 

lib/gitlab/auth/user_auth_finders.rb

@@ -18,7 +18,7 @@ module Gitlab
     end
 
     module UserAuthFinders
-      prepend ::EE::Gitlab::Auth::UserAuthFinders
+      prepend ::EE::Gitlab::Auth::UserAuthFinders # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       include Gitlab::Utils::StrongMemoize
 

lib/gitlab/background_migration/redact_links.rb

@@ -6,7 +6,7 @@ require_relative 'redact_links/redactable'
 module Gitlab
   module BackgroundMigration
     class RedactLinks
-      prepend EE::Gitlab::BackgroundMigration::RedactLinks
+      prepend EE::Gitlab::BackgroundMigration::RedactLinks # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       class Note < ActiveRecord::Base
         include EachBatch

lib/gitlab/checks/base_checker.rb

@@ -3,7 +3,7 @@
 module Gitlab
   module Checks
     class BaseChecker
-      prepend EE::Gitlab::Checks::BaseChecker
+      prepend EE::Gitlab::Checks::BaseChecker # rubocop: disable Cop/InjectEnterpriseEditionModule
       include Gitlab::Utils::StrongMemoize
 
       attr_reader :change_access

lib/gitlab/checks/change_access.rb

@@ -3,7 +3,7 @@
 module Gitlab
   module Checks
     class ChangeAccess
-      prepend EE::Gitlab::Checks::ChangeAccess
+      prepend EE::Gitlab::Checks::ChangeAccess # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       ATTRIBUTES = %i[user_access project skip_authorization
                       skip_lfs_integrity_check protocol oldrev newrev ref

lib/gitlab/checks/diff_check.rb

@@ -4,7 +4,7 @@ module Gitlab
   module Checks
     class DiffCheck < BaseChecker
       include Gitlab::Utils::StrongMemoize
-      prepend EE::Gitlab::Checks::DiffCheck
+      prepend EE::Gitlab::Checks::DiffCheck # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       LOG_MESSAGES = {
         validate_file_paths: "Validating diffs' file paths...",

lib/gitlab/ci/parsers.rb

@@ -3,7 +3,7 @@
 module Gitlab
   module Ci
     module Parsers
-      prepend ::EE::Gitlab::Ci::Parsers
+      prepend ::EE::Gitlab::Ci::Parsers # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       ParserNotFoundError = Class.new(ParserError)
 

lib/gitlab/ci/status/build/failed.rb

@@ -20,7 +20,7 @@ module Gitlab
 
           private_constant :REASONS
 
-          prepend ::EE::Gitlab::Ci::Status::Build::Failed
+          prepend ::EE::Gitlab::Ci::Status::Build::Failed # rubocop: disable Cop/InjectEnterpriseEditionModule
 
           def status_tooltip
             base_message

lib/gitlab/database.rb

@@ -12,7 +12,7 @@ module Gitlab
     MAX_TIMESTAMP_VALUE = Time.at((1 << 31) - 1).freeze
 
     class << self
-      prepend EE::Gitlab::Database
+      prepend EE::Gitlab::Database # rubocop: disable Cop/InjectEnterpriseEditionModule
     end
 
     def self.config

lib/gitlab/email/handler.rb

@@ -3,7 +3,7 @@
 module Gitlab
   module Email
     module Handler
-      prepend ::EE::Gitlab::Email::Handler
+      prepend ::EE::Gitlab::Email::Handler # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def self.handlers
         @handlers ||= load_handlers

lib/gitlab/etag_caching/router.rb

@@ -3,7 +3,7 @@
 module Gitlab
   module EtagCaching
     class Router
-      prepend EE::Gitlab::EtagCaching::Router
+      prepend EE::Gitlab::EtagCaching::Router # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       Route = Struct.new(:regexp, :name)
       # We enable an ETag for every request matching the regex.

lib/gitlab/git_access.rb

@@ -4,7 +4,7 @@
 # class return an instance of `GitlabAccessStatus`
 module Gitlab
   class GitAccess
-    prepend ::EE::Gitlab::GitAccess
+    prepend ::EE::Gitlab::GitAccess # rubocop: disable Cop/InjectEnterpriseEditionModule
     include ActionView::Helpers::SanitizeHelper
     include PathLocksHelper
     include Gitlab::Utils::StrongMemoize

lib/gitlab/git_access_wiki.rb

@@ -2,7 +2,7 @@
 
 module Gitlab
   class GitAccessWiki < GitAccess
-    prepend EE::Gitlab::GitAccessWiki
+    prepend EE::Gitlab::GitAccessWiki # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     ERROR_MESSAGES = {
       read_only:     "You can't push code to a read-only GitLab instance.",

lib/gitlab/github_import/parallel_importer.rb

@@ -5,7 +5,7 @@ module Gitlab
     # The ParallelImporter schedules the importing of a GitHub project using
     # Sidekiq.
     class ParallelImporter
-      prepend ::EE::Gitlab::GithubImport::ParallelImporter
+      prepend ::EE::Gitlab::GithubImport::ParallelImporter # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       attr_reader :project
 

lib/gitlab/hook_data/issue_builder.rb

@@ -3,7 +3,7 @@
 module Gitlab
   module HookData
     class IssueBuilder < BaseBuilder
-      prepend ::EE::Gitlab::HookData::IssueBuilder
+      prepend ::EE::Gitlab::HookData::IssueBuilder # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       SAFE_HOOK_RELATIONS = %i[
         assignees

lib/gitlab/import_export/relation_factory.rb

@@ -3,7 +3,7 @@
 module Gitlab
   module ImportExport
     class RelationFactory
-      prepend ::EE::Gitlab::ImportExport::RelationFactory
+      prepend ::EE::Gitlab::ImportExport::RelationFactory # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       OVERRIDES = { snippets: :project_snippets,
                     ci_pipelines: 'Ci::Pipeline',

lib/gitlab/import_sources.rb

@@ -24,7 +24,7 @@ module Gitlab
     ].freeze
 
     class << self
-      prepend EE::Gitlab::ImportSources
+      prepend EE::Gitlab::ImportSources # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def options
         Hash[import_table.map { |importer| [importer.title, importer.name] }]

lib/gitlab/middleware/read_only/controller.rb

@@ -4,7 +4,7 @@ module Gitlab
   module Middleware
     class ReadOnly
       class Controller
-        prepend EE::Gitlab::Middleware::ReadOnly::Controller
+        prepend EE::Gitlab::Middleware::ReadOnly::Controller # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         DISALLOWED_METHODS = %w(POST PATCH PUT DELETE).freeze
         APPLICATION_JSON = 'application/json'.freeze

lib/gitlab/omniauth_initializer.rb

@@ -2,7 +2,7 @@
 
 module Gitlab
   class OmniauthInitializer
-    prepend ::EE::Gitlab::OmniauthInitializer
+    prepend ::EE::Gitlab::OmniauthInitializer # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def initialize(devise_config)
       @devise_config = devise_config

lib/gitlab/patch/draw_route.rb

@@ -5,7 +5,7 @@
 module Gitlab
   module Patch
     module DrawRoute
-      prepend EE::Gitlab::Patch::DrawRoute
+      prepend EE::Gitlab::Patch::DrawRoute # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       RoutesNotFound = Class.new(StandardError)
 

lib/gitlab/prometheus/metric_group.rb

@@ -3,7 +3,7 @@
 module Gitlab
   module Prometheus
     class MetricGroup
-      prepend EE::Gitlab::Prometheus::MetricGroup
+      prepend EE::Gitlab::Prometheus::MetricGroup # rubocop: disable Cop/InjectEnterpriseEditionModule
       include ActiveModel::Model
 
       attr_accessor :name, :priority, :metrics

lib/gitlab/prometheus/queries/query_additional_metrics.rb

@@ -4,7 +4,7 @@ module Gitlab
   module Prometheus
     module Queries
       module QueryAdditionalMetrics
-        prepend EE::Gitlab::Prometheus::Queries::QueryAdditionalMetrics
+        prepend EE::Gitlab::Prometheus::Queries::QueryAdditionalMetrics # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         def query_metrics(project, environment, query_context)
           matched_metrics(project).map(&query_group(query_context))

lib/gitlab/regex.rb

@@ -3,7 +3,7 @@
 module Gitlab
   module Regex
     extend self
-    extend EE::Gitlab::Regex
+    extend EE::Gitlab::Regex # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def namespace_name_regex
       @namespace_name_regex ||= /\A[\p{Alnum}\p{Pd}_\. ]*\z/.freeze

lib/gitlab/search/parsed_query.rb

@@ -3,7 +3,7 @@
 module Gitlab
   module Search
     class ParsedQuery
-      prepend EE::Gitlab::Search::ParsedQuery
+      prepend EE::Gitlab::Search::ParsedQuery # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       attr_reader :term, :filters
 

lib/gitlab/slash_commands/command.rb

@@ -3,7 +3,7 @@
 module Gitlab
   module SlashCommands
     class Command < BaseCommand
-      prepend EE::Gitlab::SlashCommands::Command
+      prepend EE::Gitlab::SlashCommands::Command # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def self.commands
         [

lib/gitlab/slash_commands/presenters/issue_base.rb

@@ -4,7 +4,7 @@ module Gitlab
   module SlashCommands
     module Presenters
       module IssueBase
-        prepend EE::Gitlab::SlashCommands::Presenters::IssueBase
+        prepend EE::Gitlab::SlashCommands::Presenters::IssueBase # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         def color(issuable)
           issuable.open? ? '#38ae67' : '#d22852'

lib/gitlab/tree_summary.rb

@@ -2,7 +2,7 @@
 
 module Gitlab
   class TreeSummary
-    prepend ::EE::Gitlab::TreeSummary
+    prepend ::EE::Gitlab::TreeSummary # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     include ::Gitlab::Utils::StrongMemoize
 

lib/gitlab/usage_data.rb

@@ -5,7 +5,7 @@ module Gitlab
     APPROXIMATE_COUNT_MODELS = [Label, MergeRequest, Note, Todo].freeze
 
     class << self
-      prepend EE::Gitlab::UsageData
+      prepend EE::Gitlab::UsageData # rubocop: disable Cop/InjectEnterpriseEditionModule
 
       def data(force_refresh: false)
         Rails.cache.fetch('usage_data', force: force_refresh, expires_in: 2.weeks) { uncached_data }

qa/qa/page/admin/menu.rb

@@ -4,7 +4,7 @@ module QA
   module Page
     module Admin
       class Menu < Page::Base
-        prepend EE::Page::Admin::Menu
+        prepend EE::Page::Admin::Menu # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         view 'app/views/layouts/nav/sidebar/_admin.html.haml' do
           element :admin_sidebar

qa/qa/page/project/new.rb

@@ -3,7 +3,7 @@ module QA
     module Project
       class New < Page::Base
         include Page::Component::Select2
-        prepend EE::Page::Project::New
+        prepend EE::Page::Project::New # rubocop: disable Cop/InjectEnterpriseEditionModule
 
         view 'app/views/projects/new.html.haml' do
           element :project_create_from_template_tab

qa/qa/page/project/settings/mirroring_repositories.rb

@@ -5,7 +5,7 @@ module QA
     module Project
       module Settings
         class MirroringRepositories < Page::Base
-          prepend EE::Page::Project::Settings::MirroringRepositories
+          prepend EE::Page::Project::Settings::MirroringRepositories # rubocop: disable Cop/InjectEnterpriseEditionModule
 
           view 'app/views/projects/mirrors/_authentication_method.html.haml' do
             element :authentication_method

qa/qa/page/project/settings/protected_branches.rb

@@ -3,7 +3,7 @@ module QA
     module Project
       module Settings
         class ProtectedBranches < Page::Base
-          prepend EE::Page::Project::Settings::ProtectedBranches
+          prepend EE::Page::Project::Settings::ProtectedBranches # rubocop: disable Cop/InjectEnterpriseEditionModule
 
           view 'app/views/projects/protected_branches/shared/_dropdown.html.haml' do
             element :protected_branch_select

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])

spec/rubocop/cop/inject_enterprise_edition_module_spec.rb

@@ -19,6 +19,41 @@ describe RuboCop::Cop::InjectEnterpriseEditionModule do
     SOURCE
   end
 
+  it 'does not flag the use of `prepend EEFoo` in the middle of a file' do
+    expect_no_offenses(<<~SOURCE)
+    class Foo
+      prepend EEFoo
+    end
+    SOURCE
+  end
+
+  it 'flags the use of `prepend EE::Foo::Bar` in the middle of a file' do
+    expect_offense(<<~SOURCE)
+    class Foo
+      prepend EE::Foo::Bar
+      ^^^^^^^^^^^^^^^^^^^^ Injecting EE modules must be done on the last line of this file, outside of any class or module definitions
+    end
+    SOURCE
+  end
+
+  it 'flags the use of `prepend(EE::Foo::Bar)` in the middle of a file' do
+    expect_offense(<<~SOURCE)
+    class Foo
+      prepend(EE::Foo::Bar)
+      ^^^^^^^^^^^^^^^^^^^^^ Injecting EE modules must be done on the last line of this file, outside of any class or module definitions
+    end
+    SOURCE
+  end
+
+  it 'flags the use of `prepend EE::Foo::Bar::Baz` in the middle of a file' do
+    expect_offense(<<~SOURCE)
+    class Foo
+      prepend EE::Foo::Bar::Baz
+      ^^^^^^^^^^^^^^^^^^^^^^^^^ Injecting EE modules must be done on the last line of this file, outside of any class or module definitions
+    end
+    SOURCE
+  end
+
   it 'flags the use of `prepend ::EE` in the middle of a file' do
     expect_offense(<<~SOURCE)
     class Foo