def test_publish(self, mock_hook):
operator = PubSubPublishMessageOperator(
task_id=TASK_ID, project_id=TEST_PROJECT, topic=TEST_TOPIC, messages=TEST_MESSAGES,
)
operator.execute(None)
mock_hook.return_value.publish.assert_called_once_with(
project_id=TEST_PROJECT, topic=TEST_TOPIC, messages=TEST_MESSAGES
)
task_id="pull_messages",
ack_messages=True,
project_id=GCP_PROJECT_ID,
subscription=subscription,
)
# [END howto_operator_gcp_pubsub_pull_message_with_sensor]
# [START howto_operator_gcp_pubsub_pull_messages_result]
pull_messages_result = BashOperator(task_id="pull_messages_result",
bash_command=echo_cmd)
# [END howto_operator_gcp_pubsub_pull_messages_result]
# [START howto_operator_gcp_pubsub_publish]
publish_task = PubSubPublishMessageOperator(
task_id="publish_task",
project_id=GCP_PROJECT_ID,
topic=TOPIC_FOR_SENSOR_DAG,
messages=[MESSAGE] * 10,
)
# [END howto_operator_gcp_pubsub_publish]
# [START howto_operator_gcp_pubsub_unsubscribe]
unsubscribe_task = PubSubDeleteSubscriptionOperator(
task_id="unsubscribe_task",
project_id=GCP_PROJECT_ID,
subscription=subscription,
)
# [END howto_operator_gcp_pubsub_unsubscribe]
# [START howto_operator_gcp_pubsub_delete_topic]
delete_topic = PubSubDeleteTopicOperator(task_id="delete_topic",
topic=TOPIC_FOR_SENSOR_DAG,
ack_messages=True,
project_id=GCP_PROJECT_ID,
subscription=subscription,
)
# [END howto_operator_gcp_pubsub_pull_message]
# [START howto_operator_gcp_pubsub_pull_messages_result]
pull_messages_result = BashOperator(
task_id="pull_messages_result", bash_command=echo_cmd
)
# [END howto_operator_gcp_pubsub_pull_messages_result]
# [START howto_operator_gcp_pubsub_publish]
publish_task = PubSubPublishMessageOperator(
task_id="publish_task",
project_id=GCP_PROJECT_ID,
topic=TOPIC,
messages=[MESSAGE, MESSAGE, MESSAGE],
)
# [END howto_operator_gcp_pubsub_publish]
# [START howto_operator_gcp_pubsub_unsubscribe]
unsubscribe_task = PubSubDeleteSubscriptionOperator(
task_id="unsubscribe_task",
project_id=GCP_PROJECT_ID,
subscription="{{ task_instance.xcom_pull('subscribe_task') }}",
)
# [END howto_operator_gcp_pubsub_unsubscribe]
# [START howto_operator_gcp_pubsub_delete_topic]
delete_topic = PubSubDeleteTopicOperator(
task_id="delete_topic", topic=TOPIC, project_id=GCP_PROJECT_ID