def test_get_ip_for_subscriber_with_wildcard_apn_gw(self):
""" test wildcard apn"""
apn = 'magma'
imsi = 'IMSI110'
sid = imsi + '.' + apn + ",ipv4"
assigned_ip = '1.2.3.4'
gw_ip = "1.2.3.100"
gw_mac = "11:22:33:11:77:81"
vlan = "300"
MockedSubscriberDBStub.add_sub(sid=imsi, apn=apn, ip=assigned_ip,
gw_ip=gw_ip, gw_mac=gw_mac, vlan=vlan)
wildcard_assigned_ip = "20.20.20.20"
wildcard_gw_ip = "1.2.7.7"
wildcard_gw_mac = "11:22:33:88:77:99"
wildcard_vlan = "400"
MockedSubscriberDBStub.add_sub_ip(sid=imsi, apn="*",
ip=wildcard_assigned_ip,
gw_ip=wildcard_gw_ip,
gw_mac=wildcard_gw_mac,
vlan=wildcard_vlan)
ip0, _ = self._allocator.alloc_ip_address(sid)
ip0_returned = self._allocator.get_ip_for_sid(sid)
# check if retrieved ip is the same as the one allocated
self.assertEqual(ip0, ip0_returned)
self.assertEqual(ip0, ipaddress.ip_address(assigned_ip))
self.check_type(sid, IPType.STATIC)
self.check_gw_info(vlan, gw_ip, gw_mac)
self.check_gw_info(wildcard_vlan, None, None)
def test_get_ip_for_subscriber_with_wildcard_and_exact_apn_no_ip(self):
""" test IP assignement from multiple APNs"""
apn = 'magma'
imsi = 'IMSI110'
sid = imsi + '.' + apn + ",ipv4"
assigned_ip_wild = '22.22.22.22'
MockedSubscriberDBStub.add_sub(sid=imsi, apn="*", ip=assigned_ip_wild)
MockedSubscriberDBStub.add_sub_ip(sid=imsi, apn=apn, ip=None)
ip0, _ = self._allocator.alloc_ip_address(sid)
ip0_returned = self._allocator.get_ip_for_sid(sid)
# check if retrieved ip is the same as the one allocated
self.assertEqual(ip0, ip0_returned)
self.assertNotEqual(ip0, ipaddress.ip_address(assigned_ip_wild))
self.check_type(sid, IPType.IP_POOL)
def test_get_ipv6_for_subscriber_with_wildcard_and_no_exact_apn(self):
""" test IP assignement from multiple APNs"""
apn = 'magma'
imsi = 'IMSI110'
sid = imsi + '.' + apn + ",ipv6"
assigned_ip = '2022::1'
assigned_ip_wild = '2022::22:22'
MockedSubscriberDBStub.add_sub(sid=imsi, apn="*", ip=assigned_ip_wild)
MockedSubscriberDBStub.add_sub_ip(sid=imsi, apn="xyz", ip=assigned_ip)
ip0, _ = self._allocator.alloc_ip_address(sid, version=IPAddress.IPV6)
ip0_returned = self._allocator.get_ip_for_sid(sid)
# check if retrieved ip is the same as the one allocated
self.assertEqual(ip0, ip0_returned)
self.assertEqual(ip0, ipaddress.ip_address(assigned_ip_wild))
self.check_type(sid, IPType.STATIC)
def test_get_ip_for_subscriber_with_multi_apn_but_no_match(self):
""" test IP assignment from multiple APNs"""
apn = 'magma'
imsi = 'IMSI110'
sid = imsi + '.' + apn
assigned_ip = '1.2.3.4'
assigned_ip_wild = '22.22.22.22'
MockedSubscriberDBStub.add_sub(sid=imsi, apn="abc", ip=assigned_ip_wild)
MockedSubscriberDBStub.add_sub_ip(sid=imsi, apn="xyz", ip=assigned_ip)
ip0, _ = self._allocator.alloc_ip_address(sid)
ip0_returned = self._allocator.get_ip_for_sid(sid)
# check if retrieved ip is the same as the one allocated
self.assertEqual(ip0, ip0_returned)
self.assertNotEqual(ip0, ipaddress.ip_address(assigned_ip))
self.check_type(sid, IPType.IP_POOL)
