def test_ensure_vlan(self, mock_set_mtu, mock_dev_exists, mock_exec):
linux_net._ensure_vlan_privileged(123,
'fake-bridge',
mac_address='fake-mac',
mtu=1500)
self.assertTrue(mock_dev_exists.called)
calls = [
mock.call('ip',
'link',
'add',
'link',
'fake-bridge',
'name',
'vlan123',
'type',
'vlan',
'id',
123,
check_exit_code=[0, 2, 254]),
mock.call('ip',
'link',
'set',
'vlan123',
'address',
'fake-mac',
check_exit_code=[0, 2, 254]),
mock.call('ip',
'link',
'set',
'vlan123',
'up',
check_exit_code=[0, 2, 254])
]
self.assertEqual(calls, mock_exec.mock_calls)
mock_set_mtu.assert_called_once_with('vlan123', 1500)
def test_ensure_vlan_invalid_mtu(self, mock_set_mtu, mock_dev_exists,
mock_exec):
linux_net._ensure_vlan_privileged(123,
'fake-bridge',
mac_address='fake-mac',
mtu=None)
self.assertFalse(mock_set_mtu.called)
def test_ensure_vlan(self, mock_set_mtu,
mock_dev_exists, mock_exec):
linux_net._ensure_vlan_privileged(123, 'fake-bridge',
mac_address='fake-mac',
mtu=1500)
self.assertTrue(mock_dev_exists.called)
calls = [mock.call('ip', 'link', 'add', 'link',
'fake-bridge', 'name', 'vlan123', 'type',
'vlan', 'id', 123,
check_exit_code=[0, 2, 254]),
mock.call('ip', 'link', 'set', 'vlan123',
'address', 'fake-mac',
check_exit_code=[0, 2, 254]),
mock.call('ip', 'link', 'set', 'vlan123', 'up',
check_exit_code=[0, 2, 254])]
self.assertEqual(calls, mock_exec.mock_calls)
mock_set_mtu.assert_called_once_with('vlan123', 1500)
def test_ensure_vlan(self, mock_set_mtu, mock_dev_exists, mock_ip_set,
mock_ip_add):
linux_net._ensure_vlan_privileged(123,
'fake-bridge',
mac_address='fake-mac',
mtu=1500)
self.assertTrue(mock_dev_exists.called)
set_calls = [
mock.call('vlan123',
address='fake-mac',
check_exit_code=[0, 2, 254]),
mock.call('vlan123', state='up', check_exit_code=[0, 2, 254])
]
mock_ip_add.assert_called_once_with('vlan123',
'vlan',
link='fake-bridge',
vlan_id=123,
check_exit_code=[0, 2, 254])
mock_ip_set.assert_has_calls(set_calls)
mock_set_mtu.assert_called_once_with('vlan123', 1500)
def test_ensure_vlan_invalid_mtu(self, mock_set_mtu,
mock_dev_exists, mock_exec):
linux_net._ensure_vlan_privileged(123, 'fake-bridge',
mac_address='fake-mac', mtu=None)
self.assertFalse(mock_set_mtu.called)
