summaryrefslogtreecommitdiffstats
path: root/test/test_qos.py
diff options
context:
space:
mode:
authorPaul Vinciguerra <pvinci@vinciconsulting.com>2019-05-17 09:48:15 -0400
committerDave Wallace <dwallacelf@gmail.com>2019-05-17 21:03:29 +0000
commitb7658201d958231ac7f5aeda6184d918893d2914 (patch)
tree97f053180e0e6c1a916ea37fb3494442480eb347 /test/test_qos.py
parent0d144c78d7119a0af71eda6e67f29fa56da2614d (diff)
Tests: Refactor. Remove copies of enums.
Use the enums provided from the api definitions instead. Change-Id: I43b7591df920c984e439efc4b3ec3c4121c539d8 Signed-off-by: Paul Vinciguerra <pvinci@vinciconsulting.com>
Diffstat (limited to 'test/test_qos.py')
-rw-r--r--test/test_qos.py81
1 files changed, 46 insertions, 35 deletions
diff --git a/test/test_qos.py b/test/test_qos.py
index 8025bdcd40c..d4bd0872243 100644
--- a/test/test_qos.py
+++ b/test/test_qos.py
@@ -3,7 +3,6 @@
import unittest
from framework import VppTestCase, VppTestRunner
-from vpp_papi_provider import QOS_SOURCE
from vpp_sub_interface import VppDot1QSubint
from vpp_ip import DpoProto
from vpp_ip_route import VppIpRoute, VppRoutePath, VppMplsRoute, \
@@ -15,6 +14,7 @@ from scapy.layers.l2 import Ether, Dot1Q
from scapy.layers.inet import IP, UDP
from scapy.layers.inet6 import IPv6
from scapy.contrib.mpls import MPLS
+from vpp_papi import VppEnum
NUM_PKTS = 67
@@ -22,6 +22,13 @@ NUM_PKTS = 67
class TestQOS(VppTestCase):
""" QOS Test Case """
+ # Note: Since the enums aren't created dynamically until after
+ # the papi client attaches to VPP, we put it in a property to
+ # ensure it is the value at runtime, not at module load time.
+ @property
+ def QOS_SOURCE(self):
+ return VppEnum.vl_api_qos_source_t
+
@classmethod
def setUpClass(cls):
super(TestQOS, cls).setUpClass()
@@ -110,19 +117,19 @@ class TestQOS(VppTestCase):
# Bind interface pgN to table n
#
self.vapi.qos_mark_enable_disable(self.pg1.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
1,
1)
self.vapi.qos_mark_enable_disable(self.pg2.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
2,
1)
self.vapi.qos_mark_enable_disable(self.pg3.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
3,
1)
self.vapi.qos_mark_enable_disable(self.pg4.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
4,
1)
@@ -154,7 +161,7 @@ class TestQOS(VppTestCase):
# Enable QoS recording on IP input for pg0
#
self.vapi.qos_record_enable_disable(self.pg0.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
1)
#
@@ -212,11 +219,11 @@ class TestQOS(VppTestCase):
# remove the map on pg2 and pg3, now expect an unchanged IP tos
#
self.vapi.qos_mark_enable_disable(self.pg2.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
2,
0)
self.vapi.qos_mark_enable_disable(self.pg3.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
3,
0)
self.logger.info(self.vapi.cli("sh int feat pg2"))
@@ -243,7 +250,7 @@ class TestQOS(VppTestCase):
# disable the input recording on pg0
#
self.vapi.qos_record_enable_disable(self.pg0.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
0)
#
@@ -257,11 +264,11 @@ class TestQOS(VppTestCase):
# disable the egress map on pg1 and pg4
#
self.vapi.qos_mark_enable_disable(self.pg1.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
1,
0)
self.vapi.qos_mark_enable_disable(self.pg4.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
4,
0)
@@ -331,10 +338,10 @@ class TestQOS(VppTestCase):
# on Pg1
#
self.vapi.qos_record_enable_disable(self.pg0.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
1)
self.vapi.qos_mark_enable_disable(self.pg1.sw_if_index,
- QOS_SOURCE.MPLS,
+ self.QOS_SOURCE.QOS_API_SOURCE_MPLS,
1,
1)
@@ -378,11 +385,12 @@ class TestQOS(VppTestCase):
# enable MPLS QoS recording on the input Pg0 and IP egress marking
# on Pg1
#
- self.vapi.qos_record_enable_disable(self.pg0.sw_if_index,
- QOS_SOURCE.MPLS,
- 1)
+ self.vapi.qos_record_enable_disable(
+ self.pg0.sw_if_index,
+ self.QOS_SOURCE.QOS_API_SOURCE_MPLS,
+ 1)
self.vapi.qos_mark_enable_disable(self.pg1.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
1,
1)
@@ -436,17 +444,18 @@ class TestQOS(VppTestCase):
# cleanup
#
self.vapi.qos_record_enable_disable(self.pg0.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
0)
self.vapi.qos_mark_enable_disable(self.pg1.sw_if_index,
- QOS_SOURCE.MPLS,
+ self.QOS_SOURCE.QOS_API_SOURCE_MPLS,
1,
0)
- self.vapi.qos_record_enable_disable(self.pg0.sw_if_index,
- QOS_SOURCE.MPLS,
- 0)
+ self.vapi.qos_record_enable_disable(
+ self.pg0.sw_if_index,
+ self.QOS_SOURCE.QOS_API_SOURCE_MPLS,
+ 0)
self.vapi.qos_mark_enable_disable(self.pg1.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
1,
0)
self.vapi.qos_egress_map_delete(1)
@@ -479,11 +488,12 @@ class TestQOS(VppTestCase):
#
# enable VLAN QoS recording/marking on the input Pg0 subinterface and
#
- self.vapi.qos_record_enable_disable(sub_if.sw_if_index,
- QOS_SOURCE.VLAN,
- 1)
+ self.vapi.qos_record_enable_disable(
+ sub_if.sw_if_index,
+ self.QOS_SOURCE.QOS_API_SOURCE_VLAN,
+ 1)
self.vapi.qos_mark_enable_disable(sub_if.sw_if_index,
- QOS_SOURCE.VLAN,
+ self.QOS_SOURCE.QOS_API_SOURCE_VLAN,
1,
1)
@@ -491,10 +501,10 @@ class TestQOS(VppTestCase):
# IP marking/recording on pg1
#
self.vapi.qos_record_enable_disable(self.pg1.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
1)
self.vapi.qos_mark_enable_disable(self.pg1.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
1,
1)
@@ -570,18 +580,19 @@ class TestQOS(VppTestCase):
sub_if.unconfig_ip4()
sub_if.unconfig_ip6()
- self.vapi.qos_record_enable_disable(sub_if.sw_if_index,
- QOS_SOURCE.VLAN,
- 0)
+ self.vapi.qos_record_enable_disable(
+ sub_if.sw_if_index,
+ self.QOS_SOURCE.QOS_API_SOURCE_VLAN,
+ 0)
self.vapi.qos_mark_enable_disable(sub_if.sw_if_index,
- QOS_SOURCE.VLAN,
+ self.QOS_SOURCE.QOS_API_SOURCE_VLAN,
1,
0)
self.vapi.qos_record_enable_disable(self.pg1.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
0)
self.vapi.qos_mark_enable_disable(self.pg1.sw_if_index,
- QOS_SOURCE.IP,
+ self.QOS_SOURCE.QOS_API_SOURCE_IP,
1,
0)