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")