class TestEOSVlan(unittest.TestCase):
@mock.patch("pyeapi.api.vlans.Vlans", autospec=True)
@mock.patch("pyeapi.client.Node", autospec=True)
@mock.patch("pyntc.devices.eos_device.EOSDevice", autospec=True)
def setUp(self, mock_eos_device, mock_native_node, mock_native_vlans):
mock_eos_instance = mock_eos_device.return_value
mock_eos_instance.native = mock_native_node
mock_native_vlans.get.side_effect = get
mock_native_vlans.getall.side_effect = getall
self.vlans = EOSVlans(mock_eos_instance)
self.vlans.native_vlans = mock_native_vlans
def test_get(self):
result = self.vlans.get("10")
self.assertEqual(result.get("name"), "VLAN0010")
self.assertEqual(result.get("state"), "active")
self.assertEqual(result.get("id"), "10")
def test_bad_get(self):
with self.assertRaises(VlanNotInRangeError):
self.vlans.get("6000")
def test_get_list(self):
result = self.vlans.get_list()
self.assertEqual(result, ["1", "10"])
def test_remove(self):
self.vlans.remove("10")
self.vlans.native_vlans.delete.assert_called_with("10")
class TestEOSVlan(unittest.TestCase):
@mock.patch('pyeapi.api.vlans.Vlans', autospec=True)
@mock.patch('pyeapi.client.Node', autospec=True)
@mock.patch('pyntc.devices.eos_device.EOSDevice', autospec=True)
def setUp(self, mock_eos_device, mock_native_node, mock_native_vlans):
mock_eos_instance = mock_eos_device.return_value
mock_eos_instance.native = mock_native_node
mock_native_vlans.get.side_effect = get
mock_native_vlans.getall.side_effect = getall
self.vlans = EOSVlans(mock_eos_instance)
self.vlans.native_vlans = mock_native_vlans
def test_get(self):
result = self.vlans.get('10')
self.assertEqual(result.get('name'), 'VLAN0010')
self.assertEqual(result.get('state'), 'active')
self.assertEqual(result.get('id'), '10')
def test_bad_get(self):
with self.assertRaises(VlanNotInRangeError):
self.vlans.get('6000')
def test_get_list(self):
result = self.vlans.get_list()
self.assertEqual(result, ['1', '10'])
def test_remove(self):
self.vlans.remove('10')
self.vlans.native_vlans.delete.assert_called_with('10')