def test_resource_request(self):
logger = logging.getLogger("logger")
session = MockSession()
test_object = resource_requests.ResourceRequests(logger=logger, session=session)
response = test_object.resource_request("google")
self.assertIsNotNone(response)
self.assertEqual(response.get("resources"), [{"thing1": "data"}, {"thing2": "data"}])
def test_resource_request_requests_exception(self):
logger = logging.getLogger('logger')
session = MockSession()
test_object = resource_requests.ResourceRequests(logger=logger,
session=session)
with pytest.raises(PluginException):
response = test_object.resource_request('exception.com')
def test_init(self):
logger = logging.getLogger('logger')
session = requests.session()
test_object = resource_requests.ResourceRequests(logger=logger,
session=session)
self.assertIsNotNone(test_object)
self.assertTrue(test_object.logger.name == 'logger')
def test_resource_request_401(self):
logger = logging.getLogger('logger')
session = MockSession()
test_object = resource_requests.ResourceRequests(logger=logger,
session=session)
with pytest.raises(
PluginException,
match='InsightVM returned an error message. Unauthorized'):
response = test_object.resource_request('bad password')
def test_paged_resource_request_with_params_dict(self):
logger = logging.getLogger("logger")
session = MockSession()
test_object = resource_requests.ResourceRequests(logger=logger, session=session)
response = test_object.paged_resource_request("paged_request", params={"size": 100})
self.assertIsNotNone(response)
self.assertEqual(
response,
[{"thing1": "data"}, {"thing2": "data"}, {"thing1": "data"}, {"thing2": "data"}],
)
def test_paged_resource_request_no_params(self):
logger = logging.getLogger('logger')
session = MockSession()
test_object = resource_requests.ResourceRequests(logger=logger,
session=session)
response = test_object.paged_resource_request('paged_request')
self.assertIsNotNone(response)
self.assertEqual(response, [{
"thing1": "data"
}, {
"thing2": "data"
}, {
"thing1": "data"
}, {
"thing2": "data"
}])
def test_paged_resource_request_with_params_tuple(self):
logger = logging.getLogger('logger')
session = MockSession()
test_object = resource_requests.ResourceRequests(logger=logger,
session=session)
response = test_object.paged_resource_request('paged_request',
params=[('size', 100)])
self.assertIsNotNone(response)
self.assertEqual(response, [{
'thing1': 'data'
}, {
'thing2': 'data'
}, {
'thing1': 'data'
}, {
'thing2': 'data'
}])
def test_resource_request_requests_exception(self):
logger = logging.getLogger("logger")
session = MockSession()
test_object = resource_requests.ResourceRequests(logger=logger, session=session)
with pytest.raises(PluginException):
test_object.resource_request("exception.com")
