1 files changed, 16 insertions, 1 deletions
diff --git a/spec/views/notify/change_in_merge_request_draft_status_email.html.haml_spec.rb b/spec/views/notify/change_in_merge_request_draft_status_email.html.haml_spec.rb
index 6d56145144f..deef13fec99 100644
--- a/spec/views/notify/change_in_merge_request_draft_status_email.html.haml_spec.rb
+++ b/spec/views/notify/change_in_merge_request_draft_status_email.html.haml_spec.rb
@@ -12,10 +12,25 @@ RSpec.describe 'notify/change_in_merge_request_draft_status_email.html.haml' do
     assign(:merge_request, merge_request)
   end
 
+  it 'shows user added draft status on email' do
+    merge_request.update!(title: merge_request.draft_title)
+
+    render
+
+    expect(merge_request.draft).to be_truthy
+    expect(rendered).to have_content("#{user.name} marked merge request #{merge_request.to_reference} as draft")
+  end
+
+  it 'shows user removed draft status on email' do
+    render
+
+    expect(merge_request.draft).to be_falsy
+    expect(rendered).to have_content("#{user.name} marked merge request #{merge_request.to_reference} as ready")
+  end
+
   it 'renders the email correctly' do
     render
 
-    expect(rendered).to have_content("#{user.name} changed the draft status of merge request #{merge_request.to_reference}")
     expect(rendered).to have_link(user.name, href: user_url(user))
     expect(rendered).to have_link(merge_request.to_reference, href: merge_request_link)
   end