def __init__(self, name=NAME):
# Call to the superclass to bootstrap.
super(Application, self).__init__(name=name)
self._checker = EC2EventChecker(
boto=self.boto,
name=name,
logger=self.logger,
stats=self.stats,
)
if self.args.flowdock_token is not None:
flowdock = FlowdockListener(
flow_token=self.args.flowdock_token,
name=name,
logger=self.logger,
stats=self.stats,
)
flowdock.urgent_threshold = self.args.urgent
self._checker.add_listener(flowdock)
if self.args.jira_username is not None:
self._checker.add_listener(JiraListener(
username=self.args.jira_username,
password=self.args.jira_password,
base_url=self.args.jira_base_url,
name=name,
logger=self.logger,
stats=self.stats,
))
def setUp(self):
# Set up a mock stats. This is used to verify code execution
self._stats = MagicMock()
# Set up a mock flowdock object
self._flowdock = MagicMock()
self._flowdock_lib = MagicMock(
Chat=MagicMock(
return_value=self._flowdock
)
)
# Set up the listener to be tested with the mock flowdock object
with patch('aws_analysis_tools.ec2_events.flowdock_listener.flowdock', self._flowdock_lib):
self._listener = FlowdockListener(
flow_token = self.FLOW_TOKEN,
name = self.APP_NAME,
stats = self._stats,
)
class FlowdockListenerTest(unittest.TestCase):
APP_NAME = 'fake-unit-test-application'
FLOW_TOKEN = 'fake-flow-token'
INSTANCE_NAME = 'unit-test.krxd.net'
INSTANCE_REGION = 'test-region-1'
INSTANCE_ZONE = INSTANCE_REGION + 'a'
INSTANCE_ID = 'i-a1b2c3d4'
EVENT_CODE = 'system-maintenance'
EVENT_DESCRIPTION = 'Your instance will experience a loss of network connectivity.'
REGULAR_EVENT_COUNT = 5
URGENT_EVENT_COUT = 5
EVENT_SEPARATOR = '\n'
URGENT_EVENTS_MESSAGE_FORMAT = '@team, the following events will happen within the next {hours} hours:\n{events}'
FLOW_TAG = ['#ec2_events']
STAT_FORMAT = 'event.{region}.{event_code}'
MESSAGE_FORMAT = '{az}: {name} ({id}) - {description} ({code}) between {start_time} and {end_time}'
def setUp(self):
# Set up a mock stats. This is used to verify code execution
self._stats = MagicMock()
# Set up a mock flowdock object
self._flowdock = MagicMock()
self._flowdock_lib = MagicMock(
Chat=MagicMock(
return_value=self._flowdock
)
)
# Set up the listener to be tested with the mock flowdock object
with patch('aws_analysis_tools.ec2_events.flowdock_listener.flowdock', self._flowdock_lib):
self._listener = FlowdockListener(
flow_token = self.FLOW_TOKEN,
name = self.APP_NAME,
stats = self._stats,
)
@classmethod
def _get_event_params(cls, start_time=datetime.now()):
"""
Returns a mock instance and status event
"""
region = MagicMock()
region.name = cls.INSTANCE_REGION
instance = MagicMock(
placement=cls.INSTANCE_ZONE,
tags={'Name': cls.INSTANCE_NAME},
id=cls.INSTANCE_ID,
region=region,
)
event = MagicMock(
code=cls.EVENT_CODE,
description=cls.EVENT_DESCRIPTION,
not_before=start_time,
not_after=datetime.now()
)
return {
'instance': instance,
'event': event,
}
def test_init(self):
"""
Flowdock object is created with passed in token
"""
self._flowdock_lib.Chat.assert_called_once_with(self.FLOW_TOKEN)
def test_handle_event(self):
"""
handle_event() correctly increment the stats for the given event type and region
"""
self._listener.handle_event(**FlowdockListenerTest._get_event_params())
self._stats.incr.assert_called_once_with(self.STAT_FORMAT.format(region=self.INSTANCE_REGION, event_code=self.EVENT_CODE))
def test_handle_complete_no_events(self):
"""
handle_complete() correctly exits without any action when no event was added
"""
self._listener.handle_complete()
self.assertFalse(self._flowdock.post.called)
def test_handle_complete(self):
"""
handle_complete() correctly post a message with all events
"""
# Generate 5 fake events
regular_events = []
for i in xrange(self.REGULAR_EVENT_COUNT):
params = FlowdockListenerTest._get_event_params(start_time=datetime(9999, 12, 31))
regular_events.append(self.MESSAGE_FORMAT.format(
az=params['instance'].placement,
name=params['instance'].tags['Name'],
id=params['instance'].id,
description=params['event'].description,
code=params['event'].code,
start_time=params['event'].not_before,
end_time=params['event'].not_after,
))
self._listener.handle_event(**params)
# Generate 5 urgent events
urgent_events = []
for i in xrange(self.URGENT_EVENT_COUT):
params = FlowdockListenerTest._get_event_params()
urgent_events.append(self.MESSAGE_FORMAT.format(
az=params['instance'].placement,
name=params['instance'].tags['Name'],
id=params['instance'].id,
description=params['event'].description,
code=params['event'].code,
start_time=params['event'].not_before,
end_time=params['event'].not_after,
))
self._listener.handle_event(**params)
self._listener.handle_complete()
# Check the events are posted to Flowdock
post_calls = [
call(
self.EVENT_SEPARATOR.join(regular_events),
self.APP_NAME,
self.FLOW_TAG,
),
call(
self.URGENT_EVENTS_MESSAGE_FORMAT.format(
hours=self._listener.urgent_threshold,
events=self.EVENT_SEPARATOR.join(urgent_events),
),
self.APP_NAME,
self.FLOW_TAG,
)
]
self.assertEquals(post_calls, self._flowdock.post.call_args_list)
