spec/unit/queue_spec.rb in hushed-0.0.3 vs spec/unit/queue_spec.rb in hushed-0.0.4

@@ -22,16 +22,17 @@
       @sqs_queue.expects(:send_message).with('hello world')
 
       @queue.send(@message)
     end
 
-    it "should return a message when receiving from a queue" do
+    it "should return a message which holds a reference to the message_object when receiving from a queue" do
       @message.stubs(:body).returns(load_message('purchase_order_message'))
       @sqs_queue.expects(:receive_message).yields(@message)
 
       received_message = @queue.receive
       assert_equal 'PurchaseOrder', received_message.document_type
       assert_equal 'HUSHED_PurchaseOrder_1234_20100927_132505124.xml', received_message.document_name
+      assert_equal @message, received_message.message_object
     end
 
     it "should return a message when receiving even if the queue did not return anything" do
       @sqs_queue.expects(:receive_message)
 
\ No newline at end of file