75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
from neutron.tests.unit import testlib_api
|
|
from neutron_lib import context as ctx
|
|
|
|
from wan_qos.db import wan_qos_db
|
|
|
|
|
|
class TestTcDb(testlib_api.SqlTestCase):
|
|
def setUp(self):
|
|
super(TestTcDb, self).setUp()
|
|
self.db = wan_qos_db.WanTcDb()
|
|
self.context = ctx.get_admin_context()
|
|
|
|
def test_add_class(self):
|
|
wtc_class = {
|
|
'direction': 'both',
|
|
'min': '1mbit'
|
|
}
|
|
wtc_class_db = self.db.create_wan_tc_class(self.context, wtc_class)
|
|
|
|
assert wtc_class_db is not None
|
|
|
|
def test_get_class_by_id(self):
|
|
|
|
class_db_1 = self._add_class(None, 'both', '1mbit', '2mbit')
|
|
class_db_2 = self._add_class(class_db_1['id'], 'both', '2mbit',
|
|
'3mbit')
|
|
self._add_class(class_db_2['id'], 'both', '3mbit',
|
|
'4mbit')
|
|
|
|
class_by_id = self.db.get_class_by_id(self.context, class_db_1['id'])
|
|
# class_by_id = self.db.get_class_by_id(self.context, '111')
|
|
print (class_by_id)
|
|
|
|
def test_get_class_tree(self):
|
|
|
|
class_db_1 = self._add_class(None, 'both', '1mbit', '2mbit')
|
|
class_db_2 = self._add_class(class_db_1['id'], 'both', '2mbit',
|
|
'3mbit')
|
|
class_db_3 = self._add_class(class_db_2['id'], 'both', '3mbit',
|
|
'4mbit')
|
|
|
|
class_tree = self.db.get_class_tree()
|
|
assert class_tree is not None
|
|
print (class_tree)
|
|
|
|
class_tree = self.db.get_class_tree(class_db_1['id'])
|
|
assert class_tree is not None
|
|
print (class_tree)
|
|
|
|
class_tree = self.db.get_class_tree(class_db_2['id'])
|
|
assert class_tree is not None
|
|
print (class_tree)
|
|
|
|
class_tree = self.db.get_class_tree(class_db_3['id'])
|
|
assert class_tree is not None
|
|
print (class_tree)
|
|
|
|
def test_get_classes(self):
|
|
self.test_add_class()
|
|
all_classes = self.db.get_all_classes(self.context)
|
|
print ('all classes: %s' % all_classes)
|
|
|
|
def _add_class(self, parent, direction, min, max):
|
|
wtc_class = {
|
|
'direction': direction,
|
|
}
|
|
if min:
|
|
wtc_class['min'] = min
|
|
if parent:
|
|
wtc_class['parent'] = parent
|
|
if max:
|
|
wtc_class['max'] = max
|
|
|
|
return self.db.create_wan_tc_class(self.context, wtc_class)
|