def test_get_message(self):
self.login(self.user.email)
# Add a message.
message = Message(from_user=self.contact, to_user=self.user,
message='test message', ttl=10)
db.session.add(message)
db.session.commit()
# Get message from API.
rv = self.client.get('/%s/message/%s' % (API_VERSION, message.id))
eq_(rv.status_code, 200)
data = json.loads(rv.data)
message_data = message.to_json()
for k in data['data'].keys():
eq_(data['data'][k], message_data[k])
# Verify that the message was set to expire.
message = Message.query.filter(Message.id==message.id).one()
ok_(message.expire is not None)
