def trigger_escalation_incident(self, parsed_message, slack_message_obj):
"""
Trigger a service escalation for the specified service
"""
# This parses the service string into two components: the service name
# and optional service message.
# e.g. website please call me --> ['website', 'please call me']
# e.g. website --> ['website']
# e.g. website plz --> ['website', 'plz']
full_service_str = parsed_message.strip()
service_str_components = full_service_str.split(' ', 1)
# Check to see if the service key exists
service_name = service_str_components[0]
service_key = self.pd_service_mappings.get(service_name)
if not service_key:
error_msg = "Could not find key for service '%s', check that the service exists in the config file!" % service_name # NOQA
self.log.error(error_msg)
yield from self.slack.send_channel_message(
slack_message_obj.channel, error_msg) # NOQA
return
slack_user = SlackUser()
yield from slack_user.retrieve_slack_user_info(self.slack,
slack_message_obj.user)
# Set the custom trigger message, if one was provided
custom_message = ""
try:
custom_message = service_str_components[1]
except IndexError:
pass
payload = self.get_trigger_payload(service_key,
service_name,
slack_user.real_name,
custom_message=custom_message)
response = yield from http_post_request(
url=
"https://events.pagerduty.com/generic/2010-04-15/create_event.json", # NOQA
payload=json.dumps(payload),
)
if not response:
error_msg = "Could not escalate incident for %s. Check the logs for details." % service_name # NOQA
self.log.error(error_msg)
yield from self.slack.send_channel_message(
slack_message_obj.channel, error_msg) # NOQA
return
success_msg = "Successfully escalated this %s incident!" % service_name
self.log.info(success_msg)
yield from self.slack.send_channel_message(slack_message_obj.channel,
success_msg)
def trigger_escalation_incident(self, parsed_message, slack_message_obj):
"""
Trigger a service escalation for the specified service
"""
# This parses the service string into two components: the service name
# and optional service message.
# e.g. website please call me --> ['website', 'please call me']
# e.g. website --> ['website']
# e.g. website plz --> ['website', 'plz']
full_service_str = parsed_message.strip()
service_str_components = full_service_str.split(' ', 1)
# Check to see if the service key exists
service_name = service_str_components[0]
service_key = self.pd_service_mappings.get(service_name)
if not service_key:
error_msg = "Could not find key for service '%s', check that the service exists in the config file!" % service_name # NOQA
self.log.error(error_msg)
yield from self.slack.send_channel_message(slack_message_obj.channel, error_msg) # NOQA
return
slack_user = SlackUser()
yield from slack_user.retrieve_slack_user_info(self.slack,
slack_message_obj.user)
# Set the custom trigger message, if one was provided
custom_message = ""
try:
custom_message = service_str_components[1]
except IndexError:
pass
payload = self.get_trigger_payload(service_key,
service_name,
slack_user.real_name,
custom_message=custom_message)
response = yield from http_post_request(
url="https://events.pagerduty.com/generic/2010-04-15/create_event.json", # NOQA
payload=json.dumps(payload),
)
if not response:
error_msg = "Could not escalate incident for %s. Check the logs for details." % service_name # NOQA
self.log.error(error_msg)
yield from self.slack.send_channel_message(slack_message_obj.channel, error_msg) # NOQA
return
success_msg = "Successfully escalated this %s incident!" % service_name
self.log.info(success_msg)
yield from self.slack.send_channel_message(slack_message_obj.channel,
success_msg)
def populate_slack_user_object(self, username): # pragma: no cover
slack_user = SlackUser()
yield from slack_user.retrieve_slack_user_info(self.slack, username)
return slack_user
class TestSlackUser(asynctest.TestCase):
def setUp(self):
patcher1 = patch(
'charlesbot.slack.slack_connection.SlackConnection.api_call'
) # NOQA
self.addCleanup(patcher1.stop)
self.mock_api_call = patcher1.start()
from charlesbot.slack.slack_connection import SlackConnection
self.slack_connection = SlackConnection()
from charlesbot.slack.slack_user import SlackUser
self.su = SlackUser()
def tearDown(self):
self.slack_connection._drop()
@asynctest.ignore_loop
def test_user_equality(self):
from charlesbot.slack.slack_user import SlackUser
user1 = SlackUser(id="SU01", name="userone", color="red")
user2 = SlackUser(id="SU02", name="usertwo", color="blue")
self.assertNotEqual(user1, user2)
user2.id = "SU01"
self.assertNotEqual(user1, user2)
user2.name = "userone"
self.assertNotEqual(user1, user2)
user2.color = "red"
self.assertEqual(user1, user2)
@asynctest.ignore_loop
def test_user_return_string(self):
self.su.id = "SU01"
self.su.name = "User One"
self.su.deleted = False
self.su.is_admin = False
self.su.has_2fa = True
user_json = json.loads(str(self.su))
self.assertEqual(user_json.get('id'), "SU01")
self.assertEqual(user_json.get('name'), "User One")
self.assertEqual(user_json.get('deleted'), False)
self.assertEqual(user_json.get('is_admin'), False)
self.assertEqual(user_json.get('has_2fa'), True)
self.assertEqual(user_json.get('is_owner'), "")
def test_empty_slack_response(self):
self.su.name = "suser"
self.mock_api_call.side_effect = ["{}"]
yield from self.su.retrieve_slack_user_info(self.slack_connection,
"fake123")
expected_call = call("users.info", user="******")
self.assertEqual(self.mock_api_call.mock_calls, [expected_call]),
self.assertEqual(self.su.name, "suser")
self.assertEqual(self.su.last_name, "")
self.assertEqual(self.su.is_bot, "")
def test_no_profile_key(self):
self.su.name = "suser"
user_info = {"ok": True, "user": {"id": "U023BECGF", "name": "bobby"}}
self.mock_api_call.side_effect = [json.dumps(user_info)]
yield from self.su.retrieve_slack_user_info(self.slack_connection,
"fake123")
expected_call = call("users.info", user="******")
self.assertEqual(self.mock_api_call.mock_calls, [expected_call]),
self.assertEqual(self.su.name, "bobby")
self.assertEqual(self.su.id, "U023BECGF")
self.assertEqual(self.su.last_name, "")
self.assertEqual(self.su.is_bot, "")
def test_with_profile_key(self):
self.su.name = "suser"
user_info = {
"ok": True,
"user": {
"id": "U023BECGF",
"name": "bobby",
"profile": {
"real_name": "Bobby Tables",
"image_24": "https://www.tables.com",
}
}
}
self.mock_api_call.side_effect = [json.dumps(user_info)]
yield from self.su.retrieve_slack_user_info(self.slack_connection,
"fake123")
expected_call = call("users.info", user="******")
self.assertEqual(self.mock_api_call.mock_calls, [expected_call]),
self.assertEqual(self.su.name, "bobby")
self.assertEqual(self.su.id, "U023BECGF")
self.assertEqual(self.su.real_name, "Bobby Tables")
self.assertEqual(self.su.image_24, "https://www.tables.com")
self.assertEqual(self.su.is_bot, "")
class TestSlackUser(asynctest.TestCase):
def setUp(self):
patcher1 = patch('charlesbot.slack.slack_connection.SlackConnection.api_call') # NOQA
self.addCleanup(patcher1.stop)
self.mock_api_call = patcher1.start()
from charlesbot.slack.slack_connection import SlackConnection
self.slack_connection = SlackConnection()
from charlesbot.slack.slack_user import SlackUser
self.su = SlackUser()
def tearDown(self):
self.slack_connection._drop()
@asynctest.ignore_loop
def test_user_equality(self):
from charlesbot.slack.slack_user import SlackUser
user1 = SlackUser(id="SU01",
name="userone",
color="red")
user2 = SlackUser(id="SU02",
name="usertwo",
color="blue")
self.assertNotEqual(user1, user2)
user2.id = "SU01"
self.assertNotEqual(user1, user2)
user2.name = "userone"
self.assertNotEqual(user1, user2)
user2.color = "red"
self.assertEqual(user1, user2)
@asynctest.ignore_loop
def test_user_return_string(self):
self.su.id = "SU01"
self.su.name = "User One"
self.su.deleted = False
self.su.is_admin = False
self.su.has_2fa = True
user_json = json.loads(str(self.su))
self.assertEqual(user_json.get('id'), "SU01")
self.assertEqual(user_json.get('name'), "User One")
self.assertEqual(user_json.get('deleted'), False)
self.assertEqual(user_json.get('is_admin'), False)
self.assertEqual(user_json.get('has_2fa'), True)
self.assertEqual(user_json.get('is_owner'), "")
def test_empty_slack_response(self):
self.su.name = "suser"
self.mock_api_call.side_effect = ["{}"]
yield from self.su.retrieve_slack_user_info(self.slack_connection,
"fake123")
expected_call = call("users.info", user="******")
self.assertEqual(self.mock_api_call.mock_calls,
[expected_call]),
self.assertEqual(self.su.name, "suser")
self.assertEqual(self.su.last_name, "")
self.assertEqual(self.su.is_bot, "")
def test_no_profile_key(self):
self.su.name = "suser"
user_info = {
"ok": True,
"user": {
"id": "U023BECGF",
"name": "bobby"
}
}
self.mock_api_call.side_effect = [json.dumps(user_info)]
yield from self.su.retrieve_slack_user_info(self.slack_connection,
"fake123")
expected_call = call("users.info", user="******")
self.assertEqual(self.mock_api_call.mock_calls,
[expected_call]),
self.assertEqual(self.su.name, "bobby")
self.assertEqual(self.su.id, "U023BECGF")
self.assertEqual(self.su.last_name, "")
self.assertEqual(self.su.is_bot, "")
def test_with_profile_key(self):
self.su.name = "suser"
user_info = {
"ok": True,
"user": {
"id": "U023BECGF",
"name": "bobby",
"profile": {
"real_name": "Bobby Tables",
"image_24": "https://www.tables.com",
}
}
}
self.mock_api_call.side_effect = [json.dumps(user_info)]
yield from self.su.retrieve_slack_user_info(self.slack_connection,
"fake123")
expected_call = call("users.info", user="******")
self.assertEqual(self.mock_api_call.mock_calls,
[expected_call]),
self.assertEqual(self.su.name, "bobby")
self.assertEqual(self.su.id, "U023BECGF")
self.assertEqual(self.su.real_name, "Bobby Tables")
self.assertEqual(self.su.image_24, "https://www.tables.com")
self.assertEqual(self.su.is_bot, "")