def create_message(data, ack_id='ACKID', **attrs):
with mock.patch.object(message.Message, 'lease') as lease:
with mock.patch.object(time, 'time') as time_:
time_.return_value = 1335020400
msg = message.Message(types.PubsubMessage(
attributes=attrs,
data=data,
message_id='message_id',
publish_time=types.Timestamp(seconds=1335020400 - 86400),
), ack_id, queue.Queue())
lease.assert_called_once_with()
return msg
def test_publish_time():
msg = create_message(b'foo')
assert msg.publish_time == types.Timestamp(seconds=1335020400 - 86400)
