Add missing unit test for segment db

This patch add test for get_dynamic_segment.

Newton changes: get_dynamic_segments receives session as the first
argument, not context.

Change-Id: Icf676059a0dd24ad18b8632563c4de81190f0d04
Closes-Bug: #1684519
(cherry picked from commit 8595a01702)
This commit is contained in:
Dongcan Ye 2017-05-04 14:02:05 +08:00 committed by Ihar Hrachyshka
parent b5b68b3752
commit 00ac588968
1 changed files with 27 additions and 0 deletions

View File

@ -164,6 +164,33 @@ class Ml2DBTestCase(testlib_api.SqlTestCase):
segment_uuid)
self.assertIsNone(net_segment)
def test_get_dynamic_segment(self):
net_id = uuidutils.generate_uuid()
segment1 = {api.NETWORK_TYPE: 'vlan',
api.PHYSICAL_NETWORK: 'physnet1',
api.SEGMENTATION_ID: 1}
self._create_segments(
[segment1], is_seg_dynamic=True, network_id=net_id)
segs1 = segments_db.get_dynamic_segment(
self.ctx.session, net_id)
self.assertEqual('vlan', segs1[api.NETWORK_TYPE])
self.assertEqual('physnet1', segs1[api.PHYSICAL_NETWORK])
self.assertEqual(1, segs1[api.SEGMENTATION_ID])
segs2 = segments_db.get_dynamic_segment(
self.ctx.session, net_id, physical_network='physnet1')
self.assertEqual('vlan', segs2[api.NETWORK_TYPE])
self.assertEqual('physnet1', segs2[api.PHYSICAL_NETWORK])
self.assertEqual(1, segs2[api.SEGMENTATION_ID])
segs3 = segments_db.get_dynamic_segment(
self.ctx.session, net_id, segmentation_id=1)
self.assertEqual('vlan', segs3[api.NETWORK_TYPE])
self.assertEqual('physnet1', segs3[api.PHYSICAL_NETWORK])
self.assertEqual(1, segs3[api.SEGMENTATION_ID])
def test_add_port_binding(self):
network_id = 'foo-network-id'
port_id = 'foo-port-id'