def execute(self, args):
update_with_template_args(args)
mgr = HardwareManager(self.client)
# Disks will be a comma-separated list. Let's make it a real list.
if isinstance(args.get('--disk'), str):
args['--disk'] = args.get('--disk').split(',')
# Do the same thing for SSH keys
if isinstance(args.get('--key'), str):
args['--key'] = args.get('--key').split(',')
self._validate_args(args)
ds_options = mgr.get_dedicated_server_create_options(args['--chassis'])
order = {
'hostname': args['--hostname'],
'domain': args['--domain'],
'bare_metal': False,
'package_id': args['--chassis'],
}
# Convert the OS code back into a price ID
os_price = self._get_price_id_from_options(ds_options, 'os',
args['--os'])
if os_price:
order['os'] = os_price
else:
raise CLIAbort('Invalid operating system specified.')
order['location'] = args['--datacenter'] or 'FIRST_AVAILABLE'
order['server'] = args['--cpu']
order['ram'] = self._get_price_id_from_options(ds_options, 'memory',
int(args['--memory']))
# Set the disk sizes
disk_prices = []
disk_number = 0
for disk in args.get('--disk'):
disk_price = self._get_disk_price(ds_options, disk, disk_number)
disk_number += 1
if disk_price:
disk_prices.append(disk_price)
if not disk_prices:
disk_prices.append(self._get_default_value(ds_options, 'disk0'))
order['disks'] = disk_prices
# Set the disk controller price
if args.get('--controller'):
dc_price = self._get_price_id_from_options(
ds_options, 'disk_controller', args.get('--controller'))
else:
dc_price = self._get_price_id_from_options(ds_options,
'disk_controller',
'None')
order['disk_controller'] = dc_price
# Set the port speed
port_speed = args.get('--network') or '100'
nic_price = self._get_price_id_from_options(ds_options, 'nic',
port_speed)
if nic_price:
order['port_speed'] = nic_price
else:
raise CLIAbort('Invalid NIC speed specified.')
if args.get('--postinstall'):
order['post_uri'] = args.get('--postinstall')
# Get the SSH keys
if args.get('--key'):
keys = []
for key in args.get('--key'):
key_id = resolve_id(SshKeyManager(self.client).resolve_ids,
key, 'SshKey')
keys.append(key_id)
order['ssh_keys'] = keys
if args.get('--vlan_public'):
order['public_vlan'] = args['--vlan_public']
if args.get('--vlan_private'):
order['private_vlan'] = args['--vlan_private']
# Do not create hardware server with --test or --export
do_create = not (args['--export'] or args['--test'])
output = None
if args.get('--test'):
result = mgr.verify_order(**order)
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])
output = []
output.append(table)
output.append(FormattedItem(
'',
' -- ! Prices reflected here are retail and do not '
'take account level discounts and are not guaranteed.')
)
if args['--export']:
export_file = args.pop('--export')
export_to_template(export_file, args, exclude=['--wait', '--test'])
return 'Successfully exported options to a template file.'
if do_create:
if args['--really'] or confirm(
"This action will incur charges on your account. "
"Continue?"):
result = mgr.place_order(**order)
table = KeyValueTable(['name', 'value'])
table.align['name'] = 'r'
table.align['value'] = 'l'
table.add_row(['id', result['orderId']])
table.add_row(['created', result['orderDate']])
output = table
else:
raise CLIAbort('Aborting dedicated server order.')
return output
def execute(self, args):
update_with_template_args(args)
mgr = HardwareManager(self.client)
# Disks will be a comma-separated list. Let's make it a real list.
if isinstance(args.get('--disk'), str):
args['--disk'] = args.get('--disk').split(',')
# Do the same thing for SSH keys
if isinstance(args.get('--key'), str):
args['--key'] = args.get('--key').split(',')
self._validate_args(args)
bmi_options = mgr.get_bare_metal_create_options()
order = {
'hostname': args['--hostname'],
'domain': args['--domain'],
'bare_metal': True,
}
# Validate the CPU/Memory combination and get the price ID
server_core = self._get_cpu_and_memory_price_ids(bmi_options,
args['--cpu'],
args['--memory'])
if server_core:
order['server'] = server_core
else:
raise CLIAbort('Invalid CPU/memory combination specified.')
order['hourly'] = args['--hourly']
# Convert the OS code back into a price ID
os_price = self._get_price_id_from_options(bmi_options, 'os',
args['--os'])
if os_price:
order['os'] = os_price
else:
raise CLIAbort('Invalid operating system specified.')
order['location'] = args['--datacenter'] or 'FIRST_AVAILABLE'
# Set the disk size
disk_prices = []
for disk in args.get('--disk'):
disk_price = self._get_price_id_from_options(bmi_options, 'disk',
disk)
if disk_price:
disk_prices.append(disk_price)
if not disk_prices:
disk_prices.append(self._get_default_value(bmi_options, 'disk0'))
order['disks'] = disk_prices
# Set the port speed
port_speed = args.get('--network') or 100
nic_price = self._get_price_id_from_options(bmi_options, 'nic',
str(port_speed))
if nic_price:
order['port_speed'] = nic_price
else:
raise CLIAbort('Invalid network speed specified.')
# Get the SSH keys
if args.get('--key'):
keys = []
for key in args.get('--key'):
key_id = resolve_id(SshKeyManager(self.client).resolve_ids,
key, 'SshKey')
keys.append(key_id)
order['ssh_keys'] = keys
if args.get('--vlan_public'):
order['public_vlan'] = args['--vlan_public']
if args.get('--vlan_private'):
order['private_vlan'] = args['--vlan_private']
# Begin output
table = Table(['Item', 'cost'])
table.align['Item'] = 'r'
table.align['cost'] = 'r'
if args.get('--test'):
result = mgr.verify_order(**order)
total_monthly = 0.0
total_hourly = 0.0
for price in result['prices']:
monthly_fee = float(price.get('recurringFee', 0.0))
hourly_fee = float(price.get('hourlyRecurringFee', 0.0))
total_monthly += monthly_fee
total_hourly += hourly_fee
if args.get('--hourly'):
rate = "%.2f" % hourly_fee
else:
rate = "%.2f" % monthly_fee
table.add_row([price['item']['description'], rate])
if args.get('--hourly'):
total = total_hourly
else:
total = total_monthly
billing_rate = 'monthly'
if args.get('--hourly'):
billing_rate = 'hourly'
table.add_row(['Total %s cost' % billing_rate, "%.2f" % total])
output = SequentialOutput()
output.append(table)
output.append(FormattedItem(
'',
' -- ! Prices reflected here are retail and do not '
'take account level discounts and are not guaranteed.')
)
elif args['--really'] or confirm(
"This action will incur charges on your account. Continue?"):
result = mgr.place_order(**order)
table = KeyValueTable(['name', 'value'])
table.align['name'] = 'r'
table.align['value'] = 'l'
table.add_row(['id', result['orderId']])
table.add_row(['created', result['orderDate']])
output = table
else:
raise CLIAbort('Aborting bare metal instance order.')
return output
class HardwareTests(unittest.TestCase):
def setUp(self):
self.client = FixtureClient()
self.hardware = HardwareManager(self.client)
def test_list_hardware(self):
mcall = call(mask=ANY, filter={})
results = self.hardware.list_hardware()
self.client['Account'].getHardware.assert_has_calls(mcall)
self.assertEqual(results, Account.getHardware)
def test_list_hardware_with_filters(self):
results = self.hardware.list_hardware(
tags=['tag1', 'tag2'],
cpus=2,
memory=1,
hostname='hostname',
domain='example.com',
datacenter='dal05',
nic_speed=100,
public_ip='1.2.3.4',
private_ip='4.3.2.1',
)
self.client['Account'].getHardware.assert_has_calls(call(
filter={
'hardware': {
'datacenter': {'name': {'operation': '_= dal05'}},
'domain': {'operation': '_= example.com'},
'tagReferences': {
'tag': {'name': {
'operation': 'in',
'options': [
{'name': 'data', 'value': ['tag1', 'tag2']}]
}}
},
'memoryCapacity': {'operation': 1},
'processorPhysicalCoreAmount': {'operation': 2},
'hostname': {'operation': '_= hostname'},
'primaryIpAddress': {'operation': '_= 1.2.3.4'},
'networkComponents': {'maxSpeed': {'operation': 100}},
'primaryBackendIpAddress': {'operation': '_= 4.3.2.1'}}
},
mask=ANY,
))
self.assertEqual(results, Account.getHardware)
def test_resolve_ids_ip(self):
_id = self.hardware._get_ids_from_ip('172.16.1.100')
self.assertEqual(_id, [1000, 1001])
_id = self.hardware._get_ids_from_ip('nope')
self.assertEqual(_id, [])
# Now simulate a private IP test
self.client['Account'].getHardware.side_effect = [[], [{'id': 99}]]
_id = self.hardware._get_ids_from_ip('10.0.1.87')
self.assertEqual(_id, [99])
def test_resolve_ids_hostname(self):
_id = self.hardware._get_ids_from_hostname('hardware-test1')
self.assertEqual(_id, [1000, 1001])
def test_get_hardware(self):
result = self.hardware.get_hardware(1000)
self.client['Hardware_Server'].getObject.assert_called_once_with(
id=1000, mask=ANY)
self.assertEqual(Hardware_Server.getObject, result)
def test_reload(self):
post_uri = 'http://test.sftlyr.ws/test.sh'
self.hardware.reload(1, post_uri=post_uri, ssh_keys=[1701])
f = self.client['Hardware_Server'].reloadOperatingSystem
f.assert_called_once_with('FORCE',
{'customProvisionScriptUri': post_uri,
'sshKeyIds': [1701]}, id=1)
def test_get_bare_metal_create_options_returns_none_on_error(self):
self.client['Product_Package'].getAllObjects.return_value = [
{'name': 'No Matching Instances', 'id': 0}]
self.assertIsNone(self.hardware.get_bare_metal_create_options())
def test_get_bare_metal_create_options(self):
package_id = 50
self.hardware.get_bare_metal_create_options()
f1 = self.client['Product_Package'].getRegions
f1.assert_called_once_with(id=package_id)
f2 = self.client['Product_Package'].getConfiguration
f2.assert_called_once_with(id=package_id,
mask='mask[itemCategory[group]]')
f3 = self.client['Product_Package'].getCategories
f3.assert_called_once_with(id=package_id)
def test_generate_create_dict_with_all_bare_metal_options(self):
args = {
'server': 100,
'hostname': 'unicorn',
'domain': 'giggles.woo',
'disks': [500],
'location': 'Wyrmshire',
'os': 200,
'port_speed': 600,
'bare_metal': True,
'hourly': True,
'public_vlan': 10234,
'private_vlan': 20468,
}
expected = {
'hardware': [
{
'domain': 'giggles.woo',
'bareMetalInstanceFlag': True,
'hostname': 'unicorn',
'primaryBackendNetworkComponent':
{'networkVlan': {'id': 20468}},
'primaryNetworkComponent':
{'networkVlan': {'id': 10234}},
}
],
'prices': [
{'id': 100},
{'id': 500},
{'id': 200},
{'id': 600},
{'id': 12000}
],
'useHourlyPricing': True,
'location': 'Wyrmshire', 'packageId': 50
}
data = self.hardware._generate_create_dict(**args)
self.assertEqual(expected, data)
def test_generate_create_dict_with_all_dedicated_server_options(self):
args = {
'server': 100,
'hostname': 'unicorn',
'domain': 'giggles.woo',
'disks': [1000, 1000, 1000, 1000],
'location': 'Wyrmshire',
'os': 200,
'port_speed': 600,
'bare_metal': False,
'package_id': 13,
'ram': 1400,
'disk_controller': 1500,
'ssh_keys': [3000, 3001],
'public_vlan': 10234,
'private_vlan': 20468,
}
expected = {
'hardware': [
{
'domain': 'giggles.woo',
'bareMetalInstanceFlag': False,
'hostname': 'unicorn',
'primaryBackendNetworkComponent':
{'networkVlan': {'id': 20468}},
'primaryNetworkComponent':
{'networkVlan': {'id': 10234}},
}
],
'prices': [
{'id': 100},
{'id': 1000},
{'id': 1000},
{'id': 1000},
{'id': 1000},
{'id': 200},
{'id': 600},
{'id': 1400},
{'id': 1500}],
'sshKeys': [{'sshKeyIds': [3000, 3001]}],
'location': 'Wyrmshire', 'packageId': 13
}
data = self.hardware._generate_create_dict(**args)
self.assertEqual(expected, data)
@patch('SoftLayer.managers.hardware.HardwareManager._generate_create_dict')
def test_verify_order(self, create_dict):
create_dict.return_value = {'test': 1, 'verify': 1}
self.hardware.verify_order(test=1, verify=1)
create_dict.assert_called_once_with(test=1, verify=1)
f = self.client['Product_Order'].verifyOrder
f.assert_called_once_with({'test': 1, 'verify': 1})
@patch('SoftLayer.managers.hardware.HardwareManager._generate_create_dict')
def test_place_order(self, create_dict):
create_dict.return_value = {'test': 1, 'verify': 1}
self.hardware.place_order(test=1, verify=1)
create_dict.assert_called_once_with(test=1, verify=1)
f = self.client['Product_Order'].placeOrder
f.assert_called_once_with({'test': 1, 'verify': 1})
def test_cancel_metal_immediately(self):
b_id = 6327
result = self.hardware.cancel_metal(b_id, immediate=True)
f = self.client['Billing_Item'].cancelService
f.assert_called_once_with(id=b_id)
self.assertEqual(result, Billing_Item.cancelService)
def test_cancel_metal_on_anniversary(self):
b_id = 6327
result = self.hardware.cancel_metal(b_id, False)
f = self.client['Billing_Item'].cancelServiceOnAnniversaryDate
f.assert_called_once_with(id=b_id)
self.assertEqual(result, Billing_Item.cancelServiceOnAnniversaryDate)
def test_cancel_hardware_without_reason(self):
hw_id = 987
result = self.hardware.cancel_hardware(hw_id)
reasons = self.hardware.get_cancellation_reasons()
f = self.client['Ticket'].createCancelServerTicket
f.assert_called_once_with(hw_id, reasons['unneeded'], '', True,
'HARDWARE')
self.assertEqual(result, Ticket.createCancelServerTicket)
def test_cancel_hardware_with_reason_and_comment(self):
hw_id = 987
reason = 'sales'
comment = 'Test Comment'
self.hardware.cancel_hardware(hw_id, reason, comment)
reasons = self.hardware.get_cancellation_reasons()
f = self.client['Ticket'].createCancelServerTicket
f.assert_called_once_with(hw_id, reasons[reason], comment, True,
'HARDWARE')
def test_change_port_speed_public(self):
hw_id = 1
speed = 100
self.hardware.change_port_speed(hw_id, True, speed)
f = self.client['Hardware_Server'].setPublicNetworkInterfaceSpeed
f.assert_called_once_with(speed, id=hw_id)
def test_change_port_speed_private(self):
hw_id = 2
speed = 10
self.hardware.change_port_speed(hw_id, False, speed)
f = self.client['Hardware_Server'].setPrivateNetworkInterfaceSpeed
f.assert_called_once_with(speed, id=hw_id)
def test_get_available_dedicated_server_packages(self):
self.hardware.get_available_dedicated_server_packages()
f = self.client['Product_Package'].getObject
f.assert_has_calls([call(id=13, mask='mask[id, name, description]')])
def test_get_dedicated_server_options(self):
package_id = 13
self.hardware.get_dedicated_server_create_options(package_id)
f1 = self.client['Product_Package'].getRegions
f1.assert_called_once_with(id=package_id)
f2 = self.client['Product_Package'].getConfiguration
f2.assert_called_once_with(id=package_id,
mask='mask[itemCategory[group]]')
f3 = self.client['Product_Package'].getCategories
f3.assert_called_once_with(id=package_id)
def test_get_default_value_returns_none_for_unknown_category(self):
package_options = {'categories': ['Cat1', 'Cat2']}
self.assertEqual(None, get_default_value(package_options,
'Unknown Category'))
def test_get_default_value(self):
price_id = 9876
package_options = {'categories':
{'Cat1': {
'items': [{
'prices': [{
'setupFee': 0,
'recurringFee': 0,
}],
'price_id': price_id,
}]
}}}
self.assertEqual(price_id, get_default_value(package_options, 'Cat1'))
def test_edit(self):
# Test editing user data
self.hardware.edit(100, userdata='my data')
service = self.client['Hardware_Server']
service.setUserMetadata.assert_called_once_with(['my data'], id=100)
# Now test a blank edit
self.assertTrue(self.hardware.edit, 100)
# Finally, test a full edit
args = {
'hostname': 'new-host',
'domain': 'new.sftlyr.ws',
'notes': 'random notes',
}
self.hardware.edit(100, **args)
service.editObject.assert_called_once_with(args, id=100)
def execute(self, args):
update_with_template_args(args)
mgr = HardwareManager(self.client)
# Disks will be a comma-separated list. Let's make it a real list.
if isinstance(args.get("--disk"), str):
args["--disk"] = args.get("--disk").split(",")
# Do the same thing for SSH keys
if isinstance(args.get("--key"), str):
args["--key"] = args.get("--key").split(",")
self._validate_args(args)
bmi_options = mgr.get_bare_metal_create_options()
order = {"hostname": args["--hostname"], "domain": args["--domain"], "bare_metal": True}
# Validate the CPU/Memory combination and get the price ID
server_core = self._get_cpu_and_memory_price_ids(bmi_options, args["--cpu"], args["--memory"])
if server_core:
order["server"] = server_core
else:
raise CLIAbort("Invalid CPU/memory combination specified.")
order["hourly"] = args["--hourly"]
# Convert the OS code back into a price ID
os_price = self._get_price_id_from_options(bmi_options, "os", args["--os"])
if os_price:
order["os"] = os_price
else:
raise CLIAbort("Invalid operating system specified.")
order["location"] = args["--datacenter"] or "FIRST_AVAILABLE"
# Set the disk size
disk_prices = []
for disk in args.get("--disk"):
disk_price = self._get_price_id_from_options(bmi_options, "disk", disk)
if disk_price:
disk_prices.append(disk_price)
if not disk_prices:
disk_prices.append(self._get_default_value(bmi_options, "disk0"))
order["disks"] = disk_prices
# Set the port speed
port_speed = args.get("--network") or 100
nic_price = self._get_price_id_from_options(bmi_options, "nic", str(port_speed))
if nic_price:
order["port_speed"] = nic_price
else:
raise CLIAbort("Invalid network speed specified.")
# Get the SSH keys
if args.get("--key"):
keys = []
for key in args.get("--key"):
key_id = resolve_id(SshKeyManager(self.client).resolve_ids, key, "SshKey")
keys.append(key_id)
order["ssh_keys"] = keys
if args.get("--vlan_public"):
order["public_vlan"] = args["--vlan_public"]
if args.get("--vlan_private"):
order["private_vlan"] = args["--vlan_private"]
# Begin output
t = Table(["Item", "cost"])
t.align["Item"] = "r"
t.align["cost"] = "r"
if args.get("--test"):
result = mgr.verify_order(**order)
total_monthly = 0.0
total_hourly = 0.0
for price in result["prices"]:
monthly_fee = float(price.get("recurringFee", 0.0))
hourly_fee = float(price.get("hourlyRecurringFee", 0.0))
total_monthly += monthly_fee
total_hourly += hourly_fee
if args.get("--hourly"):
rate = "%.2f" % hourly_fee
else:
rate = "%.2f" % monthly_fee
t.add_row([price["item"]["description"], rate])
if args.get("--hourly"):
total = total_hourly
else:
total = total_monthly
billing_rate = "monthly"
if args.get("--hourly"):
billing_rate = "hourly"
t.add_row(["Total %s cost" % billing_rate, "%.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 guaranteed.",
)
)
elif args["--really"] or confirm("This action will incur charges on your account. Continue?"):
result = mgr.place_order(**order)
t = KeyValueTable(["name", "value"])
t.align["name"] = "r"
t.align["value"] = "l"
t.add_row(["id", result["orderId"]])
t.add_row(["created", result["orderDate"]])
output = t
else:
raise CLIAbort("Aborting bare metal instance order.")
return output
def execute(self, args):
update_with_template_args(args)
mgr = HardwareManager(self.client)
# Disks will be a comma-separated list. Let's make it a real list.
if isinstance(args.get('--disk'), str):
args['--disk'] = args.get('--disk').split(',')
# Do the same thing for SSH keys
if isinstance(args.get('--key'), str):
args['--key'] = args.get('--key').split(',')
self._validate_args(args)
ds_options = mgr.get_dedicated_server_create_options(args['--chassis'])
order = self._process_args(args, ds_options)
# Do not create hardware server with --test or --export
do_create = not (args['--export'] or args['--test'])
output = None
if args.get('--test'):
result = mgr.verify_order(**order)
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])
output = []
output.append(table)
output.append(FormattedItem(
'',
' -- ! Prices reflected here are retail and do not '
'take account level discounts and are not guaranteed.')
)
if args['--export']:
export_file = args.pop('--export')
export_to_template(export_file, args, exclude=['--wait', '--test'])
return 'Successfully exported options to a template file.'
if do_create:
if args['--really'] or confirm(
"This action will incur charges on your account. "
"Continue?"):
result = mgr.place_order(**order)
table = KeyValueTable(['name', 'value'])
table.align['name'] = 'r'
table.align['value'] = 'l'
table.add_row(['id', result['orderId']])
table.add_row(['created', result['orderDate']])
output = table
else:
raise CLIAbort('Aborting dedicated server order.')
return output
