def test_execute(self, mock_hook):
operator = StackdriverListNotificationChannelsOperator(task_id=TEST_TASK_ID, filter_=TEST_FILTER)
mock_hook.return_value.list_notification_channels.return_value = [
NotificationChannel(name="test-123")
]
result = operator.execute(None)
mock_hook.return_value.list_notification_channels.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 [
{
'description': '',
'display_name': '',
'labels': {},
'name': 'test-123',
'type_': '',
'user_labels': {},
'verification_status': 0,
}
] == result
def test_execute(self, mock_hook):
operator = StackdriverListNotificationChannelsOperator(task_id=TEST_TASK_ID, filter_=TEST_FILTER)
operator.execute(None)
mock_hook.return_value.list_notification_channels.assert_called_once_with(
project_id=None,
filter_=TEST_FILTER,
format_=None,
order_by=None,
page_size=None,
retry=DEFAULT,
timeout=DEFAULT,
metadata=None,
)
def test_execute(self, mock_hook):
operator = StackdriverListNotificationChannelsOperator(
task_id=TEST_TASK_ID, filter_=TEST_FILTER)
mock_hook.return_value.list_notification_channels.return_value = [
NotificationChannel(name="test-123")
]
result = operator.execute(None)
mock_hook.return_value.list_notification_channels.assert_called_once_with(
project_id=None,
filter_=TEST_FILTER,
format_=None,
order_by=None,
page_size=None,
retry=DEFAULT,
timeout=DEFAULT,
metadata=None,
)
# Depending on the version of google-apitools installed we might receive the response either with or
# without mutation_records.
assert result in [
[{
'description': '',
'display_name': '',
'labels': {},
'name': 'test-123',
'type_': '',
'user_labels': {},
'verification_status': 0,
}],
[{
'description': '',
'display_name': '',
'labels': {},
'mutation_records': [],
'name': 'test-123',
'type_': '',
'user_labels': {},
'verification_status': 0,
}],
]
)
# [END howto_operator_gcp_stackdriver_upsert_notification_channel]
# [START howto_operator_gcp_stackdriver_enable_notification_channel]
enable_notification_channel = StackdriverEnableNotificationChannelsOperator(
task_id='enable-notification-channel', filter_='type="slack"')
# [END howto_operator_gcp_stackdriver_enable_notification_channel]
# [START howto_operator_gcp_stackdriver_disable_notification_channel]
disable_notification_channel = StackdriverDisableNotificationChannelsOperator(
task_id='disable-notification-channel',
filter_='displayName="channel1"')
# [END howto_operator_gcp_stackdriver_disable_notification_channel]
# [START howto_operator_gcp_stackdriver_list_notification_channel]
list_notification_channel = StackdriverListNotificationChannelsOperator(
task_id='list-notification-channel', filter_='type="slack"')
# [END howto_operator_gcp_stackdriver_list_notification_channel]
# [START howto_operator_gcp_stackdriver_upsert_alert_policy]
create_alert_policy = StackdriverUpsertAlertOperator(
task_id='create-alert-policies',
alerts=json.dumps(
{"policies": [TEST_ALERT_POLICY_1, TEST_ALERT_POLICY_2]}),
)
# [END howto_operator_gcp_stackdriver_upsert_alert_policy]
# [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")',
)