1 files changed, 43 insertions, 13 deletions
diff --git a/spec/features/broadcast_messages_spec.rb b/spec/features/broadcast_messages_spec.rb
index 43fbf1010c9..d7c84b8085b 100644
--- a/spec/features/broadcast_messages_spec.rb
+++ b/spec/features/broadcast_messages_spec.rb
@@ -3,28 +3,58 @@
 require 'spec_helper'
 
 describe 'Broadcast Messages' do
-  let!(:broadcast_message) { create(:broadcast_message, broadcast_type: 'notification', message: 'SampleMessage') }
+  shared_examples 'a Broadcast Messages' do
+    it 'shows broadcast message' do
+      visit root_path
 
-  it 'shows broadcast message' do
-    visit root_path
+      expect(page).to have_content 'SampleMessage'
+    end
+  end
+
+  shared_examples 'a dismissable Broadcast Messages' do
+    it 'hides broadcast message after dismiss', :js do
+      visit root_path
+
+      find('.js-dismiss-current-broadcast-notification').click
+
+      expect(page).not_to have_content 'SampleMessage'
+    end
+
+    it 'broadcast message is still hidden after refresh', :js do
+      visit root_path
+
+      find('.js-dismiss-current-broadcast-notification').click
+      visit root_path
+
+      expect(page).not_to have_content 'SampleMessage'
+    end
+  end
+
+  describe 'banner type' do
+    let!(:broadcast_message) { create(:broadcast_message, message: 'SampleMessage') }
+
+    it_behaves_like 'a Broadcast Messages'
+
+    it 'shows broadcast message' do
+      visit root_path
 
-    expect(page).to have_content 'SampleMessage'
+      expect(page).not_to have_selector('.js-dismiss-current-broadcast-notification')
+    end
   end
 
-  it 'hides broadcast message after dismiss', :js do
-    visit root_path
+  describe 'dismissable banner type' do
+    let!(:broadcast_message) { create(:broadcast_message, dismissable: true, message: 'SampleMessage') }
 
-    find('.js-dismiss-current-broadcast-notification').click
+    it_behaves_like 'a Broadcast Messages'
 
-    expect(page).not_to have_content 'SampleMessage'
+    it_behaves_like 'a dismissable Broadcast Messages'
   end
 
-  it 'broadcast message is still hidden after refresh', :js do
-    visit root_path
+  describe 'notification type' do
+    let!(:broadcast_message) { create(:broadcast_message, broadcast_type: 'notification', message: 'SampleMessage') }
 
-    find('.js-dismiss-current-broadcast-notification').click
-    visit root_path
+    it_behaves_like 'a Broadcast Messages'
 
-    expect(page).not_to have_content 'SampleMessage'
+    it_behaves_like 'a dismissable Broadcast Messages'
   end
 end