def test_find_collectors(collectors_mock):
collector = Collectors(CLIENT)
collectors_mock.return_value = [
{'name': 'Found', 'id': 100}
]
find_collector = collector.find('found')
assert 'id' in find_collector
def test_delete_collector_notok(requests_delete):
collector = Collectors(CLIENT)
requests_delete.return_value.json.side_effect = ValueError
assert 'message' in collector.delete(1)
def test_delete_collector_ok(requests_delete):
collector = Collectors(CLIENT)
requests_delete.return_value.json.return_value = {}
assert collector.delete(1) == {}
def test_info(requests_get):
collector = Collectors(CLIENT)
requests_get.return_value.json.return_value = {}
assert collector.info(1) == {}
def test_get_id():
collector = Collectors(CLIENT)
assert collector.get_id() is None
def test_find_no_collectors(collectors_mock):
collector = Collectors(CLIENT)
find_collector = collector.find('missing_collector')
assert 'status' in find_collector
def test_init_auth_and_url_exceptions():
auth_mock = MagicMock(auth=('u', 'p'))
auth_mock.get_url.side_effect = AttributeError
auth_mock.get_auth.side_effect = AttributeError
collector = Collectors(auth=auth_mock)
assert collector.url.startswith('https://api.sumologic.com/api')
def test_get_collectors(requests_get):
collector = Collectors(CLIENT)
requests_get.return_value.json.return_value = {}
assert collector.get_collectors() == {}
def test_delete_collector_ok(requests_delete):
collector = Collectors(CLIENT)
requests_delete.return_value.json.return_value = {}
assert collector.delete(1) == {}
def test_delete_collector_notok(requests_delete):
collector = Collectors(CLIENT)
requests_delete.return_value.json.side_effect = ValueError
assert 'message' in collector.delete(1)
def test_info(requests_get):
collector = Collectors(CLIENT)
requests_get.return_value.json.return_value = {}
assert collector.info(1) == {}
def test_get_id():
collector = Collectors(CLIENT)
assert collector.get_id() is None
def test_find_collectors(collectors_mock):
collector = Collectors(CLIENT)
collectors_mock.return_value = [{'name': 'Found', 'id': 100}]
find_collector = collector.find('found')
assert 'id' in find_collector
def test_find_no_collectors(collectors_mock):
collector = Collectors(CLIENT)
find_collector = collector.find('missing_collector')
assert 'status' in find_collector
def test_get_collectors(requests_get):
collector = Collectors(CLIENT)
requests_get.return_value.json.return_value = {}
assert collector.get_collectors() == {}