def execute(self, args):
mgr = NetworkManager(self.client)
table = Table([
'id', 'number', 'datacenter', 'name', 'IPs', 'hardware', 'ccis',
'networking', 'firewall'
])
table.sortby = args.get('--sortby') or 'id'
vlans = mgr.list_vlans(
datacenter=args.get('--datacenter'),
vlan_number=args.get('--number'),
name=args.get('--name'),
)
for vlan in vlans:
table.add_row([
vlan['id'],
vlan['vlanNumber'],
vlan['primaryRouter']['datacenter']['name'],
vlan.get('name') or blank(),
vlan['totalPrimaryIpAddressCount'],
len(vlan['hardware']),
len(vlan['virtualGuests']),
len(vlan['networkComponents']),
'Yes' if vlan['firewallInterfaces'] else 'No',
])
return table
def execute(self, args):
mgr = NetworkManager(self.client)
update = {
'abuse_email': args.get('--abuse'),
'address1': args.get('--address1'),
'address2': args.get('--address2'),
'city': args.get('--city'),
'country': args.get('--country'),
'first_name': args.get('--firstname'),
'last_name': args.get('--lastname'),
'postal_code': args.get('--postal'),
'state': args.get('--state')
}
if args.get('--private'):
update['private_residence'] = False
elif args.get('--public'):
update['private_residence'] = True
check = [x for x in update.values() if x is not None]
if not check:
raise CLIAbort("You must specify at least one field to update.")
mgr.edit_rwhois(**update) # pylint: disable=W0142
def execute(client, args):
mgr = NetworkManager(client)
t = Table([
'id', 'ip', 'assigned', 'target'
])
t.sortby = args.get('--sortby') or 'id'
version = 0
if args.get('--v4'):
version = 4
elif args.get('--v6'):
version = 6
ips = mgr.list_global_ips(version=version)
for ip in ips:
assigned = 'No'
target = 'None'
if ip.get('destinationIpAddress'):
dest = ip['destinationIpAddress']
assigned = 'Yes'
target = dest['ipAddress']
if dest.get('virtualGuest'):
vg = dest['virtualGuest']
target += ' (' + vg['fullyQualifiedDomainName'] + ')'
elif ip['destinationIpAddress'].get('hardware'):
target += ' (' + \
dest['hardware']['fullyQualifiedDomainName'] + \
')'
t.add_row([ip['id'], ip['ipAddress']['ipAddress'], assigned,
target])
return t
def execute(client, args):
mgr = NetworkManager(client)
version = 4
if args.get('--v6'):
version = 6
if not args.get('--test') and not args['--really']:
if not confirm("This action will incur charges on your account."
"Continue?"):
raise CLIAbort('Cancelling order.')
result = mgr.add_global_ip(version=version,
test_order=args.get('--test'))
if not result:
return 'Unable to place order: No valid price IDs found.'
t = Table(['Item', 'cost'])
t.align['Item'] = 'r'
t.align['cost'] = 'r'
total = 0.0
for price in result['orderDetails']['prices']:
total += float(price.get('recurringFee', 0.0))
rate = "%.2f" % float(price['recurringFee'])
t.add_row([price['item']['description'], rate])
t.add_row(['Total monthly cost', "%.2f" % total])
output = SequentialOutput()
output.append(t)
output.append(FormattedItem(
'',
' -- ! Prices reflected here are retail and do not '
'take account level discounts and are not guarenteed.')
)
return t
def getVlanIdFromName(self, datacenter, vlan):
nwManager = NetworkManager(self.Client)
vlans = nwManager.list_vlans(datacenter=datacenter, vlan_number=vlan, mask='id')
if len(vlans) == 1:
return vlans[0]['id']
else:
return None
def execute(self, args):
mgr = NetworkManager(self.client)
version = 4
if args.get('--v6'):
version = 6
if not args.get('--test') and not args['--really']:
if not confirm("This action will incur charges on your account."
"Continue?"):
raise CLIAbort('Cancelling order.')
result = mgr.add_global_ip(version=version,
test_order=args.get('--test'))
if not result:
return 'Unable to place order: No valid price IDs found.'
table = Table(['Item', 'cost'])
table.align['Item'] = 'r'
table.align['cost'] = 'r'
total = 0.0
for price in result['orderDetails']['prices']:
total += float(price.get('recurringFee', 0.0))
rate = "%.2f" % float(price['recurringFee'])
table.add_row([price['item']['description'], rate])
table.add_row(['Total monthly cost', "%.2f" % total])
return table
def execute(self, args):
mgr = NetworkManager(self.client)
t = Table([
'id', 'identifier', 'type', 'datacenter', 'vlan id', 'IPs',
'hardware', 'ccis',
])
t.sortby = args.get('--sortby') or 'id'
version = 0
if args.get('--v4'):
version = 4
elif args.get('--v6'):
version = 6
subnets = mgr.list_subnets(
datacenter=args.get('--datacenter'),
version=version,
identifier=args.get('--identifier'),
subnet_type=args.get('--type'),
)
for subnet in subnets:
t.add_row([
subnet['id'],
subnet['networkIdentifier'] + '/' + str(subnet['cidr']),
subnet.get('subnetType', '-'),
subnet['datacenter']['name'],
subnet['networkVlanId'],
subnet['ipAddressCount'],
len(subnet['hardware']),
len(subnet['virtualGuests']),
])
return t
def execute(self, args):
mgr = NetworkManager(self.client)
vlan = mgr.get_vlan(args.get('<identifier>'))
t = KeyValueTable(['Name', 'Value'])
t.align['Name'] = 'r'
t.align['Value'] = 'l'
t.add_row(['id', vlan['id']])
t.add_row(['number', vlan['vlanNumber']])
t.add_row(['datacenter',
vlan['primaryRouter']['datacenter']['longName']])
t.add_row(['primary router',
vlan['primaryRouter']['fullyQualifiedDomainName']])
t.add_row(['firewall', 'Yes' if vlan['firewallInterfaces'] else 'No'])
subnets = []
for subnet in vlan['subnets']:
subnet_table = KeyValueTable(['Name', 'Value'])
subnet_table.align['Name'] = 'r'
subnet_table.align['Value'] = 'l'
subnet_table.add_row(['id', subnet['id']])
subnet_table.add_row(['identifier', subnet['networkIdentifier']])
subnet_table.add_row(['netmask', subnet['netmask']])
subnet_table.add_row(['gateway', subnet.get('gateway', '-')])
subnet_table.add_row(['type', subnet['subnetType']])
subnet_table.add_row(['usable ips',
subnet['usableIpAddressCount']])
subnets.append(subnet_table)
t.add_row(['subnets', subnets])
if not args.get('--no-cci'):
if vlan['virtualGuests']:
cci_table = KeyValueTable(['Hostname', 'Domain', 'IP'])
cci_table.align['Hostname'] = 'r'
cci_table.align['IP'] = 'l'
for cci in vlan['virtualGuests']:
cci_table.add_row([cci['hostname'],
cci['domain'],
cci.get('primaryIpAddress')])
t.add_row(['ccis', cci_table])
else:
t.add_row(['cci', 'none'])
if not args.get('--no-hardware'):
if vlan['hardware']:
hw_table = Table(['Hostname', 'Domain', 'IP'])
hw_table.align['Hostname'] = 'r'
hw_table.align['IP'] = 'l'
for hw in vlan['hardware']:
hw_table.add_row([hw['hostname'],
hw['domain'],
hw.get('primaryIpAddress')])
t.add_row(['hardware', hw_table])
else:
t.add_row(['hardware', 'none'])
return t
def execute(client, args):
mgr = NetworkManager(client)
id = mgr.resolve_global_ip_ids(args.get('<identifier>'))
if args['--really'] or no_going_back(id):
mgr.cancel_global_ip(id)
else:
CLIAbort('Aborted')
def execute(client, args):
mgr = NetworkManager(client)
id = mgr.resolve_global_ip_ids(args.get('<identifier>'))
if not id:
raise CLIAbort("Unable to find global IP record for " +
args['<identifier>'])
mgr.assign_global_ip(id, args['<target>'])
def execute(self, args):
mgr = NetworkManager(self.client)
subnet_id = mgr.resolve_subnet_ids(args.get('<identifier>'))
if args['--really'] or no_going_back(subnet_id):
mgr.cancel_subnet(subnet_id)
else:
CLIAbort('Aborted')
def execute(self, args):
mgr = NetworkManager(self.client)
global_ip_id = mgr.resolve_global_ip_ids(args.get('<identifier>'))
if not global_ip_id:
raise CLIAbort("Unable to find global IP record for " +
args['<identifier>'])
mgr.unassign_global_ip(global_ip_id)
def execute(self, args):
mgr = NetworkManager(self.client)
global_ip_id = resolve_id(mgr.resolve_global_ip_ids,
args.get('<identifier>'),
name='global ip')
if not global_ip_id:
raise CLIAbort("Unable to find global IP record for " +
args['<identifier>'])
mgr.assign_global_ip(global_ip_id, args['<target>'])
def execute(self, args):
mgr = NetworkManager(self.client)
global_ip_id = resolve_id(mgr.resolve_global_ip_ids,
args.get('<identifier>'),
name='global ip')
if args['--really'] or no_going_back(global_ip_id):
mgr.cancel_global_ip(global_ip_id)
else:
CLIAbort('Aborted')
def execute(self, args):
mgr = NetworkManager(self.client)
subnet_id = resolve_id(mgr.resolve_subnet_ids,
args.get('<identifier>'),
name='subnet')
subnet = mgr.get_subnet(subnet_id)
table = KeyValueTable(['Name', 'Value'])
table.align['Name'] = 'r'
table.align['Value'] = 'l'
table.add_row(['id', subnet['id']])
table.add_row(['identifier',
'%s/%s' % (subnet['networkIdentifier'],
str(subnet['cidr']))])
table.add_row(['subnet type', subnet['subnetType']])
table.add_row(['gateway', subnet.get('gateway', blank())])
table.add_row(['broadcast', subnet.get('broadcastAddress', blank())])
table.add_row(['datacenter', subnet['datacenter']['name']])
table.add_row(['usable ips',
subnet.get('usableIpAddressCount', blank())])
if not args.get('--no-cci'):
if subnet['virtualGuests']:
cci_table = Table(['Hostname', 'Domain', 'IP'])
cci_table.align['Hostname'] = 'r'
cci_table.align['IP'] = 'l'
for cci in subnet['virtualGuests']:
cci_table.add_row([cci['hostname'],
cci['domain'],
cci.get('primaryIpAddress')])
table.add_row(['ccis', cci_table])
else:
table.add_row(['cci', 'none'])
if not args.get('--no-hardware'):
if subnet['hardware']:
hw_table = Table(['Hostname', 'Domain', 'IP'])
hw_table.align['Hostname'] = 'r'
hw_table.align['IP'] = 'l'
for hardware in subnet['hardware']:
hw_table.add_row([hardware['hostname'],
hardware['domain'],
hardware.get('primaryIpAddress')])
table.add_row(['hardware', hw_table])
else:
table.add_row(['hardware', 'none'])
return table
def execute(self, args):
mgr = NetworkManager(self.client)
addr_info = mgr.ip_lookup(args['<ip>'])
if not addr_info:
return 'Not found'
table = KeyValueTable(['Name', 'Value'])
table.align['Name'] = 'r'
table.align['Value'] = 'l'
table.add_row(['id', addr_info['id']])
table.add_row(['ip', addr_info['ipAddress']])
subnet_table = KeyValueTable(['Name', 'Value'])
subnet_table.align['Name'] = 'r'
subnet_table.align['Value'] = 'l'
subnet_table.add_row(['id', addr_info['subnet']['id']])
subnet_table.add_row(['identifier',
'%s/%s'
% (addr_info['subnet']['networkIdentifier'],
str(addr_info['subnet']['cidr']))])
subnet_table.add_row(['netmask', addr_info['subnet']['netmask']])
if addr_info['subnet'].get('gateway'):
subnet_table.add_row(['gateway', addr_info['subnet']['gateway']])
subnet_table.add_row(['type', addr_info['subnet'].get('subnetType')])
table.add_row(['subnet', subnet_table])
if addr_info.get('virtualGuest') or addr_info.get('hardware'):
device_table = KeyValueTable(['Name', 'Value'])
device_table.align['Name'] = 'r'
device_table.align['Value'] = 'l'
if addr_info.get('virtualGuest'):
device = addr_info['virtualGuest']
device_type = 'cci'
else:
device = addr_info['hardware']
device_type = 'server'
device_table.add_row(['id', device['id']])
device_table.add_row(['name', device['fullyQualifiedDomainName']])
device_table.add_row(['type', device_type])
table.add_row(['device', device_table])
return table
def execute(self, args):
mgr = NetworkManager(self.client)
result = mgr.get_rwhois()
table = KeyValueTable(['Name', 'Value'])
table.align['Name'] = 'r'
table.align['Value'] = 'l'
table.add_row(['Name', result['firstName'] + ' ' + result['lastName']])
table.add_row(['Company', result['companyName']])
table.add_row(['Abuse Email', result['abuseEmail']])
table.add_row(['Address 1', result['address1']])
if result.get('address2'):
table.add_row(['Address 2', result['address2']])
table.add_row(['City', result['city']])
table.add_row(['State', result.get('state', '-')])
table.add_row(['Postal Code', result.get('postalCode', '-')])
table.add_row(['Country', result['country']])
return table
def execute(client, args):
mgr = NetworkManager(client)
ip = mgr.ip_lookup(args['<ip>'])
if not ip:
return 'Not found'
t = KeyValueTable(['Name', 'Value'])
t.align['Name'] = 'r'
t.align['Value'] = 'l'
t.add_row(['id', ip['id']])
t.add_row(['ip', ip['ipAddress']])
subnet_table = KeyValueTable(['Name', 'Value'])
subnet_table.align['Name'] = 'r'
subnet_table.align['Value'] = 'l'
subnet_table.add_row(['id', ip['subnet']['id']])
subnet_table.add_row(['identifier', ip['subnet']['networkIdentifier']])
subnet_table.add_row(['netmask', ip['subnet']['netmask']])
if ip['subnet'].get('gateway'):
subnet_table.add_row(['gateway', ip['subnet']['gateway']])
subnet_table.add_row(['type', ip['subnet'].get('subnetType')])
t.add_row(['subnet', subnet_table])
if ip.get('virtualGuest') or ip.get('hardware'):
device_table = KeyValueTable(['Name', 'Value'])
device_table.align['Name'] = 'r'
device_table.align['Value'] = 'l'
if ip.get('virtualGuest'):
device = ip['virtualGuest']
device_type = 'cci'
else:
device = ip['hardware']
device_type = 'server'
device_table.add_row(['id', device['id']])
device_table.add_row(['name', device['fullyQualifiedDomainName']])
device_table.add_row(['type', device_type])
t.add_row(['device', device_table])
return t
def execute(self, args):
mgr = NetworkManager(self.client)
datacenters = mgr.summary_by_datacenter()
t = Table([
'datacenter', 'vlans', 'subnets', 'IPs', 'networking',
'hardware', 'ccis'
])
t.sortby = args.get('--sortby') or 'datacenter'
for name, dc in datacenters.iteritems():
t.add_row([
name,
dc['vlanCount'],
dc['subnetCount'],
dc['primaryIpCount'],
dc['networkingCount'],
dc['hardwareCount'],
dc['virtualGuestCount'],
])
return t
def execute(self, args):
mgr = NetworkManager(self.client)
table = Table([
'id', 'ip', 'assigned', 'target'
])
table.sortby = args.get('--sortby') or 'id'
version = 0
if args.get('--v4'):
version = 4
elif args.get('--v6'):
version = 6
ips = mgr.list_global_ips(version=version)
for ip_address in ips:
assigned = 'No'
target = 'None'
if ip_address.get('destinationIpAddress'):
dest = ip_address['destinationIpAddress']
assigned = 'Yes'
target = dest['ipAddress']
virtual_guest = dest.get('virtualGuest')
if virtual_guest:
target += (' (%s)'
% virtual_guest['fullyQualifiedDomainName'])
elif ip_address['destinationIpAddress'].get('hardware'):
target += ' (' + \
dest['hardware']['fullyQualifiedDomainName'] + \
')'
table.add_row([ip_address['id'],
ip_address['ipAddress']['ipAddress'],
assigned,
target])
return table
def setUp(self):
self.client = FixtureClient()
self.network = NetworkManager(self.client)
def get_network_manager():
return NetworkManager(get_client())
class NetworkTests(unittest.TestCase):
def setUp(self):
self.client = FixtureClient()
self.network = NetworkManager(self.client)
def test_ip_lookup(self):
service = self.client['Network_Subnet_IpAddress']
self.network.ip_lookup('10.0.1.37')
service.getByIpAddress.assert_called_with('10.0.1.37', mask=ANY)
def test_add_subnet_raises_exception_on_failure(self):
self.assertRaises(TypeError, self.network.add_subnet, ('bad'))
def test_add_global_ip(self):
# Test a global IPv4 order
result = self.network.add_global_ip(test_order=True)
self.assertEqual(Product_Order.verifyOrder, result)
def test_add_subnet_for_ipv4(self):
# Test a four public address IPv4 order
result = self.network.add_subnet('public',
quantity=4,
vlan_id=1234,
version=4,
test_order=True)
self.assertEqual(Product_Order.verifyOrder, result)
result = self.network.add_subnet('public',
quantity=4,
vlan_id=1234,
version=4,
test_order=False)
self.assertEqual(Product_Order.verifyOrder, result)
result = self.network.add_subnet('global',
test_order=True)
self.assertEqual(Product_Order.verifyOrder, result)
def test_add_subnet_for_ipv6(self):
# Test a public IPv6 order
result = self.network.add_subnet('public',
quantity=64,
vlan_id=45678,
version=6,
test_order=True)
self.assertEqual(Product_Order.verifyOrder, result)
# Test a global IPv6 order
result = self.network.add_subnet('global',
version=6,
test_order=True)
self.assertEqual(Product_Order.verifyOrder, result)
def test_assign_global_ip(self):
self.network.assign_global_ip(9876, '172.16.24.76')
service = self.client['Network_Subnet_IpAddress_Global']
service.route.assert_called_with('172.16.24.76', id=9876)
def test_cancel_global_ip(self):
self.network.cancel_global_ip(1234)
service = self.client['Billing_Item']
service.cancelService.assert_called_with(id=1234)
def test_cancel_subnet(self):
self.network.cancel_subnet(1234)
service = self.client['Billing_Item']
service.cancelService.assert_called_with(id=1056)
def test_edit_rwhois(self):
self.network.edit_rwhois(
abuse_email='*****@*****.**',
address1='123 Test Street',
address2='Apt. #31',
city='Anywhere',
company_name='TestLayer',
country='US',
first_name='Bob',
last_name='Bobinson',
postal_code='9ba62',
private_residence=False,
state='TX')
expected = {
'abuseEmail': '*****@*****.**',
'address1': '123 Test Street',
'address2': 'Apt. #31',
'city': 'Anywhere',
'companyName': 'TestLayer',
'country': 'US',
'firstName': 'Bob',
'lastName': 'Bobinson',
'postalCode': '9ba62',
'privateResidenceFlag': False,
'state': 'TX',
}
f = self.client['Network_Subnet_Rwhois_Data'].editObject
f.assert_called_with(expected, id='id')
def test_get_rwhois(self):
self.network.get_rwhois()
self.client['Account'].getRwhoisData.assert_called()
def test_get_subnet(self):
mcall = call(id=9876, mask=ANY)
service = self.client['Network_Subnet']
self.network.get_subnet(9876)
service.getObject.assert_has_calls(mcall)
def test_get_vlan(self):
service = self.client['Network_Vlan']
self.network.get_vlan(1234)
service.getObject.assert_has_calls(call(id=1234, mask=ANY))
def test_list_global_ips_default(self):
self.network.list_global_ips()
mask = 'destinationIpAddress[hardware, virtualGuest],ipAddress'
service = self.client['Account']
service.getGlobalIpRecords.assert_has_calls(call(filter={}, mask=mask))
def test_list_global_ips_with_filter(self):
self.network.list_global_ips(version=4)
mask = 'destinationIpAddress[hardware, virtualGuest],ipAddress'
_filter = {
'globalIpRecords': {
'ipAddress': {
'subnet': {
'version': {'operation': 4},
}
}
}
}
service = self.client['Account']
service.getGlobalIpRecords.assert_has_calls(call(filter=_filter,
mask=mask))
def test_list_subnets_default(self):
_filter = {'subnets': {'subnetType': {'operation': '!= GLOBAL_IP'}}}
mask = 'hardware,datacenter,ipAddressCount,virtualGuests'
service = self.client['Account']
self.network.list_subnets()
service.getSubnets.assert_has_calls(call(filter=_filter, mask=mask))
def test_list_subnets_with_filters(self):
result = self.network.list_subnets(
identifier='10.0.0.1',
datacenter='dal00',
subnet_type='PRIMARY',
version=4,
)
_filter = {
'subnets': {
'datacenter': {
'name': {'operation': '_= dal00'}
},
'version': {'operation': 4},
'subnetType': {'operation': '_= PRIMARY'},
'networkIdentifier': {'operation': '_= 10.0.0.1'}
}
}
mask = 'hardware,datacenter,ipAddressCount,virtualGuests'
self.client['Account'].getSubnets.assert_called_with(filter=_filter,
mask=mask)
self.assertEqual(self.client['Account'].getSubnets.return_value,
result)
def test_list_vlans_default(self):
service = self.client['Account']
self.network.list_vlans()
service.getNetworkVlans.assert_has_calls(call(filter={}, mask=ANY))
def test_list_vlans_with_filters(self):
self.network.list_vlans(
vlan_number=5,
datacenter='dal00',
name='primary-vlan',
)
service = self.client['Account']
service.getNetworkVlans.assert_has_calls(call(
filter={
'networkVlans': {
'primaryRouter': {
'datacenter': {
'name': {'operation': '_= dal00'}},
},
'vlanNumber': {'operation': 5},
'name': {'operation': '_= primary-vlan'},
},
},
mask=ANY
))
def test_summary_by_datacenter(self):
result = self.network.summary_by_datacenter()
expected = {
'dal00': {
'hardwareCount': 1,
'networkingCount': 1,
'primaryIpCount': 3,
'subnetCount': 0,
'virtualGuestCount': 1,
'vlanCount': 1
}}
self.assertEqual(expected, result)
def test_resolve_global_ip_ids(self):
service = self.client['Account']
_id = self.network.resolve_global_ip_ids('10.0.0.1')
self.assertEqual(_id, ['200'])
service.getGlobalIpRecords.return_value = []
_id = self.network.resolve_global_ip_ids('nope')
self.assertEqual(_id, [])
def test_resolve_subnet_ids(self):
service = self.client['Account']
_id = self.network.resolve_subnet_ids('10.0.0.1/29')
self.assertEqual(_id, ['100'])
service.getSubnets.return_value = []
_id = self.network.resolve_subnet_ids('nope')
self.assertEqual(_id, [])
def test_resolve_vlan_ids(self):
service = self.client['Account']
service.getNetworkVlans.side_effect = [[{'id': '100'}], []]
_id = self.network.resolve_vlan_ids('vlan_name')
self.assertEqual(_id, ['100'])
_id = self.network.resolve_vlan_ids('nope')
self.assertEqual(_id, [])
def test_unassign_global_ip(self):
self.network.unassign_global_ip(9876)
service = self.client['Network_Subnet_IpAddress_Global']
service.unroute.assert_called_with(id=9876)
def listVlans():
netManager = NetworkManager(client)
vlans = netManager.list_vlans()
print json.dumps(vlans, sort_keys=True, indent=4)
return vlans
class NetworkTests(unittest.TestCase):
def setUp(self):
self.client = MagicMock()
self.network = NetworkManager(self.client)
def test_ip_lookup(self):
ip = '10.0.1.37'
mcall = call(ip, mask=ANY)
service = self.client['Network_Subnet_IpAddress']
self.network.ip_lookup(ip)
service.getByIpAddress.assert_has_calls(mcall)
def test_add_subnet_returns_none_on_failure(self):
self._setup_add_subnet_mocks()
self.assertEqual(None, self.network.add_subnet('bad'))
def test_add_global_ip(self):
self._setup_add_subnet_mocks()
# Test a global IPv4 order
expected = {'packageId': 0,
'prices': [{
'categories': [{
'categoryCode': 'global_ipv4'}],
'id': 11,
'item': {'capacity': '0',
'description': 'Global IPv4',
'id': 10},
'itemId': 10,
'recurringFee': '0'}]}
result = self.network.add_global_ip(test_order=True)
self.assertEqual(expected, result)
def test_add_subnet_for_ipv4(self):
self._setup_add_subnet_mocks()
# Test a four public address IPv4 order
expected = {'packageId': 0,
'prices': [{
'categories': [{
'categoryCode': 'sov_sec_ip_addresses_pub'}],
'id': 4444,
'item': {
'capacity': '4',
'description': '4 Portable Public IP Addresses',
'id': 4440},
'itemId': 4440,
'recurringFee': '0'}]}
result = self.network.add_subnet('public',
quantity=4,
vlan_id=1234,
version=4,
test_order=True)
self.assertEqual(expected, result)
# Test a global IPv4 order
expected = {'packageId': 0,
'prices': [{
'categories': [{
'categoryCode': 'global_ipv4'}],
'id': 11,
'item': {'capacity': '0',
'description': 'Global IPv4',
'id': 10},
'itemId': 10,
'recurringFee': '0'}]}
result = self.network.add_subnet('global',
test_order=True)
self.assertEqual(expected, result)
def test_add_subnet_for_ipv6(self):
self._setup_add_subnet_mocks()
# Test a public IPv6 order
expected = {
'packageId': 0,
'prices': [{
'categories': [{'categoryCode': 'static_ipv6_addresses'}],
'id': 664641,
'item': {
'capacity': '64',
'description': '/64 Block Portable Public IPv6 Addresses',
'id': 66464},
'itemId': 66464,
'recurringFee': '0'}]}
result = self.network.add_subnet('public',
quantity=64,
vlan_id=45678,
version=6,
test_order=True)
self.assertEqual(expected, result)
# Test a global IPv6 order
expected = {'packageId': 0,
'prices': [{
'categories': [{
'categoryCode': 'global_ipv6'}],
'id': 611,
'item': {'capacity': '0',
'description': 'Global IPv6',
'id': 610},
'itemId': 610,
'recurringFee': '0'}]}
result = self.network.add_subnet('global',
version=6,
test_order=True)
self.assertEqual(expected, result)
def test_assign_global_ip(self):
id = 9876
target = '172.16.24.76'
self.network.assign_global_ip(id, target)
service = self.client['Network_Subnet_IpAddress_Global']
service.route.assert_called_with(target, id=id)
def test_cancel_global_ip(self):
id = 9876
mcall = call(id=1056)
service = self.client['Billing_Item']
self.client['Network_Subnet'].getObject.return_value = {
'id': id,
'billingItem': {'id': 1056}
}
self.network.cancel_global_ip(id)
service.cancelService.assert_has_calls(mcall)
def test_cancel_subnet(self):
id = 9876
mcall = call(id=1056)
service = self.client['Billing_Item']
self.client['Network_Subnet'].getObject.return_value = {
'id': id,
'billingItem': {'id': 1056}
}
self.network.cancel_subnet(id)
service.cancelService.assert_has_calls(mcall)
def test_edit_rwhois(self):
self.client['Account'].getRwhoisData.return_value = {'id': 954}
expected = {
'abuseEmail': '*****@*****.**',
'address1': '123 Test Street',
'address2': 'Apt. #31',
'city': 'Anywhere',
'companyName': 'TestLayer',
'country': 'US',
'firstName': 'Bob',
'lastName': 'Bobinson',
'postalCode': '9ba62',
'privateResidenceFlag': False,
'state': 'TX',
}
self.network.edit_rwhois(
abuse_email='*****@*****.**',
address1='123 Test Street',
address2='Apt. #31',
city='Anywhere',
company_name='TestLayer',
country='US',
first_name='Bob',
last_name='Bobinson',
postal_code='9ba62',
private_residence=False,
state='TX')
f = self.client['Network_Subnet_Rwhois_Data'].editObject
f.assert_called_with(expected, id=954)
def test_get_rwhois(self):
self.network.get_rwhois()
self.client['Account'].getRwhoisData.assert_called()
def test_get_subnet(self):
id = 9876
mcall = call(id=id, mask=ANY)
service = self.client['Network_Subnet']
self.network.get_subnet(id)
service.getObject.assert_has_calls(mcall)
def test_get_vlan(self):
id = 1234
mcall = call(id=id, mask=ANY)
service = self.client['Network_Vlan']
self.network.get_vlan(id)
service.getObject.assert_has_calls(mcall)
def test_list_global_ips_default(self):
mask = 'mask[destinationIpAddress[hardware, virtualGuest],ipAddress]'
mcall = call(filter={}, mask=mask)
service = self.client['Account']
self.network.list_global_ips()
service.getGlobalIpRecords.assert_has_calls(mcall)
def test_list_global_ips_with_filter(self):
mask = 'mask[destinationIpAddress[hardware, virtualGuest],ipAddress]'
_filter = {
'globalIpRecords': {
'ipAddress': {
'subnet': {
'version': {'operation': 4},
}
}
}
}
mcall = call(filter=_filter, mask=mask)
service = self.client['Account']
self.network.list_global_ips(version=4)
service.getGlobalIpRecords.assert_has_calls(mcall)
def test_list_subnets_default(self):
_filter = {'subnets': {'subnetType': {'operation': 'not null'}}}
mask = 'mask[hardware,datacenter,ipAddressCount,virtualGuests]'
mcall = call(filter=_filter,
mask=mask)
service = self.client['Account']
self.network.list_subnets()
service.getSubnets.assert_has_calls(mcall)
def test_list_subnets_with_filters(self):
identifier = '10.0.0.1'
datacenter = 'dal00'
subnet_type = 'PRIMARY'
version = 4
service = self.client['Account']
service.getSubnets.return_value = [
{
'id': 100,
'networkIdentifier': '10.0.0.1',
'datacenter': {'name': 'dal00'},
'version': 4,
'subnetType': 'PRIMARY',
},
]
result = self.network.list_subnets(
identifier=identifier,
datacenter=datacenter,
subnet_type=subnet_type,
version=version,
)
_filter = {
'subnets': {
'datacenter': {
'name': {'operation': '_= dal00'}
},
'version': {'operation': 4},
'subnetType': {'operation': '_= PRIMARY'},
'networkIdentifier': {'operation': '_= 10.0.0.1'}
}
}
mask = 'mask[hardware,datacenter,ipAddressCount,virtualGuests]'
service.getSubnets.assert_called_with(filter=_filter, mask=mask)
self.assertEqual([service.getSubnets.return_value[0]], result)
def test_list_vlans_default(self):
mcall = call(filter={}, mask=ANY)
service = self.client['Account']
self.network.list_vlans()
service.getNetworkVlans.assert_has_calls(mcall)
def test_list_vlans_with_filters(self):
number = 5
datacenter = 'dal00'
self.network.list_vlans(
vlan_number=number,
datacenter=datacenter,
)
service = self.client['Account']
service.getNetworkVlans.assert_has_calls(call(
filter={
'networkVlans': {
'primaryRouter': {
'datacenter': {
'name': {'operation': '_= ' + datacenter}},
},
'vlanNumber': {'operation': number},
},
},
mask=ANY
))
def test_summary_by_datacenter(self):
mcall = call(mask=ANY)
service = self.client['Account']
service.getNetworkVlans.return_value = [
{
'name': 'dal00',
'hardware': [{'id': 1}],
'networkComponents': [{'id': 2}],
'primaryRouter': {
'datacenter': {'name': 'dal00'}
},
'totalPrimaryIpAddressCount': 3,
'subnets': [],
'virtualGuests': [{'id': 3}]
}
]
expected = {'dal00': {
'hardwareCount': 1,
'networkingCount': 1,
'primaryIpCount': 3,
'subnetCount': 0,
'virtualGuestCount': 1,
'vlanCount': 1
}}
result = self.network.summary_by_datacenter()
service.getNetworkVlans.assert_has_calls(mcall)
self.assertEqual(expected, result)
def test_resolve_global_ip_ids(self):
service = self.client['Account']
service.getGlobalIpRecords.side_effect = [[
{
'id': '200',
'ipAddress': {
'subnet': {
'networkIdentifier': '10.0.0.1',
},
},
},
], []]
_id = self.network.resolve_global_ip_ids('10.0.0.1')
self.assertEqual(_id, '200')
_id = self.network.resolve_global_ip_ids('nope')
self.assertEqual(_id, None)
def test_resolve_subnet_ids(self):
service = self.client['Account']
service.getSubnets.side_effect = [[
{
'id': '100',
'networkIdentifier': '10.0.0.1',
'datacenter': {'name': 'dal00'},
'version': 4,
'subnetType': 'PRIMARY'
},
], []]
_id = self.network.resolve_subnet_ids('10.0.0.1/29')
self.assertEqual(_id, '100')
_id = self.network.resolve_subnet_ids('nope')
self.assertEqual(_id, None)
def test_unassign_global_ip(self):
id = 9876
self.network.unassign_global_ip(id)
service = self.client['Network_Subnet_IpAddress_Global']
service.unroute.assert_called_with(id=id)
def _setup_add_subnet_mocks(self):
package_mock = self.client['Product_Package']
package_mock.getItems.return_value = [
{
'id': 4440,
'capacity': '4',
'description': '4 Portable Public IP Addresses',
'itemCategory': {'categoryCode': 'sov_sec_ip_addresses_pub'},
'prices': [{'id': 4444}],
},
{
'id': 8880,
'capacity': '8',
'description': '8 Portable Public IP Addresses',
'itemCategory': {'categoryCode': 'sov_sec_ip_addresses_pub'},
'prices': [{'id': 8888}],
},
{
'id': 44400,
'capacity': '4',
'description': '4 Portable Private IP Addresses',
'itemCategory': {'categoryCode': 'sov_sec_ip_addresses_priv'},
'prices': [{'id': 44441}],
},
{
'id': 88800,
'capacity': '8',
'description': '8 Portable Private IP Addresses',
'itemCategory': {'categoryCode': 'sov_sec_ip_addresses_priv'},
'prices': [{'id': 88881}],
},
{
'id': 10,
'capacity': '0',
'description': 'Global IPv4',
'itemCategory': {'categoryCode': 'global_ipv4'},
'prices': [{'id': 11}],
},
{
'id': 66464,
'capacity': '64',
'description': '/64 Block Portable Public IPv6 Addresses',
'itemCategory': {'categoryCode': 'static_ipv6_addresses'},
'prices': [{'id': 664641}],
},
{
'id': 610,
'capacity': '0',
'description': 'Global IPv6',
'itemCategory': {'categoryCode': 'global_ipv6'},
'prices': [{'id': 611}],
},
]
def vlan_return_mock(id, mask):
return {'primaryRouter': {'datacenter': {'id': id * 10}}}
vlan_mock = self.client['Network_Vlan']
vlan_mock.getObject.side_effect = vlan_return_mock
def order_return_mock(order):
mock_item = {}
for item in package_mock.getItems.return_value:
if item['prices'][0]['id'] == order['prices'][0]['id']:
mock_item = item
result = {
'packageId': 0,
'prices': [
{
'itemId': mock_item['id'],
'recurringFee': '0',
'id': mock_item['prices'][0]['id'],
'item': {
'capacity': mock_item['capacity'],
'description': mock_item['description'],
'id': mock_item['id']
},
'categories': [{
'categoryCode':
mock_item['itemCategory']['categoryCode']
}],
}
],
}
return result
order_mock = self.client['Product_Order']
order_mock.verifyOrder.side_effect = order_return_mock
def setUp(self):
self.client = MagicMock()
self.network = NetworkManager(self.client)
def listVlans():
netManager = NetworkManager(client)
vlans = netManager.list_vlans()
print json.dumps(vlans, sort_keys=True, indent=4)
return vlans
class NetworkTests(unittest.TestCase):
def setUp(self):
self.client = FixtureClient()
self.network = NetworkManager(self.client)
def test_ip_lookup(self):
service = self.client["Network_Subnet_IpAddress"]
self.network.ip_lookup("10.0.1.37")
service.getByIpAddress.assert_called_with("10.0.1.37", mask=ANY)
def test_add_subnet_raises_exception_on_failure(self):
self.assertRaises(TypeError, self.network.add_subnet, ("bad"))
def test_add_global_ip(self):
# Test a global IPv4 order
result = self.network.add_global_ip(test_order=True)
self.assertEqual(Product_Order.verifyOrder, result)
def test_add_subnet_for_ipv4(self):
# Test a four public address IPv4 order
result = self.network.add_subnet("public", quantity=4, vlan_id=1234, version=4, test_order=True)
self.assertEqual(Product_Order.verifyOrder, result)
result = self.network.add_subnet("public", quantity=4, vlan_id=1234, version=4, test_order=False)
self.assertEqual(Product_Order.verifyOrder, result)
result = self.network.add_subnet("global", test_order=True)
self.assertEqual(Product_Order.verifyOrder, result)
def test_add_subnet_for_ipv6(self):
# Test a public IPv6 order
result = self.network.add_subnet("public", quantity=64, vlan_id=45678, version=6, test_order=True)
self.assertEqual(Product_Order.verifyOrder, result)
# Test a global IPv6 order
result = self.network.add_subnet("global", version=6, test_order=True)
self.assertEqual(Product_Order.verifyOrder, result)
def test_assign_global_ip(self):
id = 9876
target = "172.16.24.76"
self.network.assign_global_ip(id, target)
service = self.client["Network_Subnet_IpAddress_Global"]
service.route.assert_called_with(target, id=id)
def test_cancel_global_ip(self):
service = self.client["Billing_Item"]
self.network.cancel_global_ip(1234)
service.cancelService.assert_called_with(id=1234)
def test_cancel_subnet(self):
service = self.client["Billing_Item"]
self.network.cancel_subnet(1234)
service.cancelService.assert_called_with(id=1056)
def test_edit_rwhois(self):
self.client["Account"].getRwhoisData.return_value = {"id": 954}
expected = {
"abuseEmail": "*****@*****.**",
"address1": "123 Test Street",
"address2": "Apt. #31",
"city": "Anywhere",
"companyName": "TestLayer",
"country": "US",
"firstName": "Bob",
"lastName": "Bobinson",
"postalCode": "9ba62",
"privateResidenceFlag": False,
"state": "TX",
}
self.network.edit_rwhois(
abuse_email="*****@*****.**",
address1="123 Test Street",
address2="Apt. #31",
city="Anywhere",
company_name="TestLayer",
country="US",
first_name="Bob",
last_name="Bobinson",
postal_code="9ba62",
private_residence=False,
state="TX",
)
f = self.client["Network_Subnet_Rwhois_Data"].editObject
f.assert_called_with(expected, id=954)
def test_get_rwhois(self):
self.network.get_rwhois()
self.client["Account"].getRwhoisData.assert_called()
def test_get_subnet(self):
id = 9876
mcall = call(id=id, mask=ANY)
service = self.client["Network_Subnet"]
self.network.get_subnet(id)
service.getObject.assert_has_calls(mcall)
def test_get_vlan(self):
id = 1234
mcall = call(id=id, mask=ANY)
service = self.client["Network_Vlan"]
self.network.get_vlan(id)
service.getObject.assert_has_calls(mcall)
def test_list_global_ips_default(self):
mask = "destinationIpAddress[hardware, virtualGuest],ipAddress"
mcall = call(filter={}, mask=mask)
service = self.client["Account"]
self.network.list_global_ips()
service.getGlobalIpRecords.assert_has_calls(mcall)
def test_list_global_ips_with_filter(self):
mask = "destinationIpAddress[hardware, virtualGuest],ipAddress"
_filter = {"globalIpRecords": {"ipAddress": {"subnet": {"version": {"operation": 4}}}}}
mcall = call(filter=_filter, mask=mask)
service = self.client["Account"]
self.network.list_global_ips(version=4)
service.getGlobalIpRecords.assert_has_calls(mcall)
def test_list_subnets_default(self):
_filter = {"subnets": {"subnetType": {"operation": "!= GLOBAL_IP"}}}
mask = "hardware,datacenter,ipAddressCount,virtualGuests"
mcall = call(filter=_filter, mask=mask)
service = self.client["Account"]
self.network.list_subnets()
service.getSubnets.assert_has_calls(mcall)
def test_list_subnets_with_filters(self):
identifier = "10.0.0.1"
datacenter = "dal00"
subnet_type = "PRIMARY"
version = 4
service = self.client["Account"]
service.getSubnets.return_value = [
{
"id": 100,
"networkIdentifier": "10.0.0.1",
"datacenter": {"name": "dal00"},
"version": 4,
"subnetType": "PRIMARY",
}
]
result = self.network.list_subnets(
identifier=identifier, datacenter=datacenter, subnet_type=subnet_type, version=version
)
_filter = {
"subnets": {
"datacenter": {"name": {"operation": "_= dal00"}},
"version": {"operation": 4},
"subnetType": {"operation": "_= PRIMARY"},
"networkIdentifier": {"operation": "_= 10.0.0.1"},
}
}
mask = "hardware,datacenter,ipAddressCount,virtualGuests"
service.getSubnets.assert_called_with(filter=_filter, mask=mask)
self.assertEqual([service.getSubnets.return_value[0]], result)
def test_list_vlans_default(self):
mcall = call(filter={}, mask=ANY)
service = self.client["Account"]
self.network.list_vlans()
service.getNetworkVlans.assert_has_calls(mcall)
def test_list_vlans_with_filters(self):
self.network.list_vlans(vlan_number=5, datacenter="dal00", name="primary-vlan")
service = self.client["Account"]
service.getNetworkVlans.assert_has_calls(
call(
filter={
"networkVlans": {
"primaryRouter": {"datacenter": {"name": {"operation": "_= dal00"}}},
"vlanNumber": {"operation": 5},
"name": {"operation": "_= primary-vlan"},
}
},
mask=ANY,
)
)
def test_summary_by_datacenter(self):
mcall = call(mask=ANY, filter=ANY)
service = self.client["Account"]
service.getNetworkVlans.return_value = [
{
"name": "dal00",
"hardware": [{"id": 1}],
"networkComponents": [{"id": 2}],
"primaryRouter": {"datacenter": {"name": "dal00"}},
"totalPrimaryIpAddressCount": 3,
"subnets": [],
"virtualGuests": [{"id": 3}],
}
]
expected = {
"dal00": {
"hardwareCount": 1,
"networkingCount": 1,
"primaryIpCount": 3,
"subnetCount": 0,
"virtualGuestCount": 1,
"vlanCount": 1,
}
}
result = self.network.summary_by_datacenter()
service.getNetworkVlans.assert_has_calls(mcall)
self.assertEqual(expected, result)
def test_resolve_global_ip_ids(self):
service = self.client["Account"]
_id = self.network.resolve_global_ip_ids("10.0.0.1")
self.assertEqual(_id, ["200"])
service.getGlobalIpRecords.return_value = []
_id = self.network.resolve_global_ip_ids("nope")
self.assertEqual(_id, [])
def test_resolve_subnet_ids(self):
service = self.client["Account"]
_id = self.network.resolve_subnet_ids("10.0.0.1/29")
self.assertEqual(_id, ["100"])
service.getSubnets.return_value = []
_id = self.network.resolve_subnet_ids("nope")
self.assertEqual(_id, [])
def test_resolve_vlan_ids(self):
service = self.client["Account"]
service.getNetworkVlans.side_effect = [[{"id": "100"}], []]
_id = self.network.resolve_vlan_ids("vlan_name")
self.assertEqual(_id, ["100"])
_id = self.network.resolve_vlan_ids("nope")
self.assertEqual(_id, [])
def test_unassign_global_ip(self):
id = 9876
self.network.unassign_global_ip(id)
service = self.client["Network_Subnet_IpAddress_Global"]
service.unroute.assert_called_with(id=id)
def render(self):
mgr = NetworkManager(get_client())
return render_template('network_widget_summary.html',
stats=mgr.summary_by_datacenter())
