""" SoftLayer.tests.CLI.modules.order_tests ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ :license: MIT, see LICENSE for more details. """ import json import sys import tempfile from unittest import mock as mock from SoftLayer.CLI import exceptions from SoftLayer import testing class OrderTests(testing.TestCase): def test_category_list(self): cat1 = {'itemCategory': {'name': 'cat1', 'categoryCode': 'code1'}, 'isRequired': 1} cat2 = {'itemCategory': {'name': 'cat2', 'categoryCode': 'code2'}, 'isRequired': 0} p_mock = self.set_mock('SoftLayer_Product_Package', 'getConfiguration') p_mock.return_value = [cat1, cat2] result = self.run_command(['order', 'category-list', 'package']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Package', 'getConfiguration') self.assertEqual([{'name': 'cat1', 'categoryCode': 'code1', 'isRequired': 'Y'}, {'name': 'cat2', 'categoryCode': 'code2', 'isRequired': 'N'}], json.loads(result.output)) def test_item_list(self): category = {'categoryCode': 'testing'} item1 = {'keyName': 'item1', 'description': 'description1', 'itemCategory': category} item2 = {'keyName': 'item2', 'description': 'description2', 'itemCategory': category} p_mock = self.set_mock('SoftLayer_Product_Package', 'getItems') p_mock.return_value = [item1, item2] result = self.run_command(['order', 'item-list', 'package']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Package', 'getItems') self.assertIn('description2', result.output) self.assertIn('testing', result.output) self.assertIn('item2', result.output) def test_item_list_prices(self): result = self.run_command(['order', 'item-list', 'package', '--prices']) self.assert_no_fail(result) output = json.loads(result.output) self.assertEqual(output[0][0]['Price Id'], 1007) self.assertEqual(output[0][1]['Restriction'], '- - - -') self.assertEqual(output[0][1]['KeyName'], 'KeyName015') self.assert_called_with('SoftLayer_Product_Package', 'getItems') def test_item_list_location_keyname(self): result = self.run_command(['order', 'item-list', 'package', '--prices', 'DALLAS13', ]) self.assert_no_fail(result) output = json.loads(result.output) self.assertEqual(output[0][0]['Hourly'], 0.0) self.assertEqual(output[0][1]['KeyName'], 'KeyName015') self.assertEqual(output[0][1]['Price Id'], 1144) self.assert_called_with('SoftLayer_Product_Package', 'getItemPrices') def test_item_list_location_name(self): result = self.run_command(['order', 'item-list', 'package', '--prices', 'dal13', ]) self.assert_no_fail(result) output = json.loads(result.output) self.assertEqual(output[0][0]['Hourly'], 0.0) self.assertEqual(output[0][1]['KeyName'], 'KeyName015') self.assertEqual(output[0][1]['Price Id'], 1144) self.assert_called_with('SoftLayer_Product_Package', 'getItemPrices') def test_item_list_category_keyword(self): result = self.run_command(['order', 'item-list', 'package', '--prices', 'dal13', '-c', 'os', '-k' 'test']) self.assert_no_fail(result) output = json.loads(result.output) self.assertEqual(output[0][0]['Hourly'], 0.0) self.assertEqual(output[0][1]['KeyName'], 'KeyName015') self.assertEqual(output[0][1]['Price Id'], 1144) self.assert_called_with('SoftLayer_Product_Package', 'getItemPrices') def test_package_list(self): p_mock = self.set_mock('SoftLayer_Product_Package', 'getAllObjects') p_mock.return_value = _get_all_packages() _filter = {'type': {'keyName': {'operation': '!= BLUEMIX_SERVICE'}}} result = self.run_command(['order', 'package-list']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Package', 'getAllObjects', filter=_filter) expected_results = [{'id': 1, 'name': 'package1', 'keyName': 'PACKAGE1', 'type': 'BARE_METAL_CPU'}, {'id': 2, 'name': 'package2', 'keyName': 'PACKAGE2', 'type': 'BARE_METAL_CPU'}] self.assertEqual(expected_results, json.loads(result.output)) def test_package_list_keyword(self): p_mock = self.set_mock('SoftLayer_Product_Package', 'getAllObjects') p_mock.return_value = _get_all_packages() _filter = {'type': {'keyName': {'operation': '!= BLUEMIX_SERVICE'}}} _filter['name'] = {'operation': '*= package1'} result = self.run_command(['order', 'package-list', '--keyword', 'package1']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Package', 'getAllObjects', filter=_filter) expected_results = [{'id': 1, 'name': 'package1', 'keyName': 'PACKAGE1', 'type': 'BARE_METAL_CPU'}, {'id': 2, 'name': 'package2', 'keyName': 'PACKAGE2', 'type': 'BARE_METAL_CPU'}] self.assertEqual(expected_results, json.loads(result.output)) def test_package_list_type(self): p_mock = self.set_mock('SoftLayer_Product_Package', 'getAllObjects') p_mock.return_value = _get_all_packages() _filter = {'type': {'keyName': {'operation': 'BARE_METAL_CPU'}}} result = self.run_command(['order', 'package-list', '--package_type', 'BARE_METAL_CPU']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Package', 'getAllObjects', filter=_filter) expected_results = [{'id': 1, 'name': 'package1', 'keyName': 'PACKAGE1', 'type': 'BARE_METAL_CPU'}, {'id': 2, 'name': 'package2', 'keyName': 'PACKAGE2', 'type': 'BARE_METAL_CPU'}] self.assertEqual(expected_results, json.loads(result.output)) def test_place(self): order_date = '2017-04-04 07:39:20' order = {'orderId': 1234, 'orderDate': order_date, 'placedOrder': {'status': 'APPROVED'}} verify_mock = self.set_mock('SoftLayer_Product_Order', 'verifyOrder') place_mock = self.set_mock('SoftLayer_Product_Order', 'placeOrder') items_mock = self.set_mock('SoftLayer_Product_Package', 'getItems') verify_mock.return_value = self._get_verified_order_return() place_mock.return_value = order items_mock.return_value = self._get_order_items() result = self.run_command(['-y', 'order', 'place', 'package', 'DALLAS13', 'ITEM1', '--complex-type', 'SoftLayer_Container_Product_Order_Thing']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Order', 'placeOrder') self.assertNotIn('Warning: Closed soon: dal13', result.output) self.assertIn('"status": "APPROVED"', result.output) @mock.patch('SoftLayer.managers.network.NetworkManager.get_datacenter_by_keyname') def test_place_with_quantity(self, keyname_mock): keyname_mock.return_value = {"name": "ams01"} order_date = '2017-04-04 07:39:20' order = {'orderId': 1234, 'orderDate': order_date, 'placedOrder': {'status': 'APPROVED'}} verify_mock = self.set_mock('SoftLayer_Product_Order', 'verifyOrder') place_mock = self.set_mock('SoftLayer_Product_Order', 'placeOrder') items_mock = self.set_mock('SoftLayer_Product_Package', 'getItems') verify_mock.return_value = self._get_verified_order_return() place_mock.return_value = order items_mock.return_value = self._get_order_items() result = self.run_command(['-y', 'order', 'place', '--quantity=2', 'package', 'AMSTERDAM', 'ITEM1', '--complex-type', 'SoftLayer_Container_Product_Order_Thing']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Order', 'placeOrder') self.assertIn('Warning: Closed soon: ams01.pod01', result.output) self.assertIn('"status": "APPROVED"', result.output) def test_place_extras_parameter_fail(self): result = self.run_command(['-y', 'order', 'place', 'package', 'DALLAS13', 'ITEM1', '--extras', '{"device":[']) self.assertEqual(result.exit_code, 2) self.assertIsInstance(result.exception, exceptions.CLIAbort) def test_place_quote(self): order_date = '2018-04-04 07:39:20' expiration_date = '2018-05-04 07:39:20' quote_name = 'foobar' order = {'orderDate': order_date, 'quote': { 'id': 1234, 'name': quote_name, 'expirationDate': expiration_date, 'status': 'PENDING' }} place_quote_mock = self.set_mock('SoftLayer_Product_Order', 'placeQuote') items_mock = self.set_mock('SoftLayer_Product_Package', 'getItems') place_quote_mock.return_value = order items_mock.return_value = self._get_order_items() result = self.run_command(['order', 'place-quote', '--name', 'foobar', 'package', 'DALLAS13', 'ITEM1', '--complex-type', 'SoftLayer_Container_Product_Order_Thing']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Order', 'placeQuote') self.assertEqual({'id': 1234, 'name': quote_name, 'created': order_date, 'expires': expiration_date, 'status': 'PENDING'}, json.loads(result.output)) def test_place_quote_extras_parameter_fail(self): result = self.run_command(['-y', 'order', 'place-quote', 'package', 'DALLAS13', 'ITEM1', '--extras', '{"device":[']) self.assertEqual(result.exit_code, 2) self.assertIsInstance(result.exception, exceptions.CLIAbort) def test_verify_hourly(self): order_date = '2017-04-04 07:39:20' order = {'orderId': 1234, 'orderDate': order_date, 'placedOrder': {'status': 'APPROVED'}} verify_mock = self.set_mock('SoftLayer_Product_Order', 'verifyOrder') items_mock = self.set_mock('SoftLayer_Product_Package', 'getItems') order = self._get_verified_order_return() verify_mock.return_value = order items_mock.return_value = self._get_order_items() result = self.run_command(['order', 'place', '--billing', 'hourly', '--verify', '--complex-type', 'SoftLayer_Container_Product_Order_Thing', 'package', 'DALLAS13', 'ITEM1', 'ITEM2']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Order', 'verifyOrder') expected_results = [] for price in order['orderContainers'][0]['prices']: expected_results.append({'keyName': price['item']['keyName'], 'description': price['item']['description'], 'cost': price['hourlyRecurringFee']}) self.assertEqual(expected_results, json.loads(result.output)) def test_verify_monthly(self): order_date = '2017-04-04 07:39:20' order = {'orderId': 1234, 'orderDate': order_date, 'placedOrder': {'status': 'APPROVED'}} verify_mock = self.set_mock('SoftLayer_Product_Order', 'verifyOrder') items_mock = self.set_mock('SoftLayer_Product_Package', 'getItems') order = self._get_verified_order_return() verify_mock.return_value = order items_mock.return_value = self._get_order_items() result = self.run_command(['order', 'place', '--billing', 'monthly', '--verify', '--complex-type', 'SoftLayer_Container_Product_Order_Thing', 'package', 'DALLAS13', 'ITEM1', 'ITEM2']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Order', 'verifyOrder') expected_results = [] for price in order['orderContainers'][0]['prices']: expected_results.append({'keyName': price['item']['keyName'], 'description': price['item']['description'], 'cost': price['recurringFee']}) self.assertEqual(expected_results, json.loads(result.output)) def test_preset_list(self): active_preset1 = {'name': 'active1', 'keyName': 'PRESET1', 'description': 'description1'} active_preset2 = {'name': 'active2', 'keyName': 'PRESET2', 'description': 'description2'} acc_preset = {'name': 'account1', 'keyName': 'PRESET3', 'description': 'description3'} active_mock = self.set_mock('SoftLayer_Product_Package', 'getActivePresets') account_mock = self.set_mock('SoftLayer_Product_Package', 'getAccountRestrictedActivePresets') active_mock.return_value = [active_preset1, active_preset2] account_mock.return_value = [acc_preset] result = self.run_command(['order', 'preset-list', 'package']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Package', 'getActivePresets') self.assert_called_with('SoftLayer_Product_Package', 'getAccountRestrictedActivePresets') self.assertIn('"Key Name": "PRESET1",', result.output) self.assertIn('"Description": "description2"', result.output) self.assertIn('"Name": "account1",', result.output) def test_preset_list_keywork(self): result = self.run_command(['order', 'preset-list', 'package', '--keyword', 'testKeyWord']) _filter = {'activePresets': {'name': {'operation': '*= testKeyWord'}}} self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Package', 'getActivePresets', filter=_filter) def test_preset_list_prices(self): active_preset1 = { 'name': 'active1', 'keyName': 'PRESET1', 'description': 'description1', 'locations': [], 'id': 118822, 'prices': [ {'hourlyRecurringFee': 100, 'recurringFee': 55}, {'hourlyRecurringFee': 110, 'recurringFee': 44}, ] } active_mock = self.set_mock('SoftLayer_Product_Package', 'getActivePresets') active_mock.return_value = [active_preset1] result = self.run_command(['order', 'preset-list', 'package', '--prices']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Product_Package', 'getActivePresets') self.assertIn("210", result.output) self.assertIn("99", result.output) def test_location_list(self): result = self.run_command(['order', 'package-locations', 'package']) self.assert_no_fail(result) expected_results = [ {'id': 2017603, 'dc': 'wdc07', 'description': 'WDC07 - Washington, DC', 'keyName': 'WASHINGTON07', 'Note': 'closed soon: wdc07.pod01'} ] self.assertEqual(expected_results, json.loads(result.output)) def test_quote_verify(self): result = self.run_command([ 'order', 'quote', '12345', '--verify', '--fqdn', 'test01.test.com', '--complex-type', 'SoftLayer_Container_Product_Order_Virtual_Guest']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Billing_Order_Quote', 'verifyOrder', identifier='12345') def test_quote_verify_image(self): result = self.run_command([ 'order', 'quote', '12345', '--verify', '--fqdn', 'test01.test.com', '--image', '1234', '--complex-type', 'SoftLayer_Container_Product_Order_Virtual_Guest']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Virtual_Guest_Block_Device_Template_Group', 'getObject', identifier='1234') self.assert_called_with('SoftLayer_Billing_Order_Quote', 'verifyOrder', identifier='12345') verify_call = self.calls('SoftLayer_Billing_Order_Quote', 'verifyOrder') verify_args = getattr(verify_call[0], 'args')[0] self.assertEqual('0B5DEAF4-643D-46CA-A695-CECBE8832C9D', verify_args['imageTemplateGlobalIdentifier']) def test_quote_verify_image_guid(self): result = self.run_command([ 'order', 'quote', '12345', '--verify', '--fqdn', 'test01.test.com', '--image', '0B5DEAF4-643D-46CA-A695-CECBE8832C9D', '--complex-type', 'SoftLayer_Container_Product_Order_Virtual_Guest']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Billing_Order_Quote', 'verifyOrder', identifier='12345') verify_call = self.calls('SoftLayer_Billing_Order_Quote', 'verifyOrder') verify_args = getattr(verify_call[0], 'args')[0] self.assertEqual('0B5DEAF4-643D-46CA-A695-CECBE8832C9D', verify_args['imageTemplateGlobalIdentifier']) def test_quote_verify_userdata(self): result = self.run_command([ 'order', 'quote', '12345', '--verify', '--fqdn', 'test01.test.com', '--userdata', 'aaaa1234', '--complex-type', 'SoftLayer_Container_Product_Order_Virtual_Guest']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Billing_Order_Quote', 'verifyOrder', identifier='12345') verify_call = self.calls('SoftLayer_Billing_Order_Quote', 'verifyOrder') verify_args = getattr(verify_call[0], 'args')[0] self.assertEqual([{'value': 'aaaa1234'}], verify_args['hardware'][0]['userData']) def test_quote_verify_userdata_file(self): if (sys.platform.startswith("win")): self.skipTest("TempFile tests doesn't work in Windows") with tempfile.NamedTemporaryFile() as userfile: userfile.write(b"some data") userfile.flush() result = self.run_command([ 'order', 'quote', '12345', '--verify', '--fqdn', 'test01.test.com', '--userfile', userfile.name, '--complex-type', 'SoftLayer_Container_Product_Order_Virtual_Guest']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Billing_Order_Quote', 'verifyOrder', identifier='12345') verify_call = self.calls('SoftLayer_Billing_Order_Quote', 'verifyOrder') verify_args = getattr(verify_call[0], 'args')[0] self.assertEqual([{'value': 'some data'}], verify_args['hardware'][0]['userData']) def test_quote_verify_sshkey(self): result = self.run_command([ 'order', 'quote', '12345', '--verify', '--fqdn', 'test01.test.com', '--key', 'Test 1', '--complex-type', 'SoftLayer_Container_Product_Order_Virtual_Guest']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Account', 'getSshKeys') self.assert_called_with('SoftLayer_Billing_Order_Quote', 'verifyOrder', identifier='12345') verify_call = self.calls('SoftLayer_Billing_Order_Quote', 'verifyOrder') verify_args = getattr(verify_call[0], 'args')[0] self.assertEqual(['100'], verify_args['sshKeys']) def test_quote_verify_postinstall_others(self): result = self.run_command([ 'order', 'quote', '12345', '--verify', '--fqdn', 'test01.test.com', '--quantity', '2', '--postinstall', 'https://127.0.0.1/test.sh', '--complex-type', 'SoftLayer_Container_Product_Order_Virtual_Guest']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Billing_Order_Quote', 'verifyOrder', identifier='12345') verify_call = self.calls('SoftLayer_Billing_Order_Quote', 'verifyOrder') verify_args = getattr(verify_call[0], 'args')[0] self.assertEqual(['https://127.0.0.1/test.sh'], verify_args['provisionScripts']) self.assertEqual(2, verify_args['quantity']) def test_quote_place(self): result = self.run_command([ 'order', 'quote', '12345', '--fqdn', 'test01.test.com', '--complex-type', 'SoftLayer_Container_Product_Order_Virtual_Guest']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Billing_Order_Quote', 'placeOrder', identifier='12345') def test_quote_detail(self): result = self.run_command(['order', 'quote-detail', '12345']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Billing_Order_Quote', 'getObject', identifier='12345') def test_quote_save(self): result = self.run_command(['order', 'quote-save', '12345']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Billing_Order_Quote', 'saveQuote', identifier='12345') def test_quote_list(self): result = self.run_command(['order', 'quote-list']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Account', 'getActiveQuotes') def _get_order_items(self): item1 = {'keyName': 'ITEM1', 'description': 'description1', 'itemCategory': {'categoryCode': 'cat1'}, 'prices': [{'id': 1111, 'locationGroupId': None, 'categories': [{'categoryCode': 'cat1'}]}]} item2 = {'keyName': 'ITEM2', 'description': 'description2', 'itemCategory': {'categoryCode': 'cat2'}, 'prices': [{'id': 2222, 'locationGroupId': None, 'categories': [{'categoryCode': 'cat2'}]}]} return [item1, item2] def _get_verified_order_return(self): item1, item2 = self._get_order_items() price1 = {'item': item1, 'hourlyRecurringFee': '0.04', 'recurringFee': '120'} price2 = {'item': item2, 'hourlyRecurringFee': '0.05', 'recurringFee': '150'} return {'orderContainers': [{'prices': [price1, price2]}]} def test_order_lookup(self): result = self.run_command(['order', 'lookup', '12345']) self.assert_no_fail(result) self.assert_called_with('SoftLayer_Billing_Order', 'getObject', identifier='12345') self.assertIn('Ordered By', result.output) self.assertIn('Initial Invoice', result.output) def _get_all_packages(): package_type = {'keyName': 'BARE_METAL_CPU'} all_packages = [ {'id': 1, 'name': 'package1', 'keyName': 'PACKAGE1', 'type': package_type, 'isActive': 1}, {'id': 2, 'name': 'package2', 'keyName': 'PACKAGE2', 'type': package_type, 'isActive': 1}, {'id': 3, 'name': 'package2', 'keyName': 'PACKAGE2', 'type': package_type, 'isActive': 0} ] return all_packages