444 lines
20 KiB
Python
444 lines
20 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2014 Red Hat, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import os.path
|
|
import random
|
|
import sys
|
|
import yaml
|
|
|
|
import os_net_config
|
|
from os_net_config import cli
|
|
from os_net_config import sriov_config
|
|
from os_net_config.tests import base
|
|
from os_net_config import utils
|
|
import six
|
|
|
|
|
|
REALPATH = os.path.dirname(os.path.realpath(__file__))
|
|
SAMPLE_BASE = os.path.join(REALPATH, '../../', 'etc',
|
|
'os-net-config', 'samples')
|
|
|
|
|
|
class TestCli(base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(TestCli, self).setUp()
|
|
rand = str(int(random.random() * 100000))
|
|
sriov_config._SRIOV_CONFIG_FILE = '/tmp/sriov_config_' + rand + '.yaml'
|
|
|
|
def stub_is_ovs_installed():
|
|
return True
|
|
self.stub_out('os_net_config.utils.is_ovs_installed',
|
|
stub_is_ovs_installed)
|
|
|
|
def tearDown(self):
|
|
super(TestCli, self).tearDown()
|
|
if os.path.isfile(sriov_config._SRIOV_CONFIG_FILE):
|
|
os.remove(sriov_config._SRIOV_CONFIG_FILE)
|
|
|
|
def run_cli(self, argstr, exitcodes=(0,)):
|
|
orig = sys.stdout
|
|
orig_stderr = sys.stderr
|
|
|
|
sys.stdout = six.StringIO()
|
|
sys.stderr = six.StringIO()
|
|
ret = cli.main(argstr.split())
|
|
self.assertIn(ret, exitcodes)
|
|
|
|
stdout = sys.stdout.getvalue()
|
|
sys.stdout.close()
|
|
sys.stdout = orig
|
|
stderr = sys.stderr.getvalue()
|
|
sys.stderr.close()
|
|
sys.stderr = orig_stderr
|
|
return (stdout, stderr)
|
|
|
|
def stub_get_stored_pci_address(self, ifname, noop):
|
|
if 'eth0' in ifname:
|
|
return "0000:00:07.0"
|
|
if 'eth1' in ifname:
|
|
return "0000:00:08.0"
|
|
if 'eth2' in ifname:
|
|
return "0000:00:09.0"
|
|
if 'em3' in ifname:
|
|
return "0000:00:03.0"
|
|
if 'em1' in ifname:
|
|
return "0000:00:01.0"
|
|
|
|
def test_bond_noop_output(self):
|
|
bond_yaml = os.path.join(SAMPLE_BASE, 'bond.yaml')
|
|
bond_json = os.path.join(SAMPLE_BASE, 'bond.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % bond_yaml)
|
|
self.assertEqual('', stderr)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % bond_json)
|
|
self.assertEqual('', stderr)
|
|
sanity_devices = ['DEVICE=br-ctlplane',
|
|
'DEVICE=em2',
|
|
'DEVICE=em1',
|
|
'DEVICE=bond1',
|
|
'DEVICETYPE=ovs']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
|
|
def test_ivs_noop_output(self):
|
|
ivs_yaml = os.path.join(SAMPLE_BASE, 'ivs.yaml')
|
|
ivs_json = os.path.join(SAMPLE_BASE, 'ivs.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % ivs_yaml)
|
|
self.assertEqual('', stderr)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % ivs_json)
|
|
self.assertEqual('', stderr)
|
|
sanity_devices = ['DEVICE=nic2',
|
|
'DEVICE=nic3',
|
|
'DEVICE=api201',
|
|
'DEVICE=storage202',
|
|
'DEVICETYPE=ivs']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
|
|
def test_bridge_noop_output(self):
|
|
bridge_yaml = os.path.join(SAMPLE_BASE, 'bridge_dhcp.yaml')
|
|
bridge_json = os.path.join(SAMPLE_BASE, 'bridge_dhcp.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=eni --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % bridge_yaml)
|
|
self.assertEqual('', stderr)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=eni --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % bridge_json)
|
|
self.assertEqual('', stderr)
|
|
sanity_devices = ['iface br-ctlplane inet dhcp',
|
|
'iface em1',
|
|
'ovs_type OVSBridge']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
|
|
def test_vlan_noop_output(self):
|
|
vlan_yaml = os.path.join(SAMPLE_BASE, 'bridge_vlan.yaml')
|
|
vlan_json = os.path.join(SAMPLE_BASE, 'bridge_vlan.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % vlan_yaml)
|
|
self.assertEqual('', stderr)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % vlan_json)
|
|
self.assertEqual('', stderr)
|
|
sanity_devices = ['DEVICE=br-ctlplane',
|
|
'DEVICE=em1',
|
|
'DEVICE=vlan16',
|
|
'DEVICETYPE=ovs']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
|
|
def test_interface_noop_output(self):
|
|
interface_yaml = os.path.join(SAMPLE_BASE, 'interface.yaml')
|
|
interface_json = os.path.join(SAMPLE_BASE, 'interface.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % interface_yaml)
|
|
self.assertEqual('', stderr)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % interface_json)
|
|
self.assertEqual('', stderr)
|
|
sanity_devices = ['DEVICE=em1',
|
|
'BOOTPROTO=static',
|
|
'IPADDR=192.0.2.1']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
|
|
def test_bridge_noop_rootfs(self):
|
|
for provider in ('ifcfg', 'eni'):
|
|
bond_yaml = os.path.join(SAMPLE_BASE, 'bridge_dhcp.yaml')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=%s --noop '
|
|
'--exit-on-validation-errors '
|
|
'--root-dir=/rootfs '
|
|
'-c %s' % (provider, bond_yaml))
|
|
self.assertEqual('', stderr)
|
|
self.assertIn('File: /rootfs/', stdout_yaml)
|
|
|
|
def test_interface_noop_detailed_exit_codes(self):
|
|
interface_yaml = os.path.join(SAMPLE_BASE, 'interface.yaml')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s --detailed-exit-codes'
|
|
% interface_yaml, exitcodes=(2,))
|
|
|
|
def test_interface_noop_detailed_exit_codes_no_changes(self):
|
|
interface_yaml = os.path.join(SAMPLE_BASE, 'interface.yaml')
|
|
|
|
class TestImpl(os_net_config.NetConfig):
|
|
|
|
def add_interface(self, interface):
|
|
pass
|
|
|
|
def apply(self, cleanup=False, activate=True):
|
|
# this fake implementation returns no changes
|
|
return {}
|
|
|
|
self.stub_out('os_net_config.impl_ifcfg.IfcfgNetConfig', TestImpl)
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s --detailed-exit-codes'
|
|
% interface_yaml, exitcodes=(0,))
|
|
|
|
def test_sriov_noop_output(self):
|
|
def test_get_vf_devname(device, vfid):
|
|
return device + '_' + str(vfid)
|
|
|
|
def test_get_pci_address(ifname, noop):
|
|
return '0000:79:10.2'
|
|
|
|
def test_interface_mac(name):
|
|
return 'AA:BB:CC:DD:EE:FF'
|
|
|
|
self.stub_out('os_net_config.utils.get_vf_devname',
|
|
test_get_vf_devname)
|
|
self.stub_out('os_net_config.utils.get_pci_address',
|
|
test_get_pci_address)
|
|
self.stub_out('os_net_config.utils.interface_mac',
|
|
test_interface_mac)
|
|
ivs_yaml = os.path.join(SAMPLE_BASE, 'sriov_pf.yaml')
|
|
ivs_json = os.path.join(SAMPLE_BASE, 'sriov_pf.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % ivs_yaml)
|
|
self.assertEqual('', stderr)
|
|
contents = utils.get_file_data(sriov_config._SRIOV_CONFIG_FILE)
|
|
sriov_config_yaml = yaml.safe_load(contents)
|
|
os.remove(sriov_config._SRIOV_CONFIG_FILE)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % ivs_json)
|
|
self.assertEqual('', stderr)
|
|
contents = utils.get_file_data(sriov_config._SRIOV_CONFIG_FILE)
|
|
sriov_config_json = yaml.safe_load(contents)
|
|
sanity_devices = ['DEVICE=p2p1',
|
|
'DEVICE=p2p1_5',
|
|
'DEVICE=p2p1_1',
|
|
'DEVICE=br-vfs',
|
|
'DEVICE=br-bond',
|
|
'TYPE=OVSBridge']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
self.assertItemsEqual(sriov_config_yaml, sriov_config_json)
|
|
|
|
def test_sriov_vf_with_dpdk_noop_output(self):
|
|
def test_get_vf_devname(device, vfid):
|
|
return device + '_' + str(vfid)
|
|
|
|
def test_get_pci_address(ifname, noop):
|
|
return '0000:79:10.2'
|
|
|
|
self.stub_out('os_net_config.utils.get_vf_devname',
|
|
test_get_vf_devname)
|
|
self.stub_out('os_net_config.utils.get_pci_address',
|
|
test_get_pci_address)
|
|
ivs_yaml = os.path.join(SAMPLE_BASE, 'sriov_pf_ovs_dpdk.yaml')
|
|
ivs_json = os.path.join(SAMPLE_BASE, 'sriov_pf_ovs_dpdk.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % ivs_yaml)
|
|
self.assertEqual('', stderr)
|
|
contents = utils.get_file_data(sriov_config._SRIOV_CONFIG_FILE)
|
|
sriov_config_yaml = yaml.safe_load(contents)
|
|
os.remove(sriov_config._SRIOV_CONFIG_FILE)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % ivs_json)
|
|
self.assertEqual('', stderr)
|
|
contents = utils.get_file_data(sriov_config._SRIOV_CONFIG_FILE)
|
|
sriov_config_json = yaml.safe_load(contents)
|
|
sanity_devices = ['DEVICE=p2p1',
|
|
'DEVICE=p2p1_5',
|
|
'DEVICE=br-vfs',
|
|
'TYPE=OVSUserBridge',
|
|
'DEVICE=dpdk0',
|
|
'TYPE=OVSDPDKPort']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
self.assertItemsEqual(sriov_config_yaml, sriov_config_json)
|
|
|
|
def test_ovs_dpdk_bond_noop_output(self):
|
|
ivs_yaml = os.path.join(SAMPLE_BASE, 'ovs_dpdk_bond.yaml')
|
|
ivs_json = os.path.join(SAMPLE_BASE, 'ovs_dpdk_bond.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % ivs_yaml)
|
|
self.assertEqual('', stderr)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % ivs_json)
|
|
self.assertEqual('', stderr)
|
|
sanity_devices = ['DEVICE=br-link',
|
|
'TYPE=OVSUserBridge',
|
|
'DEVICE=dpdkbond0',
|
|
'TYPE=OVSDPDKBond']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
|
|
def test_nfvswitch_noop_output(self):
|
|
nfvswitch_yaml = os.path.join(SAMPLE_BASE, 'nfvswitch.yaml')
|
|
nfvswitch_json = os.path.join(SAMPLE_BASE, 'nfvswitch.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % nfvswitch_yaml)
|
|
self.assertEqual('', stderr)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % nfvswitch_json)
|
|
self.assertEqual('', stderr)
|
|
sanity_devices = ['DEVICE=nic2',
|
|
'DEVICE=nic3',
|
|
'DEVICE=api201',
|
|
'DEVICE=storage202',
|
|
'DEVICETYPE=nfvswitch']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
|
|
def test_ovs_dpdk_noop_output(self):
|
|
ivs_yaml = os.path.join(SAMPLE_BASE, 'ovs_dpdk.yaml')
|
|
ivs_json = os.path.join(SAMPLE_BASE, 'ovs_dpdk.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % ivs_yaml)
|
|
self.assertEqual('', stderr)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % ivs_json)
|
|
self.assertEqual('', stderr)
|
|
sanity_devices = ['DEVICE=br-link',
|
|
'TYPE=OVSUserBridge',
|
|
'DEVICE=dpdk0',
|
|
'TYPE=OVSDPDKPort']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
|
|
def test_nic_mapping_report_output(self):
|
|
mapping_report = os.path.join(SAMPLE_BASE, 'mapping_report.yaml')
|
|
|
|
def dummy_mapped_nics(nic_mapping=None):
|
|
return nic_mapping
|
|
self.stub_out('os_net_config.objects.mapped_nics', dummy_mapped_nics)
|
|
|
|
stdout, stderr = self.run_cli('ARG0 --interfaces '
|
|
'--exit-on-validation-errors '
|
|
'-m %s' % mapping_report)
|
|
self.assertEqual('', stderr)
|
|
stdout_list = yaml.safe_load(stdout)
|
|
self.assertEqual(stdout_list['nic1'], 'em1')
|
|
self.assertEqual(stdout_list['nic2'], 'em2')
|
|
self.assertEqual(stdout_list['nic3'], 'em4')
|
|
self.assertEqual(stdout_list['nic4'], 'em3')
|
|
|
|
def test_nic_mapping_report_with_explicit_interface_name(self):
|
|
mapping_report = os.path.join(SAMPLE_BASE, 'mapping_report.yaml')
|
|
|
|
def dummy_mapped_nics(nic_mapping=None):
|
|
return nic_mapping
|
|
self.stub_out('os_net_config.objects.mapped_nics', dummy_mapped_nics)
|
|
|
|
stdout, stderr = self.run_cli('ARG0 --interfaces em2 em3 '
|
|
'--exit-on-validation-errors '
|
|
'-m %s' % mapping_report)
|
|
self.assertEqual('', stderr)
|
|
stdout_list = yaml.safe_load(stdout)
|
|
self.assertNotIn('em1', stdout_list.keys())
|
|
self.assertNotIn('em1', stdout_list.values())
|
|
self.assertEqual(stdout_list['em2'], 'em2')
|
|
self.assertEqual(stdout_list['em3'], 'em3')
|
|
self.assertNotIn('em4', stdout_list.keys())
|
|
self.assertNotIn('em4', stdout_list.values())
|
|
|
|
def test_contrail_vrouter_noop_output(self):
|
|
cvi_yaml = os.path.join(SAMPLE_BASE, 'contrail_vrouter.yaml')
|
|
cvi_json = os.path.join(SAMPLE_BASE, 'contrail_vrouter.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % cvi_yaml)
|
|
self.assertEqual('', stderr)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % cvi_json)
|
|
self.assertEqual('', stderr)
|
|
sanity_devices = ['DEVICE=vhost0',
|
|
'BIND_INT=em3',
|
|
'DEVICETYPE=vhost',
|
|
'TYPE=kernel_mode']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
|
|
def test_contrail_vrouter_vlan_noop_output(self):
|
|
cvi_yaml = os.path.join(SAMPLE_BASE, 'contrail_vrouter_vlan.yaml')
|
|
cvi_json = os.path.join(SAMPLE_BASE, 'contrail_vrouter_vlan.json')
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % cvi_yaml)
|
|
self.assertEqual('', stderr)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'-c %s' % cvi_json)
|
|
self.assertEqual('', stderr)
|
|
sanity_devices = ['DEVICE=vhost0',
|
|
'BIND_INT=vlan100',
|
|
'DEVICETYPE=vhost',
|
|
'TYPE=kernel_mode']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|
|
|
|
def test_contrail_vrouter_dpdk_noop_output(self):
|
|
cvi_yaml = os.path.join(SAMPLE_BASE, 'contrail_vrouter_dpdk.yaml')
|
|
cvi_json = os.path.join(SAMPLE_BASE, 'contrail_vrouter_dpdk.json')
|
|
self.stub_out('os_net_config.utils.get_stored_pci_address',
|
|
self.stub_get_stored_pci_address)
|
|
stdout_yaml, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'--debug '
|
|
'-c %s' % cvi_yaml)
|
|
self.assertEqual('', stderr)
|
|
stdout_json, stderr = self.run_cli('ARG0 --provider=ifcfg --noop '
|
|
'--exit-on-validation-errors '
|
|
'--debug '
|
|
'-c %s' % cvi_json)
|
|
self.assertEqual('', stderr)
|
|
sanity_devices = ['DEVICE=vhost0',
|
|
'BIND_INT=0000:00:03.0',
|
|
'DEVICETYPE=vhost',
|
|
'TYPE=dpdk']
|
|
for dev in sanity_devices:
|
|
self.assertIn(dev, stdout_yaml)
|
|
self.assertEqual(stdout_yaml, stdout_json)
|