def test_execute(self, mock_hook):
operator = StackdriverListAlertPoliciesOperator(task_id=TEST_TASK_ID,
filter_=TEST_FILTER)
mock_hook.return_value.list_alert_policies.return_value = [
AlertPolicy(name="test-name")
]
result = operator.execute(None)
mock_hook.return_value.list_alert_policies.assert_called_once_with(
project_id=None,
filter_=TEST_FILTER,
format_=None,
order_by=None,
page_size=None,
retry=DEFAULT,
timeout=DEFAULT,
metadata=None,
)
assert [{
'combiner': 0,
'conditions': [],
'display_name': '',
'name': 'test-name',
'notification_channels': [],
'user_labels': {},
}] == result
def test_execute(self, mock_hook):
operator = StackdriverListAlertPoliciesOperator(task_id=TEST_TASK_ID, filter_=TEST_FILTER)
operator.execute(None)
mock_hook.return_value.list_alert_policies.assert_called_once_with(
project_id=None,
filter_=TEST_FILTER,
format_=None,
order_by=None,
page_size=None,
retry=DEFAULT,
timeout=DEFAULT,
metadata=None,
)
# [START howto_operator_gcp_stackdriver_enable_alert_policy]
enable_alert_policy = StackdriverEnableAlertPoliciesOperator(
task_id='enable-alert-policies',
filter_='(displayName="test alert 1" OR displayName="test alert 2")',
)
# [END howto_operator_gcp_stackdriver_enable_alert_policy]
# [START howto_operator_gcp_stackdriver_disable_alert_policy]
disable_alert_policy = StackdriverDisableAlertPoliciesOperator(
task_id='disable-alert-policies',
filter_='displayName="test alert 1"',
)
# [END howto_operator_gcp_stackdriver_disable_alert_policy]
# [START howto_operator_gcp_stackdriver_list_alert_policy]
list_alert_policies = StackdriverListAlertPoliciesOperator(
task_id='list-alert-policies', )
# [END howto_operator_gcp_stackdriver_list_alert_policy]
# [START howto_operator_gcp_stackdriver_delete_notification_channel]
delete_notification_channel = StackdriverDeleteNotificationChannelOperator(
task_id='delete-notification-channel',
name=
"{{ task_instance.xcom_pull('list-notification-channel')[0]['name'] }}",
)
# [END howto_operator_gcp_stackdriver_delete_notification_channel]
delete_notification_channel_2 = StackdriverDeleteNotificationChannelOperator(
task_id='delete-notification-channel-2',
name=
"{{ task_instance.xcom_pull('list-notification-channel')[1]['name'] }}",
)
