spec/email_message_spec.rb in govdelivery-tms-0.8.5 vs spec/email_message_spec.rb in govdelivery-tms-0.8.7

@@ -77,23 +77,26 @@
     before do
       # blank hash prevents the client from doing a GET in the initialize method
       @message = GovDelivery::TMS::EmailMessage.new(client, '/messages/99', {})
     end
     it 'should GET cleanly' do
-      response = { body: 'processed',
-                   subject:    'hey',
-                   from_email: '[e-mail]',
-                   errors_to:  '[e-mail]',
-                   reply_to:   '[e-mail]',
-                   recipients: [{ email: '[e-mail]' }],
-                   created_at: 'time' }
+      response = { 'body'       => 'processed',
+                   'subject'    => 'hey',
+                   'from_email' => '[e-mail]',
+                   'errors_to'  => '[e-mail]',
+                   'reply_to'   => '[e-mail]',
+                   'recipients' => [{ email: '[e-mail]' }],
+                   'created_at' => 'time',
+                   '_links'     => { 'self' => '/messages/email/1', 'email_template' => '/templates/email/1' }
+                  }
       expect(@message.client).to receive('get').with(@message.href).and_return(double('response', status: 200, body: response))
       @message.get
       expect(@message.body).to eq('processed')
       expect(@message.subject).to eq('hey')
       expect(@message.from_email).to eq('[e-mail]')
       expect(@message.reply_to).to eq('[e-mail]')
       expect(@message.errors_to).to eq('[e-mail]')
       expect(@message.created_at).to eq('time')
+      expect(@message.email_template).to be_a(GovDelivery::TMS::EmailTemplate)
     end
   end
 end