def execute(self, args):
mgr = NetworkManager(self.client)
_type = 'private'
if args['public']:
_type = 'public'
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_subnet(_type,
quantity=args['<quantity>'],
vlan_id=args['<vlan>'],
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['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(client, args):
mgr = NetworkManager(client)
_type = 'private'
if args['public']:
_type = 'public'
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_subnet(type=_type,
quantity=args['<quantity>'],
vlan_id=args['<vlan>'],
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
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
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)
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)