From 241bcc25aed5298db4e203390ba75f4869cce042 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 386d73e61154..cedbfcc0d182 100644
--- a/spec/lib/gitlab/email/handler_spec.rb
+++ b/spec/lib/gitlab/email/handler_spec.rb
@@ -25,12 +25,12 @@
         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 @@
       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
-- 
GitLab