def test_delete_tc_policy_class_htb(self):
priv_tc_lib.add_tc_qdisc(self.device,
parent=rtnl.TC_H_ROOT,
kind='htb',
handle='1:',
namespace=self.namespace)
for classid, rates in self.CLASSES.items():
priv_tc_lib.add_tc_policy_class(self.device,
'1:',
classid,
'htb',
namespace=self.namespace,
**rates)
tc_classes = priv_tc_lib.list_tc_policy_classes(
self.device, namespace=self.namespace)
self.assertEqual(len(self.CLASSES), len(tc_classes))
for classid in self.CLASSES:
priv_tc_lib.delete_tc_policy_class(self.device,
'1:',
classid,
namespace=self.namespace)
tc_classes = priv_tc_lib.list_tc_policy_classes(
self.device, namespace=self.namespace)
for tc_class in tc_classes:
handle = tc_lib._handle_from_hex_to_string(tc_class['handle'])
self.assertIsNot(classid, handle)
tc_classes = priv_tc_lib.list_tc_policy_classes(
self.device, namespace=self.namespace)
self.assertEqual(0, len(tc_classes))
def test_add_tc_policy_class_htb(self):
priv_tc_lib.add_tc_qdisc(self.device,
parent=rtnl.TC_H_ROOT,
kind='htb',
handle='1:',
namespace=self.namespace)
for classid, rates in self.CLASSES.items():
priv_tc_lib.add_tc_policy_class(self.device,
'1:',
classid,
'htb',
namespace=self.namespace,
**rates)
tc_classes = priv_tc_lib.list_tc_policy_classes(
self.device, namespace=self.namespace)
self.assertEqual(len(self.CLASSES), len(tc_classes))
for tc_class in tc_classes:
handle = tc_lib._handle_from_hex_to_string(tc_class['handle'])
tca_options = tc_lib._get_attr(tc_class, 'TCA_OPTIONS')
tca_htb_params = tc_lib._get_attr(tca_options, 'TCA_HTB_PARMS')
self.assertEqual(self.CLASSES[handle]['rate'],
tca_htb_params['rate'])
self.assertEqual(self.CLASSES[handle]['ceil'],
tca_htb_params['ceil'])
burst = tc_lib._calc_burst(self.CLASSES[handle]['rate'],
tca_htb_params['buffer'])
self.assertEqual(self.CLASSES[handle]['burst'], burst)
def test_delete_tc_policy_class_htb(self):
priv_tc_lib.add_tc_qdisc(
self.device, parent=rtnl.TC_H_ROOT, kind='htb', handle='1:',
namespace=self.namespace)
for classid, rates in self.CLASSES.items():
priv_tc_lib.add_tc_policy_class(
self.device, '1:', classid, 'htb', namespace=self.namespace,
**rates)
tc_classes = priv_tc_lib.list_tc_policy_classes(
self.device, namespace=self.namespace)
self.assertEqual(len(self.CLASSES), len(tc_classes))
for classid in self.CLASSES:
priv_tc_lib.delete_tc_policy_class(
self.device, '1:', classid, namespace=self.namespace)
tc_classes = priv_tc_lib.list_tc_policy_classes(
self.device, namespace=self.namespace)
for tc_class in tc_classes:
handle = tc_lib._handle_from_hex_to_string(tc_class['handle'])
self.assertIsNot(classid, handle)
tc_classes = priv_tc_lib.list_tc_policy_classes(
self.device, namespace=self.namespace)
self.assertEqual(0, len(tc_classes))
def test_run(self):
test_cases = [(0x1, '0:1'),
(0x2a003f, '2a:3f'),
(0xf0000, 'f:0'),
(0xffffffff, 'ffff:ffff'),
(0x12345678, '1234:5678')]
for _in, expected in test_cases:
self.assertEqual(expected, tc_lib._handle_from_hex_to_string(_in))
def test_add_tc_policy_class_htb(self):
priv_tc_lib.add_tc_qdisc(
self.device, parent=rtnl.TC_H_ROOT, kind='htb', handle='1:',
namespace=self.namespace)
for classid, rates in self.CLASSES.items():
priv_tc_lib.add_tc_policy_class(
self.device, '1:', classid, 'htb', namespace=self.namespace,
**rates)
tc_classes = priv_tc_lib.list_tc_policy_classes(
self.device, namespace=self.namespace)
self.assertEqual(len(self.CLASSES), len(tc_classes))
for tc_class in tc_classes:
handle = tc_lib._handle_from_hex_to_string(tc_class['handle'])
tca_options = tc_lib._get_attr(tc_class, 'TCA_OPTIONS')
tca_htb_params = tc_lib._get_attr(tca_options, 'TCA_HTB_PARMS')
self.assertEqual(self.CLASSES[handle]['rate'],
tca_htb_params['rate'])
self.assertEqual(self.CLASSES[handle]['ceil'],
tca_htb_params['ceil'])
burst = tc_lib._calc_burst(self.CLASSES[handle]['rate'],
tca_htb_params['buffer'])
self.assertEqual(self.CLASSES[handle]['burst'], burst)
def test_run(self):
test_cases = [(0x1, '0:1'), (0x2a003f, '2a:3f'), (0xf0000, 'f:0'),
(0xffffffff, 'ffff:ffff'), (0x12345678, '1234:5678')]
for _in, expected in test_cases:
self.assertEqual(expected, tc_lib._handle_from_hex_to_string(_in))
