def test_get_contacts_by_external_ids(mock_session):
"""
Should send GET request to specific URL
"""
extids = (
'&[email protected]' +
'&[email protected]' +
'&[email protected]')
data = create_everbridge_contacts([1, 2, 3,], True)
raw = {
'message': 'OK',
'page': {
'pageSize': 100,
'start': 0,
'totalCount': 3,
'totalPageCount': 0,
'currentPageNo': 0,
'data': data
}
}
params = '?sortBy=externalId&direction=ASC&searchType=AND' + extids
expected = json.loads(json.dumps(raw))
# Set up mocks
session = create_session_mock(expected, code=200, method='GET')
mock_session.return_value = session
ever = create_everbridge_instance()
expected_url = ever.contacts_url(params)
# call get_contacts_by_external_ids
rslt = ever.get_contacts_by_external_ids(extids)
# Check if correct arguments are passed to session functions
session.get.assert_called_with(expected_url)
session.headers.update.assert_called_with(expected_header())
assert rslt == data
def test_delete_group(mock_session):
"""
Should send DELETE request to specific URL
"""
raw = {'message': 'OK'}
expected = json.loads(json.dumps(raw))
# Set up mocks
session = create_session_mock(expected, code=200, method='DELETE')
mock_session.return_value = session
ever = create_everbridge_instance()
expected_url = ever.groups_url('123')
# call delete_group
ever.delete_group(123)
# Check if correct arguments are passed to session functions
session.delete.assert_called_with(expected_url)
session.headers.update.assert_called_with(expected_header())
def test_get_group_id_by_name(mock_session):
"""
Should send GET request to specific URL
"""
raw = {'message': 'OK', 'result': {'id': 123}}
expected = json.loads(json.dumps(raw))
# Set up mocks
session = create_session_mock(expected, code=200, method='GET')
mock_session.return_value = session
ever = create_everbridge_instance()
expected_url = ever.groups_url('xxxxx?queryType=name')
# call get_group_by_name
ever.get_group_id_by_name('xxxxx')
# Check if correct arguments are passed to session functions
session.get.assert_called_with(expected_url)
session.headers.update.assert_called_with(expected_header())
def test_upsert_contacts(mock_session):
"""
Should send POST request to specific URL with data
"""
contacts = create_everbridge_contacts([1, 2, 3,], True)
raw = {'code': 100, 'message': 'OK'}
expected = json.loads(json.dumps(raw))
# Set up mocks
session = create_session_mock(expected, code=200, method='POST')
mock_session.return_value = session
ever = create_everbridge_instance()
expected_url = ever.contacts_url('batch?version=1')
# call upsert_contacts
ever.upsert_contacts(contacts)
# Check if correct arguments are passed to session functions
session.post.assert_called_with(expected_url, json=contacts)
session.headers.update.assert_called_with(expected_header())
def test_add_group(mock_session):
"""
Should send POST request to specific URL
"""
raw = {'message': 'OK'}
expected = json.loads(json.dumps(raw))
# Set up mocks
session = create_session_mock(expected, code=200, method='POST')
mock_session.return_value = session
ever = create_everbridge_instance()
expected_url = ever.groups_url('')
data = {'name': 'xxxxx', 'organizationId': '1234567', 'parentId': None}
# call add_group
ever.add_group('xxxxx')
# Check if correct arguments are passed to session functions
session.post.assert_called_with(expected_url, json=data)
session.headers.update.assert_called_with(expected_header())
def test_add_members_to_group(mock_session):
"""
Should send POST request to specific URL
"""
members = [1, 2, 3]
raw = {'message': 'OK', 'code': 100}
expected = json.loads(json.dumps(raw))
# Set up mocks
session = create_session_mock(expected, code=200, method='POST')
mock_session.return_value = session
ever = create_everbridge_instance()
expected_url = ever.groups_url('contacts?byType=id&groupId=123&idType=id')
# call add_members_to_group
ever.add_members_to_group(123, members)
# Check if correct arguments are passed to session functions
session.post.assert_called_with(expected_url, json=members)
session.headers.update.assert_called_with(expected_header())
def test_paged_group_members(mock_session):
"""
Should send GET request to specific URL
"""
raw = {'message': 'OK', 'page': {'data': []}}
expected = json.loads(json.dumps(raw))
# Set up mocks
session = create_session_mock(expected, code=200, method='GET')
mock_session.return_value = session
ever = create_everbridge_instance()
params = '?groupIds=123&pageSize=100&pageNumber=1&sortBy=externalId&direction=ASC'
expected_url = ever.contacts_url(params)
# call get_group_by_name
ever.get_paged_group_members(123)
# Check if correct arguments are passed to session functions
session.get.assert_called_with(expected_url)
session.headers.update.assert_called_with(expected_header())
def test_delete_contacts(mock_session):
"""
Should send DELETE request to specific URL with data
"""
contacts = [1, 2, 3]
raw = {'message': 'OK', 'code': 100}
expected = json.loads(json.dumps(raw))
# Set up mocks
session = create_session_mock(expected, code=200, method='DELETE')
mock_session.return_value = session
ever = create_everbridge_instance()
expected_url = ever.contacts_url('batch')
# call delete_contacts
ever.delete_contacts(contacts)
# Check if correct arguments are passed to session functions
session.delete.assert_called_with(expected_url, json=contacts)
session.headers.update.assert_called_with(expected_header())