Add some tests for user_can_view? on outgoing message.

1 files changed, 52 insertions, 0 deletions

diff --git a/spec/models/outgoing_message_spec.rb b/spec/models/outgoing_message_spec.rb
index d0816d01f..4bd9361a7 100644
--- a/spec/models/outgoing_message_spec.rb
+++ b/spec/models/outgoing_message_spec.rb
@@ -66,6 +66,58 @@ describe OutgoingMessage, " when making an outgoing message" do
         outgoing_message.body.should include(expected_text)
     end
 
+    describe 'when asked if a user can view it' do
+
+        before do
+            @info_request = FactoryGirl.create(:info_request)
+            @outgoing_message = @info_request.outgoing_messages.first
+        end
+
+        context 'if the prominence is hidden' do
+
+            before do
+                @outgoing_message.prominence = 'hidden'
+            end
+
+            it 'should return true for an admin user' do
+                @outgoing_message.user_can_view?(FactoryGirl.create(:admin_user)).should be_true
+            end
+
+            it 'should return false for a non-admin user' do
+                @outgoing_message.user_can_view?(FactoryGirl.create(:user)).should be_false
+            end
+
+        end
+
+        context 'if the prominence is requester_only' do
+
+            before do
+                @outgoing_message.prominence = 'requester_only'
+            end
+
+            it 'should return true if the user owns the associated request' do
+                @outgoing_message.user_can_view?(@info_request.user).should be_true
+            end
+
+            it 'should return false if the user does not own the associated request' do
+                @outgoing_message.user_can_view?(FactoryGirl.create(:user)).should be_false
+            end
+        end
+
+        context 'if the prominence is normal' do
+
+            before do
+                @outgoing_message.prominence = 'normal'
+            end
+
+            it 'should return true for a non-admin user' do
+                @outgoing_message.user_can_view?(FactoryGirl.create(:user)).should be_true
+            end
+
+        end
+
+    end
+
 end