def test_b_initrest(self):
m = mock.MagicMock()
with mock.patch.object(Session, 'post', return_value=m):
m.raise_for_status.return_value = 201
m.json.return_value = {'key': '1&2F28CA9FC1EA0B8EAB80E9D8FD'}
kwargs = ACCESS_INFO
rc = RestClient(**kwargs)
RestHandler(rc)
def test_b_initrest(self):
m = mock.MagicMock()
with mock.patch.object(Session, 'post', return_value=m):
m.raise_for_status.return_value = 201
m.json.return_value = {'key': '1&2F28CA9FC1EA0B8EAB80E9D8FD'}
kwargs = ACCESS_INFO
with self.assertRaises(Exception) as exc:
rc = RestClient(**kwargs)
rh = RestHandler(rc)
rh.login()
self.assertIn('Bad response from server', str(exc.exception))
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.rest_client = RestClient(**kwargs)
self.rest_client.verify = kwargs.get('verify', False)
self.rest_handler = rest_handler.RestHandler(self.rest_client)
self.rest_handler.login()
self.ssh_handler = ssh_handler.SSHHandler(**kwargs)
self.version = self.ssh_handler.login(context)
self.comhandler = component_handler.ComponentHandler(
rest_handler=self.rest_handler, ssh_handler=self.ssh_handler)
self.alert_handler = alert_handler.AlertHandler(
rest_handler=self.rest_handler, ssh_handler=self.ssh_handler)