def test_visibility_timeout(sqs, random_string):
try:
create_standard_queue(sqs, random_string, visibility_timeout=1)
create_fifo_queue(sqs, random_string + ".fifo", visibility_timeout=1)
finally:
try:
delete_queue(sqs, random_string)
except Exception:
pass
try:
delete_queue(sqs, random_string + ".fifo")
except Exception:
pass
def queue_name_with_redrive(sqs_session, sqs, random_string):
# dead letter queue_name
queue = random_string
dead_queue = random_string + "_dead"
create_standard_queue(sqs, dead_queue)
# standard queue_name
# 1 is the minimal value for redrive, and means
# "put this to the dead letter queue_name after two failed attempts"
create_standard_queue(sqs, queue, redrive_policy=sqs.redrive_policy(dead_queue, 1))
yield queue, dead_queue
# delete all the queues
delete_queue(sqs, queue)
delete_queue(sqs, dead_queue)
def test_message_retention_period(sqs_session, sqs, random_string):
if isinstance(sqs_session, MemorySession):
pytest.skip("Message Retention id not implemented with MemorySession")
try:
create_standard_queue(sqs, random_string, message_retention_period=600)
create_fifo_queue(sqs,
random_string + ".fifo",
message_retention_period=600)
finally:
try:
delete_queue(sqs, random_string)
except Exception:
pass
try:
delete_queue(sqs, random_string + ".fifo")
except Exception:
pass
# Create a new queue.
queue_name = str(uuid.uuid4())
create_standard_queue(sqs, queue_name)
# Get the queue object
queue = sqs.queue(queue_name, NiFiQueue)
@queue.processor("ping")
def ping(flowfile: FlowFile):
print("Ping...")
@queue.processor("ping/success")
def pong(flowfile: FlowFile):
print("Pong...")
# Start PingPong
ff = FlowFile({"a": "1"})
try:
queue.add_flowfile("ping",
ff,
response_queue_name=queue_name,
response_port_prefix="ping")
queue.process_queue(shutdown_policy=MaxTasksShutdown(2))
finally:
delete_queue(sqs, queue_name)
def fifo_queue_name(sqs_session, sqs, random_string):
# type: (SQSEnv) -> string
queue_name = random_string + ".fifo"
create_fifo_queue(sqs, queue_name)
yield queue_name
delete_queue(sqs, queue_name)
def queue_name2(sqs_session, sqs, random_string):
# type: (SQSEnv) -> string
create_standard_queue(sqs, random_string + "_2")
yield random_string + "_2"
delete_queue(sqs, random_string + "_2")
