def test_segmentation_id_is_nothing(self):
self.net.network_segments[0]["segmentation_id"] = None
network_type, vlan_id = fj_util.get_network_segments(self.net)
self.assertEqual('vlan', network_type)
self.assertIsNone(vlan_id)
def test_network_type_is_not_vlan(self):
self.net.network_segments[0]["network_type"] = "vxlan"
network_type, vlan_id = fj_util.get_network_segments(self.net)
self.assertEqual('vxlan', network_type)
self.assertEqual(100, vlan_id)
def test_normal_case(self):
network_type, vlan_id = fj_util.get_network_segments(self.net)
self.assertEqual("vlan", network_type)
self.assertEqual(100, vlan_id)
