"""
    SoftLayer.tests.CLI.modules.vs.vs_capacity_tests
    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    :license: MIT, see LICENSE for more details.
"""
import json
from SoftLayer.fixtures import SoftLayer_Product_Order
from SoftLayer.fixtures import SoftLayer_Product_Package
from SoftLayer import testing


class VSCapacityTests(testing.TestCase):

    def test_list(self):
        result = self.run_command(['vs', 'capacity', 'list'])
        self.assert_no_fail(result)

    def test_list_no_billing(self):
        account_mock = self.set_mock('SoftLayer_Account', 'getReservedCapacityGroups')
        account_mock.return_value = [
            {
                'id': 3103,
                'name': 'test-capacity',
                'createDate': '2018-09-24T16:33:09-06:00',
                'availableInstanceCount': 1,
                'instanceCount': 3,
                'occupiedInstanceCount': 1,
                'backendRouter': {
                    'hostname': 'bcr02a.dal13',
                },
                'instances': [{'id': 3501}]
            }
        ]
        result = self.run_command(['vs', 'capacity', 'list'])
        self.assert_no_fail(result)
        self.assertEqual(json.loads(result.output)[0]['Flavor'], 'Unknown Billing Item')

    def test_detail(self):
        result = self.run_command(['vs', 'capacity', 'detail', '1234'])
        self.assert_no_fail(result)

    def test_detail_pending(self):
        # Instances don't have a billing item if they haven't been approved yet.
        capacity_mock = self.set_mock('SoftLayer_Virtual_ReservedCapacityGroup', 'getObject')
        get_object = {
            'name': 'test-capacity',
            'instances': [
                {
                    'createDate': '2018-09-24T16:33:09-06:00',
                    'guestId': 62159257,
                    'id': 3501,
                }
            ]
        }
        capacity_mock.return_value = get_object
        result = self.run_command(['vs', 'capacity', 'detail', '1234'])
        self.assert_no_fail(result)

    def test_create_test(self):
        item_mock = self.set_mock('SoftLayer_Product_Package', 'getItems')
        item_mock.return_value = SoftLayer_Product_Package.getItems_RESERVED_CAPACITY
        order_mock = self.set_mock('SoftLayer_Product_Order', 'verifyOrder')
        order_mock.return_value = SoftLayer_Product_Order.rsc_verifyOrder
        result = self.run_command(['vs', 'capacity', 'create', '--name=TEST', '--test',
                                   '--backend_router_id=1234', '--flavor=B1_1X2_1_YEAR_TERM', '--instances=10'])
        self.assert_no_fail(result)

    def test_create(self):
        item_mock = self.set_mock('SoftLayer_Product_Package', 'getItems')
        item_mock.return_value = SoftLayer_Product_Package.getItems_RESERVED_CAPACITY
        order_mock = self.set_mock('SoftLayer_Product_Order', 'placeOrder')
        order_mock.return_value = SoftLayer_Product_Order.rsc_placeOrder
        result = self.run_command(['vs', 'capacity', 'create', '--name=TEST', '--instances=10',
                                   '--backend_router_id=1234', '--flavor=B1_1X2_1_YEAR_TERM'])
        self.assert_no_fail(result)

    def test_create_options(self):
        result = self.run_command(['vs', 'capacity', 'create_options'])
        self.assert_no_fail(result)

    def test_create_guest_test(self):
        result = self.run_command(['vs', 'capacity', 'create-guest', '--capacity-id=3103', '--primary-disk=25',
                                   '-H ABCDEFG', '-D test_list.com', '-o UBUNTU_LATEST_64', '-kTest 1', '--test'])
        self.assert_no_fail(result)

    def test_create_guest(self):
        order_mock = self.set_mock('SoftLayer_Product_Order', 'placeOrder')
        order_mock.return_value = SoftLayer_Product_Order.rsi_placeOrder
        result = self.run_command(['vs', 'capacity', 'create-guest', '--capacity-id=3103', '--primary-disk=25',
                                   '-H ABCDEFG', '-D test_list.com', '-o UBUNTU_LATEST_64', '-kTest 1'])
        self.assert_no_fail(result)