def test_filter_group_alerts(import_alerts, import_groups):
assert alerts.list_groups(limit=1)[0][0] == IMPORT_GROUP_ALERTS[0]
assert alerts.list_groups(limit=1, page=2)[0][0] == IMPORT_GROUP_ALERTS[1]
assert alerts.list_groups(limit=2, page=1)[0][1] == IMPORT_GROUP_ALERTS[1]
assert alerts.list_groups(sort='-name')[0][0] == IMPORT_GROUP_ALERTS[1]
assert alerts.list_groups(search='last')[0][0] == IMPORT_GROUP_ALERTS[1]
def test_remove_from_group(import_groups):
alerts.remove_from_group('first_two', ['test_text_2'])
group_alerts = [
group for group in alerts.list_groups()[0]
if group['name'] == 'first_two'
][0]['alerts']
expected = ['test_text_1']
assert expected == group_alerts
def test_remove_alert(import_groups):
alerts.remove_alert('test_text_2')
all_alerts = [alert['name'] for alert in alerts.list_alerts()[0]]
assert 'test_text_2' not in all_alerts
first_two_group_alerts = [
group for group in alerts.list_groups()[0]
if group['name'] == 'first_two'
][0]['alerts']
assert 'test_text_2' not in first_two_group_alerts
def list_groups_get():
if request.method == 'GET':
sort = request.args.get('sort')
page = request.args.get('page')
limit = request.args.get('limit')
search = request.args.get('search')
try:
list_groups, page_metadata = alerts.list_groups(sort=sort, page=page, limit=limit, search=search)
except Exception as e:
raise InvalidUsage(str(e))
return jsonify({'groups': list_groups, 'page_metadata': page_metadata})
else:
data = request.get_json()
group_data = {
'groups': data.get('groups', None),
}
try:
alerts.import_groups(**group_data)
except Exception as e:
raise InvalidUsage(str(e))
return jsonify({'message': 'Groups imported'})
def test_remove_group(import_groups):
alerts.remove_group('last_two')
assert 'last_two' not in alerts.list_groups()[0]
def test_import_export_groups(import_groups):
assert alerts.list_groups()[0] == IMPORT_GROUP_ALERTS