def save_cmd_details(self, data, params): if 'deployment_id' in data: deployment_id = data['deployment_id'] item = Notifications(deployment_id, cmd_channel_id=params['channel']['id'], cmd_response_url=params['response_url'], cmd_username=params['user']['name']) item.put_item()
def post_to_slack(self): notification = Notifications(self.deployment_id()) item = notification.get_item() if item: self.post(item, msg_ts_type='slack_ts') if 'cmd_channel_id' in item: self.post(item, msg_ts_type='cmd_ts') else: self.post(msg_ts_type='slack_ts')
def test_as_dict(self): expected = { 'deployment': 'ecs-svc/9223370476922431730', 'cmd_username': '******', 'slack_ts': 'slack_ts', 'cmd_response_url': 'cmd_response_url', 'cmd_channel_id': 'cmd_channel_id', 'cmd_ts': 'cmd_ts' } n = Notifications(**expected) expected['TTL'] = int(time.time()) + int(2592000) assert n.as_dict() == expected
def update_notifications_item(self, slack_response, saved_item=None, msg_ts_type='slack_ts'): try: ts = slack_response['ts'] except KeyError: log.error('Error: Cannot get slack timestamp. Slack response:') log.error(slack_response) return if saved_item: saved_item[msg_ts_type] = ts item = Notifications(**saved_item) else: item = Notifications(self.deployment_id(), ts) item.put_item()
def test_as_dict_bare(self): n = Notifications('ecs-svc/9223370476922431730') assert n.as_dict() == { 'deployment': 'ecs-svc/9223370476922431730', 'TTL': int(time.time()) + int(2592000), }
def test_read_missing_notification(self): n = Notifications(deployment='missing') assert n.get_item() == None
def save_cmd_details(self, deployment_id, params): item = Notifications(deployment_id, cmd_channel_id=params['channel_id'], cmd_response_url=params['response_url'], cmd_username=params['user_name']) item.put_item()