def _register_mapping_member(self): session = self.ib_cxt.context.session mapping_relation = utils.get_mapping_relation( self.ib_cxt.mapping.authority_member.member_type) mapping_members = dbi.get_mapping_members( session, self.ib_cxt.mapping.network_view_id, self.ib_cxt.mapping.authority_member.member_id, self.grid_id, mapping_relation) if not mapping_members: dbi.add_mapping_member( session, self.ib_cxt.mapping.network_view_id, self.ib_cxt.mapping.authority_member.member_id, mapping_relation)
def _sync_network_mapping(self, discovered_networks, discovered_delegations): """Discover networks and sync with db. The discovered network json contains the following data: - network view - network - cloud_info for delegated member if cloud platform is supported - dhcp members :param discovered_networks: discovered network json :param discovered_delegations: discovered delegation members :return: None """ session = self._context.session self._load_persisted_mappings() discovered_mapping = self._get_member_mapping(discovered_networks, discovered_delegations) # add or remove authority mapping members persisted_authority_members = utils.get_composite_values_from_records( ['network_view_id', 'member_id', 'mapping_relation'], self.db_authority_members, DELIMITER) persisted_set = set(persisted_authority_members) discovered_set = set(discovered_mapping['authority_members']) addable_set = discovered_set.difference(persisted_set) removable_set = persisted_set.difference(discovered_set) for authority_member_info in addable_set: authority_member = authority_member_info.split(DELIMITER) network_view_id = authority_member[0] member_id = authority_member[1] mapping_relation = authority_member[2] dbi.add_mapping_member(session, network_view_id, member_id, mapping_relation) for authority_member_info in removable_set: authority_member = authority_member_info.split(DELIMITER) network_view_id = authority_member[0] member_id = authority_member[1] dbi.remove_mapping_member(session, network_view_id, member_id) # add or remove service members persisted_service_members = utils.get_composite_values_from_records( ['network_view_id', 'member_id', 'service'], self.db_service_members, DELIMITER) persisted_set = set(persisted_service_members) discovered_set = set(discovered_mapping['service_members']) addable_set = discovered_set.difference(persisted_set) removable_set = persisted_set.difference(discovered_set) for service_member_info in addable_set: service_member = service_member_info.split(DELIMITER) network_view_id = service_member[0] member_id = service_member[1] service = service_member[2] dbi.add_service_member(session, network_view_id, member_id, service) for service_member_info in removable_set: service_member = service_member_info.split(DELIMITER) network_view_id = service_member[0] member_id = service_member[1] service = service_member[2] dbi.remove_service_member(session, network_view_id, member_id=member_id, service=service)
def test_get_next_dhcp_member(self): # prepare grid self._create_default_grid() # prepare grid members member_list = [{'member_id': 'm1', 'member_name': 'm1.com', 'member_ip': '10.10.1.1', 'member_ipv6': None, 'member_type': const.MEMBER_TYPE_GRID_MASTER, 'member_status': 'ON'}, {'member_id': 'm2', 'member_name': 'm2.com', 'member_ip': '10.10.1.2', 'member_ipv6': 'fd44:acb:5df6:1083::22', 'member_type': const.MEMBER_TYPE_CP_MEMBER, 'member_status': 'ON'}, {'member_id': 'm3', 'member_name': 'm3.com', 'member_ip': '10.10.1.3', 'member_ipv6': None, 'member_type': const.MEMBER_TYPE_CP_MEMBER, 'member_status': 'ON'}, {'member_id': 'm4', 'member_name': 'm4.com', 'member_ip': '10.10.1.4', 'member_ipv6': None, 'member_type': const.MEMBER_TYPE_REGULAR_MEMBER, 'member_status': 'ON'}] self._create_members(member_list, self.grid_id) db_members = infoblox_db.get_members(self.ctx.session, grid_id=self.grid_id) gm_member = utils.find_one_in_list('member_type', 'GM', db_members) m2_member = utils.find_one_in_list('member_id', 'm2', db_members) m3_member = utils.find_one_in_list('member_id', 'm3', db_members) m4_member = utils.find_one_in_list('member_id', 'm4', db_members) # create network views netview_dict = {'default': gm_member.member_id, 'testview': m2_member.member_id} self._create_network_views(netview_dict) db_network_views = infoblox_db.get_network_views(self.ctx.session) netview_default = utils.find_one_in_list('network_view', 'default', db_network_views) netview_testview = utils.find_one_in_list('network_view', 'testview', db_network_views) # add mapping members # - m1 (gm) is the authority member for 'default' view # - m2 is the authority member for 'testview' view as CPM infoblox_db.add_mapping_member(self.ctx.session, netview_default.id, gm_member.member_id, const.MAPPING_RELATION_GM_OWNED) infoblox_db.add_mapping_member(self.ctx.session, netview_testview.id, m2_member.member_id, const.MAPPING_RELATION_DELEGATED) # add service members # - m4 is a member that serves DHCP/DNS for a network under # the non-delegated view, 'default' # - m2 is the delegated member to the network view, 'testview' infoblox_db.add_service_member(self.ctx.session, netview_default.id, m4_member.member_id, const.SERVICE_TYPE_DHCP) infoblox_db.add_service_member(self.ctx.session, netview_default.id, m4_member.member_id, const.SERVICE_TYPE_DNS) infoblox_db.add_service_member(self.ctx.session, netview_default.id, m2_member.member_id, const.SERVICE_TYPE_DHCP) infoblox_db.add_service_member(self.ctx.session, netview_default.id, m2_member.member_id, const.SERVICE_TYPE_DNS) dhcp_member = infoblox_db.get_next_dhcp_member(self.ctx.session, self.grid_id) # only available dhcp member should be 'm3' self.assertEqual(m3_member.member_id, dhcp_member.member_id)
def test_mapping_management_mapping_members(self): # prepare grid self._create_default_grid() # prepare grid members self._create_simple_members() db_members = infoblox_db.get_members(self.ctx.session) gm_member = utils.find_one_in_list('member_type', 'GM', db_members) # prepare network views netview_dict = {'default': gm_member.member_id} self._create_network_views(netview_dict) # get network view id db_network_views = infoblox_db.get_network_views(self.ctx.session) netview_default_row = utils.find_one_in_list('network_view', 'default', db_network_views) netview_id = netview_default_row.id # should be no mapping members db_mapping_members = infoblox_db.get_mapping_members(self.ctx.session) self.assertEqual(0, len(db_mapping_members)) # test mapping member additions expected_rows = [] member_id = db_members[0]['member_id'] relation = const.MAPPING_RELATION_DELEGATED infoblox_db.add_mapping_member(self.ctx.session, netview_id, member_id, relation) expected_rows.append(netview_id + ':' + member_id + ':' + relation) member_id = db_members[1]['member_id'] relation = const.MAPPING_RELATION_DELEGATED infoblox_db.add_mapping_member(self.ctx.session, netview_id, member_id, relation) expected_rows.append(netview_id + ':' + member_id + ':' + relation) db_mapping_members = infoblox_db.get_mapping_members(self.ctx.session) actual_rows = utils.get_composite_values_from_records( ['network_view_id', 'member_id', 'mapping_relation'], db_mapping_members, ':') self.assertEqual(expected_rows, actual_rows) # test mapping member update member_id = db_members[0]['member_id'] relation = const.MAPPING_RELATION_GM_OWNED infoblox_db.update_mapping_member(self.ctx.session, netview_id, member_id, relation) expected_rows[0] = netview_id + ':' + member_id + ':' + relation db_mapping_members = infoblox_db.get_mapping_members(self.ctx.session) actual_rows = utils.get_composite_values_from_records( ['network_view_id', 'member_id', 'mapping_relation'], db_mapping_members, ':') self.assertEqual(expected_rows, actual_rows) # test mapping member removals member_id = db_members[0]['member_id'] infoblox_db.remove_mapping_member(self.ctx.session, netview_id, member_id) db_mapping_members = infoblox_db.get_mapping_members( self.ctx.session, member_id=member_id) self.assertEqual([], db_mapping_members)
def test_get_next_dhcp_member(self): # prepare grid self._create_default_grid() # prepare grid members member_list = [{ 'member_id': 'm1', 'member_name': 'm1.com', 'member_ip': '10.10.1.1', 'member_ipv6': None, 'member_type': const.MEMBER_TYPE_GRID_MASTER, 'member_status': 'ON' }, { 'member_id': 'm2', 'member_name': 'm2.com', 'member_ip': '10.10.1.2', 'member_ipv6': 'fd44:acb:5df6:1083::22', 'member_type': const.MEMBER_TYPE_CP_MEMBER, 'member_status': 'ON' }, { 'member_id': 'm3', 'member_name': 'm3.com', 'member_ip': '10.10.1.3', 'member_ipv6': None, 'member_type': const.MEMBER_TYPE_CP_MEMBER, 'member_status': 'ON' }, { 'member_id': 'm4', 'member_name': 'm4.com', 'member_ip': '10.10.1.4', 'member_ipv6': None, 'member_type': const.MEMBER_TYPE_REGULAR_MEMBER, 'member_status': 'ON' }] self._create_members(member_list, self.grid_id) db_members = infoblox_db.get_members(self.ctx.session, grid_id=self.grid_id) gm_member = utils.find_one_in_list('member_type', 'GM', db_members) m2_member = utils.find_one_in_list('member_id', 'm2', db_members) m3_member = utils.find_one_in_list('member_id', 'm3', db_members) m4_member = utils.find_one_in_list('member_id', 'm4', db_members) # create network views netview_dict = { 'default': gm_member.member_id, 'testview': m2_member.member_id } self._create_network_views(netview_dict) db_network_views = infoblox_db.get_network_views(self.ctx.session) netview_default = utils.find_one_in_list('network_view', 'default', db_network_views) netview_testview = utils.find_one_in_list('network_view', 'testview', db_network_views) # add mapping members # - m1 (gm) is the authority member for 'default' view # - m2 is the authority member for 'testview' view as CPM infoblox_db.add_mapping_member(self.ctx.session, netview_default.id, gm_member.member_id, const.MAPPING_RELATION_GM_OWNED) infoblox_db.add_mapping_member(self.ctx.session, netview_testview.id, m2_member.member_id, const.MAPPING_RELATION_DELEGATED) # add service members # - m4 is a member that serves DHCP/DNS for a network under # the non-delegated view, 'default' # - m2 is the delegated member to the network view, 'testview' infoblox_db.add_service_member(self.ctx.session, netview_default.id, m4_member.member_id, const.SERVICE_TYPE_DHCP) infoblox_db.add_service_member(self.ctx.session, netview_default.id, m4_member.member_id, const.SERVICE_TYPE_DNS) infoblox_db.add_service_member(self.ctx.session, netview_default.id, m2_member.member_id, const.SERVICE_TYPE_DHCP) infoblox_db.add_service_member(self.ctx.session, netview_default.id, m2_member.member_id, const.SERVICE_TYPE_DNS) dhcp_member = infoblox_db.get_next_dhcp_member(self.ctx.session, self.grid_id) # only available dhcp member should be 'm3' self.assertEqual(m3_member.member_id, dhcp_member.member_id)