From 8595a017028450ba1c5f7cd0da4e6a9bd624666e Mon Sep 17 00:00:00 2001 From: Dongcan Ye Date: Thu, 4 May 2017 14:02:05 +0800 Subject: [PATCH] Add missing unit test for segment db This patch add test for get_dynamic_segment. Change-Id: Icf676059a0dd24ad18b8632563c4de81190f0d04 Closes-Bug: #1684519 --- neutron/tests/unit/plugins/ml2/test_db.py | 27 +++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/neutron/tests/unit/plugins/ml2/test_db.py b/neutron/tests/unit/plugins/ml2/test_db.py index 1cc76205fac..285d04edd0d 100644 --- a/neutron/tests/unit/plugins/ml2/test_db.py +++ b/neutron/tests/unit/plugins/ml2/test_db.py @@ -177,6 +177,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, 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, 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, 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 = uuidutils.generate_uuid() port_id = uuidutils.generate_uuid()