"""
    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
