openstack
/
deb-ryu
Code Issues Proposed changes

utils/test_bgp: Add unit tests for L2VPN Flow Spec 

Signed-off-by: Satoshi Fujimoto <satoshi.fujimoto7@gmail.com>
Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>
This commit is contained in:
Satoshi Fujimoto 2017-05-09 16:09:57 +09:00 committed by FUJITA Tomonori
parent e82d26b1b5
commit 12b809f5f7
1 changed files with 79 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
ryu/tests/unit/services/protocols/bgp/utils/test_bgp.py
View File

@ -24,11 +24,14 @@ from ryu.lib.packet.bgp import (
BGPFlowSpecTrafficActionCommunity,
BGPFlowSpecRedirectCommunity,
BGPFlowSpecTrafficMarkingCommunity,
BGPFlowSpecVlanActionCommunity,
BGPFlowSpecTPIDActionCommunity,
)
from ryu.services.protocols.bgp.core import BgpCoreError
from ryu.services.protocols.bgp.utils.bgp import create_v4flowspec_actions
from ryu.services.protocols.bgp.utils.bgp import create_v6flowspec_actions
from ryu.services.protocols.bgp.utils.bgp import create_l2vpnflowspec_actions
LOG = logging.getLogger(__name__)
@ -130,3 +133,79 @@ class Test_Utils_BGP(unittest.TestCase):
}
expected_communities = []
self._test_create_v6flowspec_actions(actions, expected_communities)
def _test_create_l2vpnflowspec_actions(self, actions, expected_communities):
communities = create_l2vpnflowspec_actions(actions)
expected_communities.sort(key=lambda x: x.subtype)
communities.sort(key=lambda x: x.subtype)
eq_(str(expected_communities), str(communities))
def test_create_l2vpnflowspec_actions_all_actions(self):
actions = {
'traffic_rate': {
'as_number': 0,
'rate_info': 100.0,
},
'traffic_action': {
'action': 3,
},
'redirect': {
'as_number': 10,
'local_administrator': 10,
},
'traffic_marking': {
'dscp': 24,
},
'vlan_action': {
'actions_1': (BGPFlowSpecVlanActionCommunity.POP |
BGPFlowSpecVlanActionCommunity.SWAP),
'vlan_1': 3000,
'cos_1': 3,
'actions_2': BGPFlowSpecVlanActionCommunity.PUSH,
'vlan_2': 4000,
'cos_2': 2,
},
'tpid_action': {
'actions': (BGPFlowSpecTPIDActionCommunity.TI |
BGPFlowSpecTPIDActionCommunity.TO),
'tpid_1': 5,
'tpid_2': 6,
}
}
expected_communities = [
BGPFlowSpecTrafficRateCommunity(as_number=0, rate_info=100.0),
BGPFlowSpecTrafficActionCommunity(action=3),
BGPFlowSpecRedirectCommunity(as_number=10, local_administrator=10),
BGPFlowSpecTrafficMarkingCommunity(dscp=24),
BGPFlowSpecVlanActionCommunity(
actions_1=(BGPFlowSpecVlanActionCommunity.POP |
BGPFlowSpecVlanActionCommunity.SWAP),
vlan_1=3000,
cos_1=3,
actions_2=BGPFlowSpecVlanActionCommunity.PUSH,
vlan_2=4000,
cos_2=2,
),
BGPFlowSpecTPIDActionCommunity(
actions=(BGPFlowSpecTPIDActionCommunity.TI |
BGPFlowSpecTPIDActionCommunity.TO),
tpid_1=5,
tpid_2=6,
),
]
self._test_create_l2vpnflowspec_actions(actions, expected_communities)
def test_create_l2vpnflowspec_actions_without_actions(self):
actions = None
expected_communities = []
self._test_create_l2vpnflowspec_actions(actions, expected_communities)
@raises(ValueError)
def test_create_l2vpnflowspec_actions_not_exist_actions(self):
actions = {
'traffic_test': {
'test': 10,
},
}
expected_communities = []
self._test_create_l2vpnflowspec_actions(actions, expected_communities)