def delete_network_precommit(self, context): """Delete the network infromation from the DB.""" network = context.current network_id = network['id'] tenant_id = network['tenant_id'] with self.eos_sync_lock: if db_lib.is_network_provisioned(tenant_id, network_id): db_lib.forget_network(tenant_id, network_id)
def test_network_is_removed(self): tenant_id = 'test' network_id = '123' db_lib.remember_network(tenant_id, network_id, '123') db_lib.forget_network(tenant_id, network_id) net_provisioned = db_lib.is_network_provisioned(tenant_id, network_id) self.assertFalse(net_provisioned, 'The network should be deleted')
def _cleanup_db(self): """Clean up any uncessary entries in our DB.""" db_tenants = db_lib.get_tenants() for tenant in db_tenants: neutron_nets = self.ndb.get_all_networks_for_tenant(tenant) neutron_nets_id = [] for net in neutron_nets: neutron_nets_id.append(net['id']) db_nets = db_lib.get_networks(tenant) for net_id in db_nets.keys(): if net_id not in neutron_nets_id: db_lib.forget_network(tenant, net_id)
def _cleanup_db(self): """Clean up any uncessary entries in our DB.""" neutron_nets = self.ndb.get_all_networks() arista_db_nets = db_lib.get_networks(tenant_id='any') neutron_net_ids = set() for net in neutron_nets: neutron_net_ids.add(net['id']) # Remove networks from the Arista DB if the network does not exist in # Neutron DB for net_id in set(arista_db_nets.keys()).difference(neutron_net_ids): tenant_network = arista_db_nets[net_id] db_lib.forget_network(tenant_network['tenantId'], net_id)
def delete_network_precommit(self, context): """Delete the network information from the DB.""" network = context.current network_id = network['id'] tenant_id = network['tenant_id'] or INTERNAL_TENANT_ID with self.eos_sync_lock: if db_lib.is_network_provisioned(tenant_id, network_id): if db_lib.are_ports_attached_to_network(network_id): LOG.info(_LI('Network %s can not be deleted as it ' 'has ports attached to it'), network_id) raise ml2_exc.MechanismDriverError( method='delete_network_precommit') else: db_lib.forget_network(tenant_id, network_id)
def test_removes_all_networks(self): tenant_id = 'test' num_nets = 100 old_nets = db_lib.num_nets_provisioned(tenant_id) nets = ['id_%s' % n for n in range(num_nets)] for net_id in nets: db_lib.remember_network(tenant_id, net_id, 123) for net_id in nets: db_lib.forget_network(tenant_id, net_id) num_nets_provisioned = db_lib.num_nets_provisioned(tenant_id) expected = old_nets self.assertEqual(expected, num_nets_provisioned, 'There should be %d nets, not %d' % (expected, num_nets_provisioned))
def test_removes_all_networks(self): tenant_id = 'test' num_nets = 100 old_nets = db_lib.num_nets_provisioned(tenant_id) nets = ['id_%s' % n for n in range(num_nets)] for net_id in nets: db_lib.remember_network(tenant_id, net_id, 123) for net_id in nets: db_lib.forget_network(tenant_id, net_id) num_nets_provisioned = db_lib.num_nets_provisioned(tenant_id) expected = old_nets self.assertEqual( expected, num_nets_provisioned, 'There should be %d nets, not %d' % (expected, num_nets_provisioned))
def test_synchronize_required(self): """Tests whether synchronize() sends the right commands. This test verifies a scenario when the sync is required. """ region_updated_time = { 'regionName': 'RegionOne', 'regionTimestamp': '12345' } self.rpc.get_region_updated_time.return_value = region_updated_time self.sync_service._region_updated_time = { 'regionName': 'RegionOne', 'regionTimestamp': '0', } tenant_id = 'tenant-1' network_id = 'net-1' segmentation_id = 42 db_lib.remember_tenant(tenant_id) db_lib.remember_network(tenant_id, network_id, segmentation_id) self.rpc.get_tenants.return_value = {} self.rpc.sync_start.return_value = True self.rpc.sync_end.return_value = True self.sync_service.do_synchronize() expected_calls = [ mock.call.get_region_updated_time(), mock.call.sync_start(), mock.call.register_with_eos(sync=True), mock.call.get_tenants(), mock.call.create_network_bulk( tenant_id, [{'network_id': network_id, 'segmentation_id': segmentation_id, 'network_name': '', 'shared': False}], sync=True), mock.call.sync_end(), mock.call.get_region_updated_time() ] assert self.rpc.mock_calls == expected_calls db_lib.forget_network(tenant_id, network_id) db_lib.forget_tenant(tenant_id)
def test_synchronize_all_networks(self): """Test to ensure that only the required resources are sent to EOS.""" # Store two tenants in a db and none on EOS. # The sync should send details of all tenants to EOS tenant_1_id = u'tenant-1' tenant_1_net_1_id = u'ten-1-net-1' tenant_1_net_1_seg_id = 11 db_lib.remember_tenant(tenant_1_id) db_lib.remember_network(tenant_1_id, tenant_1_net_1_id, tenant_1_net_1_seg_id) tenant_2_id = u'tenant-2' tenant_2_net_1_id = u'ten-2-net-1' tenant_2_net_1_seg_id = 21 db_lib.remember_tenant(tenant_2_id) db_lib.remember_network(tenant_2_id, tenant_2_net_1_id, tenant_2_net_1_seg_id) self.rpc.get_tenants.return_value = {} self.sync_service.do_synchronize() expected_calls = [ mock.call.get_region_updated_time(), mock.call._run_openstack_cmds(['sync start']), mock.call.register_with_eos(), mock.call.get_tenants(), mock.call.create_network_bulk( tenant_1_id, [{'network_id': tenant_1_net_1_id, 'segmentation_id': tenant_1_net_1_seg_id, 'network_name': '', 'shared': False}]), mock.call.create_network_bulk( tenant_2_id, [{'network_id': tenant_2_net_1_id, 'segmentation_id': tenant_2_net_1_seg_id, 'network_name': '', 'shared': False}]), mock.call._run_openstack_cmds(['sync end']), mock.call.get_region_updated_time() ] # The create_network_bulk() can be called in different order. So split # it up. The first part checks if the initial set of methods are # invoked. self.assertTrue(self.rpc.mock_calls[:4] == expected_calls[:4], "Seen: %s\nExpected: %s" % ( self.rpc.mock_calls, expected_calls, ) ) # Check if tenant 1 networks are created. It must be one of the two # methods. self.assertTrue(self.rpc.mock_calls[4] in expected_calls[4:6], "Seen: %s\nExpected: %s" % ( self.rpc.mock_calls, expected_calls, ) ) # Check if tenant 2 networks are created. It must be one of the two # methods. self.assertTrue(self.rpc.mock_calls[5] in expected_calls[4:6], "Seen: %s\nExpected: %s" % ( self.rpc.mock_calls, expected_calls, ) ) # Check if the sync end methods are invoked. self.assertTrue(self.rpc.mock_calls[6:8] == expected_calls[6:8], "Seen: %s\nExpected: %s" % ( self.rpc.mock_calls, expected_calls, ) ) db_lib.forget_network(tenant_1_id, tenant_1_net_1_id) db_lib.forget_network(tenant_2_id, tenant_2_net_1_id) db_lib.forget_tenant(tenant_1_id) db_lib.forget_tenant(tenant_2_id)
def test_synchronize_one_network(self): """Test to ensure that only the required resources are sent to EOS.""" # Store two tenants in a db and a single tenant in EOS. # The sync should send details of the second tenant to EOS tenant_1_id = 'tenant-1' tenant_1_net_1_id = 'ten-1-net-1' tenant_1_net_1_seg_id = 11 db_lib.remember_tenant(tenant_1_id) db_lib.remember_network(tenant_1_id, tenant_1_net_1_id, tenant_1_net_1_seg_id) tenant_2_id = 'tenant-2' tenant_2_net_1_id = 'ten-2-net-1' tenant_2_net_1_seg_id = 21 db_lib.remember_tenant(tenant_2_id) db_lib.remember_network(tenant_2_id, tenant_2_net_1_id, tenant_2_net_1_seg_id) self.rpc.get_tenants.return_value = { tenant_1_id: { 'tenantVmInstances': {}, 'tenantNetworks': { tenant_1_net_1_id: { 'networkId': tenant_1_net_1_id, 'shared': False, 'networkName': 'Net1', 'segmenationType': 'vlan', 'segmentationTypeId': tenant_1_net_1_seg_id, } } } } self.sync_service.do_synchronize() expected_calls = [ mock.call.get_region_updated_time(), mock.call._run_openstack_cmds(['sync start']), mock.call.register_with_eos(), mock.call.get_tenants(), mock.call.create_network_bulk( tenant_2_id, [{'network_id': tenant_2_net_1_id, 'segmentation_id': tenant_2_net_1_seg_id, 'network_name': '', 'shared': False}]), mock.call._run_openstack_cmds(['sync end']), mock.call.get_region_updated_time() ] self.assertTrue(self.rpc.mock_calls == expected_calls, "Seen: %s\nExpected: %s" % ( self.rpc.mock_calls, expected_calls, ) ) db_lib.forget_network(tenant_1_id, tenant_1_net_1_id) db_lib.forget_network(tenant_2_id, tenant_2_net_1_id) db_lib.forget_tenant(tenant_1_id) db_lib.forget_tenant(tenant_2_id)
def test_synchronize_all_networks(self): """Test to ensure that only the required resources are sent to EOS.""" # Store two tenants in a db and none on EOS. # The sync should send details of all tenants to EOS tenant_1_id = u'tenant-1' tenant_1_net_1_id = u'ten-1-net-1' tenant_1_net_1_seg_id = 11 db_lib.remember_tenant(tenant_1_id) db_lib.remember_network(tenant_1_id, tenant_1_net_1_id, tenant_1_net_1_seg_id) tenant_2_id = u'tenant-2' tenant_2_net_1_id = u'ten-2-net-1' tenant_2_net_1_seg_id = 21 db_lib.remember_tenant(tenant_2_id) db_lib.remember_network(tenant_2_id, tenant_2_net_1_id, tenant_2_net_1_seg_id) self.rpc.get_tenants.return_value = {} self.sync_service.do_synchronize() expected_calls = [ mock.call.get_region_updated_time(), mock.call._run_openstack_cmds(['sync start']), mock.call.register_with_eos(), mock.call.get_tenants(), mock.call.create_network_bulk(tenant_1_id, [{ 'network_id': tenant_1_net_1_id, 'segmentation_id': tenant_1_net_1_seg_id, 'network_name': '', 'shared': False }]), mock.call.create_network_bulk(tenant_2_id, [{ 'network_id': tenant_2_net_1_id, 'segmentation_id': tenant_2_net_1_seg_id, 'network_name': '', 'shared': False }]), mock.call._run_openstack_cmds(['sync end']), mock.call.get_region_updated_time() ] # The create_network_bulk() can be called in different order. So split # it up. The first part checks if the initial set of methods are # invoked. self.assertTrue( self.rpc.mock_calls[:4] == expected_calls[:4], "Seen: %s\nExpected: %s" % ( self.rpc.mock_calls, expected_calls, )) # Check if tenant 1 networks are created. It must be one of the two # methods. self.assertTrue( self.rpc.mock_calls[4] in expected_calls[4:6], "Seen: %s\nExpected: %s" % ( self.rpc.mock_calls, expected_calls, )) # Check if tenant 2 networks are created. It must be one of the two # methods. self.assertTrue( self.rpc.mock_calls[5] in expected_calls[4:6], "Seen: %s\nExpected: %s" % ( self.rpc.mock_calls, expected_calls, )) # Check if the sync end methods are invoked. self.assertTrue( self.rpc.mock_calls[6:8] == expected_calls[6:8], "Seen: %s\nExpected: %s" % ( self.rpc.mock_calls, expected_calls, )) db_lib.forget_network(tenant_1_id, tenant_1_net_1_id) db_lib.forget_network(tenant_2_id, tenant_2_net_1_id) db_lib.forget_tenant(tenant_1_id) db_lib.forget_tenant(tenant_2_id)
def test_synchronize_one_network(self): """Test to ensure that only the required resources are sent to EOS.""" # Store two tenants in a db and a single tenant in EOS. # The sync should send details of the second tenant to EOS tenant_1_id = 'tenant-1' tenant_1_net_1_id = 'ten-1-net-1' tenant_1_net_1_seg_id = 11 db_lib.remember_tenant(tenant_1_id) db_lib.remember_network(tenant_1_id, tenant_1_net_1_id, tenant_1_net_1_seg_id) tenant_2_id = 'tenant-2' tenant_2_net_1_id = 'ten-2-net-1' tenant_2_net_1_seg_id = 21 db_lib.remember_tenant(tenant_2_id) db_lib.remember_network(tenant_2_id, tenant_2_net_1_id, tenant_2_net_1_seg_id) self.rpc.get_tenants.return_value = { tenant_1_id: { 'tenantVmInstances': {}, 'tenantNetworks': { tenant_1_net_1_id: { 'networkId': tenant_1_net_1_id, 'shared': False, 'networkName': 'Net1', 'segmenationType': 'vlan', 'segmentationTypeId': tenant_1_net_1_seg_id, } } } } self.sync_service.do_synchronize() expected_calls = [ mock.call.get_region_updated_time(), mock.call._run_openstack_cmds(['sync start']), mock.call.register_with_eos(), mock.call.get_tenants(), mock.call.create_network_bulk(tenant_2_id, [{ 'network_id': tenant_2_net_1_id, 'segmentation_id': tenant_2_net_1_seg_id, 'network_name': '', 'shared': False }]), mock.call._run_openstack_cmds(['sync end']), mock.call.get_region_updated_time() ] self.assertTrue( self.rpc.mock_calls == expected_calls, "Seen: %s\nExpected: %s" % ( self.rpc.mock_calls, expected_calls, )) db_lib.forget_network(tenant_1_id, tenant_1_net_1_id) db_lib.forget_network(tenant_2_id, tenant_2_net_1_id) db_lib.forget_tenant(tenant_1_id) db_lib.forget_tenant(tenant_2_id)