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)