def test_delete_connected(self):
connection = Rest(device=self.device, alias='rest', via='rest')
self.assertEqual(connection.connected, False)
with patch('requests.Session') as req:
resp = Response()
resp.status_code = 200
req().get.return_value = resp
req().delete.return_value = resp
connection.connect()
resp.json = MagicMock(return_value={'imdata': []})
connection.delete(dn='temp')
connection.disconnect()
self.assertEqual(connection.connected, False)
def test_delete_connected_change_expected(self):
connection = Rest(device=self.device, alias='rest', via='rest')
self.assertEqual(connection.connected, False)
with patch('requests.Session') as req:
resp = Response()
resp.status_code = 200
resp2 = Response()
resp2.status_code = 300
req().delete.return_value = resp2
req().post.side_effect = [resp, resp, resp2]
connection.connect()
resp.json = MagicMock(return_value={'imdata': []})
resp2.json = MagicMock(return_value={'imdata': []})
connection.delete(dn='temp')
self.assertEqual(connection.connected, True)
connection.disconnect()
self.assertEqual(connection.connected, False)
def test_delete_connected_wrong_status_change_expected(self):
connection = Rest(device=self.device, alias='rest', via='rest')
self.assertEqual(connection.connected, False)
with patch('requests.Session') as req:
resp = Response()
resp.status_code = 200
resp2 = Response()
resp2.status_code = 400
req().post.side_effect = [resp, resp]
req().request.return_value = resp2
connection.connect()
with self.assertRaises(RequestException):
connection.delete(dn='/temp')
self.assertEqual(connection.connected, True)
connection.disconnect()
self.assertEqual(connection.connected, False)
def test_delete_connected_wrong_status(self):
connection = Rest(device=self.device, alias='rest', via='rest')
self.assertEqual(connection.connected, False)
with patch('requests.Session') as req:
resp = Response()
resp.status_code = 200
resp2 = Response()
resp2.status_code = 400
req().request.return_value = resp2
req().get.return_value = resp
req().post.return_value = resp
connection.connect()
resp.json = MagicMock(return_value={'imdata': []})
resp2.json = MagicMock(return_value={'imdata': []})
with self.assertRaises(RequestException):
connection.delete(dn='temp')
self.assertEqual(connection.connected, True)
connection.disconnect()
self.assertEqual(connection.connected, False)
def test_delete_not_connected(self):
connection = Rest(device=self.device, alias='rest', via='rest')
with self.assertRaises(Exception):
connection.delete(dn='temp')
