def test_get_message_exist_uncallable_message():
queue = Queue("test-queue")
with mock.patch("faws.sqs.message.Message.is_callable",
return_value=False):
queue.add_message("takerun")
assert queue.get_message() == []
def test_add_message(dt, delay_seconds):
now = datetime(2020, 5, 20, 0, 0, 0)
expected_deliverable_time = datetime(2020, 5, 20, 0, 0, 0 + delay_seconds)
dt.now.return_value = now
queue = Queue("test-queue")
message = queue.add_message("takerun", delay_seconds=delay_seconds)
assert message.message_deliverable_time == expected_deliverable_time
def test_get_message(visibility_timeout, deliverable_second, num_of_message):
queue = Queue("test-queue")
now = datetime(2020, 5, 1, 0, 0, 0)
with mock.patch("datetime.datetime") as dt:
dt.now.return_value = now
added_message = []
for i in range(num_of_message):
message = queue.add_message("takerun")
added_message.append(message)
received_message = queue.get_message(
visibility_timeout, max_number_of_messages=num_of_message)
assert (received_message == added_message and len(
# deliverable timeが正常な値を持たないmessageを抽出し、0件であることをチェック
[
message for message in added_message
if message.message_deliverable_time != datetime(
2020, 5, 1, 0, 0, deliverable_second)
]) == 0)
def test_purge_message():
queue = Queue("test-queue")
queue.add_message("hoge")
queue.purge_message()
assert queue.get_message() == []