1 files changed, 35 insertions, 0 deletions
diff --git a/spec/workers/build_email_worker_spec.rb b/spec/workers/build_email_worker_spec.rb
new file mode 100644
index 00000000000..98deae0a588
--- /dev/null
+++ b/spec/workers/build_email_worker_spec.rb
@@ -0,0 +1,35 @@
+require 'spec_helper'
+
+describe BuildEmailWorker do
+  include RepoHelpers
+
+  let(:build) { create(:ci_build) }
+  let(:user) { create(:user) }
+  let(:data) { Gitlab::BuildDataBuilder.build(build) }
+
+  subject { BuildEmailWorker.new }
+
+  before do
+    allow(build).to receive(:execute_hooks).and_return(false)
+    build.success
+  end
+
+  describe "#perform" do
+    it "sends mail" do
+      subject.perform(build.id, [user.email], data.stringify_keys)
+
+      email = ActionMailer::Base.deliveries.last
+      expect(email.subject).to include('Build success for')
+      expect(email.to).to eq([user.email])
+    end
+
+    it "gracefully handles an input SMTP error" do
+      ActionMailer::Base.deliveries.clear
+      allow(Notify).to receive(:build_success_email).and_raise(Net::SMTPFatalError)
+
+      subject.perform(build.id, [user.email], data.stringify_keys)
+
+      expect(ActionMailer::Base.deliveries.count).to eq(0)
+    end
+  end
+end