From e67759bbc911a453d47e6d3afab8ad1de51bf1ce Mon Sep 17 00:00:00 2001
From: Michael Kozono <mkozono@gmail.com>
Date: Mon, 9 Apr 2018 13:30:43 -0700
Subject: [PATCH] Fix email handler spec for EE

---
 spec/lib/gitlab/email/handler_spec.rb | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/spec/lib/gitlab/email/handler_spec.rb b/spec/lib/gitlab/email/handler_spec.rb
index 386d73e6115..cedbfcc0d18 100644
--- a/spec/lib/gitlab/email/handler_spec.rb
+++ b/spec/lib/gitlab/email/handler_spec.rb
@@ -25,12 +25,12 @@ describe Gitlab::Email::Handler do
         described_class.for('email', address).class
       end
 
-      expect(matched_handlers.uniq).to match_array(Gitlab::Email::Handler::HANDLERS)
+      expect(matched_handlers.uniq).to match_array(ce_handlers)
     end
 
     it 'can pick exactly one handler for each address' do
       addresses.each do |address|
-        matched_handlers = Gitlab::Email::Handler::HANDLERS.select do |handler|
+        matched_handlers = ce_handlers.select do |handler|
           handler.new('email', address).can_handle?
         end
 
@@ -38,4 +38,10 @@ describe Gitlab::Email::Handler do
       end
     end
   end
+
+  def ce_handlers
+    @ce_handlers ||= Gitlab::Email::Handler::HANDLERS.reject do |handler|
+      handler.name.start_with?('Gitlab::Email::Handler::EE::')
+    end
+  end
 end