def test_check_mirror_config():
faucet_conf_str = """
dps:
s1:
interfaces:
1:
output_only: true
mirror: [2]
2:
native_vlan: 100
3:
native_vlan: 100
"""
faucet_conf = yaml_load(faucet_conf_str)
with tempfile.TemporaryDirectory() as tmpdir:
faucetconfgetsetter_cl = FaucetLocalConfGetSetter
faucetconfgetsetter_cl.DEFAULT_CONFIG_FILE = os.path.join(
tmpdir, 'faucet.yaml')
faucet = _get_proxy(faucetconfgetsetter_cl=faucetconfgetsetter_cl,
mirror_ports={'s1': 1},
proxy_mirror_ports={'sx': ['s1', 99]})
faucet.frpc.faucet_conf = faucet_conf
port = faucet.mirror_switch_port('s1')
faucet.frpc.write_faucet_conf()
assert port == 1
def test_proxy_mirror_config():
faucet_conf_str = """
dps:
s1:
interfaces:
1:
output_only: true
2:
native_vlan: 100
3:
native_vlan: 100
99:
native_vlan: 100
sx:
interfaces:
1:
native_vlan: 100
"""
faucet_conf = yaml_load(faucet_conf_str)
with tempfile.TemporaryDirectory() as tmpdir:
faucetconfgetsetter_cl = FaucetLocalConfGetSetter
faucetconfgetsetter_cl.DEFAULT_CONFIG_FILE = os.path.join(
tmpdir, 'faucet.yaml')
faucet = _get_proxy(faucetconfgetsetter_cl=faucetconfgetsetter_cl,
mirror_ports={'s1': 1},
proxy_mirror_ports={'sx': ['s1', 99]})
# returns s1:99, not sx.
faucet.frpc.faucet_conf = faucet_conf
assert faucet.proxy_mirror_port('sx', 1) == ('s1', 99)
def test_set_mirror_config():
faucet_conf_str = """
dps:
s1:
interfaces:
1:
output_only: true
mirror: [2]
2:
native_vlan: 100
3:
native_vlan: 100
"""
def mirrors(faucet):
faucet_conf = faucet.frpc.faucet_conf
switch_conf = faucet_conf['dps']['s1']
mirror_interface_conf = switch_conf['interfaces'][1]
return mirror_interface_conf.get('mirror', None)
with tempfile.TemporaryDirectory() as tmpdir:
faucetconfgetsetter_cl = FaucetLocalConfGetSetter
faucetconfgetsetter_cl.DEFAULT_CONFIG_FILE = os.path.join(
tmpdir, 'faucet.yaml')
faucet = _get_proxy(faucetconfgetsetter_cl=faucetconfgetsetter_cl,
mirror_ports={'s1': 1},
proxy_mirror_ports={'sx': ['s1', 99]})
faucet.frpc.faucet_conf = yaml_load(faucet_conf_str)
assert mirrors(faucet) == [2]
faucet.frpc.mirror_port('s1', 1, 3)
assert mirrors(faucet) == [2, 3]
faucet.frpc.mirror_port('s1', 1, 2)
assert mirrors(faucet) == [2, 3]
faucet.frpc.clear_mirror_port('s1', 1)
assert mirrors(faucet) is None
def parse_volos_cfg(self, volos_cfg_file):
cfg = None
container_cfg = None
if os.path.exists(volos_cfg_file):
try:
with open(volos_cfg_file, 'r') as f:
cfg = yaml_load(f)
except Exception as e: # pragma: no cover
self.logger.warning(
'Volos configuration could not be loaded, disabling Volos')
self.logger.error(
'Failed to load Volos config with error: {0}'.format(str(e)))
self.enabled = False
container_cfg = []
if cfg:
for repo in cfg:
item = {}
for name in cfg[repo]:
item['repo'] = repo
item['name'] = name
item['branch'] = cfg[repo][name]['branch']
item['ports'] = []
for port in cfg[repo][name]['ports']:
if port:
cfg_port = port.get('port', None)
if cfg_port:
mapping = cfg_port['mapping']
protocol = cfg_port['protocol']
cfg_p = {
'proto': protocol,
'proto_id': PROTOCOL_MAP[protocol],
'host': mapping[:mapping.index(':')],
'dest': mapping[mapping.index(':'):]
}
item['ports'].append(cfg_p)
container_cfg.append(item)
else:
self.enabled = False
else:
self.logger.debug(
'Volos configuration could not be found. disabling volos')
self.enabled = False
return container_cfg
def test_config():
"""
Tests Config Operations
"""
def check_config(obj, endpoints):
for endpoint in endpoints:
obj.mac_table[endpoint.endpoint_data['mac']] = [
endpoint.endpoint_data
]
obj.mirror_mac('00:00:00:00:00:00', 't1-1', 1)
obj.mirror_mac('00:00:00:00:00:00', 0x1, 2)
obj.mirror_mac('00:00:00:00:00:00', 't1-1', 2)
obj.mirror_mac('00:00:00:00:00:00', 't1-1', 3)
obj.mirror_mac('00:00:00:00:00:00', 't2-1', 5)
obj.mirror_mac('00:00:00:00:00:00', 'bad', 6)
obj.unmirror_mac('00:00:00:00:00:01', None, None)
obj.unmirror_mac('00:00:00:00:00:01', 't1-1', 1)
obj.unmirror_mac('00:00:00:00:00:01', 't1-1', 3)
obj.mirror_mac('00:00:00:00:00:01', 't1-1', 3)
obj.unmirror_mac('00:00:00:00:00:01', 't1-1', 3)
obj.update_acls()
obj.update_acls(endpoints=endpoints,
rules_file=os.path.join(os.getcwd(),
'config/rules.yaml'))
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 't1-1',
'port': '1',
'ipv4': '0.0.0.0',
'ipv6': '1212::1'
}
endpoint.metadata = {
'mac_addresses': {
'00:00:00:00:00:00': {
'1551805502.0': {
'labels': ['developer workstation']
}
}
},
'ipv4_addresses': {
'0.0.0.0': {
'os': 'windows'
}
},
'ipv6_addresses': {
'1212::1': {
'os': 'windows'
}
}
}
endpoint2 = endpoint_factory('foo')
endpoint2.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:01',
'segment': 't1-1',
'port': '3',
'ipv4': '0.0.0.0',
'ipv6': '1212::1'
}
endpoint2.metadata = {
'mac_addresses': {
'00:00:00:00:00:01': {
'1551805502.0': {
'labels': ['developer workstation']
}
}
},
'ipv4_addresses': {
'0.0.0.0': {
'os': 'windows'
}
},
'ipv6_addresses': {
'1212::1': {
'os': 'windows'
}
}
}
endpoints = [endpoint, endpoint2]
with tempfile.TemporaryDirectory() as tmpdir:
faucetconfgetsetter_cl = FaucetLocalConfGetSetter
faucetconfgetsetter_cl.DEFAULT_CONFIG_FILE = os.path.join(
tmpdir, 'faucet.yaml')
shutil.copy(SAMPLE_CONFIG, faucetconfgetsetter_cl.DEFAULT_CONFIG_FILE)
parser = _get_proxy(faucetconfgetsetter_cl=faucetconfgetsetter_cl,
mirror_ports={'t1-1': 2},
proxy_mirror_ports={'sx': ['s1', 99]})
parser.frpc.faucet_conf = yaml_load(
faucetconfgetsetter_cl.DEFAULT_CONFIG_FILE)
parser2 = _get_proxy(faucetconfgetsetter_cl=faucetconfgetsetter_cl)
parser2.frpc.faucet_conf = yaml_load(
faucetconfgetsetter_cl.DEFAULT_CONFIG_FILE)
config = Config().get_config()
proxy = _get_proxy(faucetconfgetsetter_cl=faucetconfgetsetter_cl,
config=config)
check_config(parser, endpoints)
check_config(parser2, endpoints)
check_config(proxy, endpoints)
def test_stack_default_config():
faucet_conf_str = """
dps:
s1:
stack:
priority: 1
dp_id: 0x1
interfaces:
1:
output_only: true
2:
native_vlan: 100
3:
native_vlan: 100
4:
stack:
dp: s2
port: 4
s2:
dp_id: 0x2
interfaces:
1:
output_only: true
2:
native_vlan: 100
3:
native_vlan: 100
4:
stack:
dp: s1
port: 4
acls:
existing_acl:
- rule:
actions:
allow: 1
"""
new_faucet_conf_str = """
dps:
s1:
stack:
priority: 1
dp_id: 0x1
arp_neighbor_timeout: 123
timeout: 247
interfaces:
1:
output_only: true
description: Poseidon local mirror
2:
native_vlan: 100
3:
native_vlan: 100
4:
stack:
dp: s2
port: 4
s2:
dp_id: 0x2
arp_neighbor_timeout: 123
timeout: 247
interfaces:
1:
description: Poseidon remote mirror (loopback plug)
acls_in: [poseidon_tunnel]
coprocessor:
strategy: vlan_vid
2:
native_vlan: 100
3:
native_vlan: 100
4:
stack:
dp: s1
port: 4
acls:
existing_acl:
- rule:
actions:
allow: 1
poseidon_tunnel:
- rule:
vlan_vid: 999
actions:
allow: 0
- rule:
actions:
allow: 0
output:
tunnel:
type: vlan
tunnel_id: 999
dp: s1
port: 1
"""
orig_faucet_conf = yaml_load(faucet_conf_str)
test_faucet_conf = yaml_load(new_faucet_conf_str)
with tempfile.TemporaryDirectory() as tmpdir:
faucetconfgetsetter_cl = FaucetLocalConfGetSetter
faucetconfgetsetter_cl.DEFAULT_CONFIG_FILE = os.path.join(
tmpdir, 'faucet.yaml')
faucet = _get_proxy(faucetconfgetsetter_cl=faucetconfgetsetter_cl,
mirror_ports={
's1': 1,
's2': 1
},
proxy_mirror_ports={'sx': ['s1', 99]},
tunnel_vlan=999,
tunnel_name='poseidon_tunnel')
faucet.reinvestigation_frequency = 123
faucet.frpc.faucet_conf = orig_faucet_conf
faucet.frpc.write_faucet_conf()
faucet._set_default_switch_conf()
faucet.frpc.read_faucet_conf(config_file=None)
assert faucet.frpc.faucet_conf['dps']['s1'] == test_faucet_conf['dps'][
's1']
assert faucet.frpc.faucet_conf['dps']['s2'] == test_faucet_conf['dps'][
's2']
assert faucet.frpc.faucet_conf['acls'] == test_faucet_conf['acls']