# Copyright 2016 EricssonAB.
#
#    Licensed under the Apache License, Version 2.0 (the "License");
#    you may not use this file except in compliance with the License.
#    You may obtain a copy of the License at
#
#        http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS,
#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#    See the License for the specific language governing permissions and
#    limitations under the License.
#
# Copyright (c) 2017 Wind River Systems, Inc.
#
# The right to copy, distribute, modify, or otherwise make use
# of this software may be licensed only pursuant to the terms
# of an applicable Wind River license agreement.
#

import mock

from dcmanagerclient.tests import base_shell_test as base


class TestShell(base.BaseShellTests):

    @mock.patch('dcmanagerclient.api.client.determine_client_version')
    def test_dcmanager_version(self, mock):
        self.shell(
            '--os-dcmanager-version=v1 quota-defaults'
        )
        self.assertTrue(mock.called)
        dcmanager_version = mock.call_args
        self.assertEqual('v1', dcmanager_version[0][0])

    @mock.patch('dcmanagerclient.api.client.determine_client_version')
    def test_default_dcmanager_version(self, mock):
        default_version = 'v1.0'
        self.shell('quota defaults')
        self.assertTrue(mock.called)
        dcmanager_version = mock.call_args
        self.assertEqual(default_version, dcmanager_version[0][0])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_env_variables(self, mock):
        self.shell(
            '--os-auth-url=https://127.0.0.1:35357/v3 '
            '--os-username=admin '
            '--os-password=1234 '
            '--os-tenant-name=admin '
            'quota defaults'
        )
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('https://127.0.0.1:35357/v3', params[1]['auth_url'])
        self.assertEqual('admin', params[1]['username'])
        self.assertEqual('admin', params[1]['project_name'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_env_without_auth_url(self, mock):
        self.shell(
            '--os-username=admin '
            '--os-password=1234 '
            '--os-tenant-name=admin '
            'quota defaults'
        )
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('', params[1]['auth_url'])
        self.assertEqual('admin', params[1]['username'])
        self.assertEqual('admin', params[1]['project_name'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_kb_service_type(self, mock):
        self.shell('--os-service-type=dcmanager')
        self.assertTrue(mock.called)
        parameters = mock.call_args
        self.assertEqual('dcmanager', parameters[1]['service_type'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_kb_default_service_type(self, mock):
        self.shell('quota defaults')
        self.assertTrue(mock.called)
        params = mock.call_args
        # Default service type is dcmanager
        self.assertEqual('dcmanager', params[1]['service_type'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_kb_endpoint_type(self, mock):
        self.shell('--os-dcmanager-endpoint-type=adminURL quota-defaults')
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('adminURL', params[1]['endpoint_type'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_kb_default_endpoint_type(self, mock):
        self.shell('quota defaults')
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('internalURL', params[1]['endpoint_type'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_os_auth_token(self, mock):
        self.shell(
            '--os-auth-token=abcd1234 '
            'quota defaults'
        )
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('abcd1234', params[1]['auth_token'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_command_without_dcmanager_url(self, mock):
        self.shell(
            'quota defaults'
        )
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('', params[1]['dcmanager_url'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_command_with_dcmanager_url(self, mock):
        self.shell(
            '--os-dcmanager-url=http://localhost:8118/v1 quota-defaults'
        )
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('http://localhost:8118/v1',
                         params[1]['dcmanager_url'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_command_without_project_name(self, mock):
        self.shell(
            'quota defaults'
        )
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('', params[1]['project_name'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_dcmanager_profile(self, mock):
        self.shell('--profile=SECRET_HMAC_KEY quota defaults')
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('SECRET_HMAC_KEY', params[1]['profile'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_dcmanager_without_profile(self, mock):
        self.shell('quota defaults')
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual(None, params[1]['profile'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_dcmanager_project_name(self, mock):
        self.shell('--os-project-name default quota defaults')
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('default', params[1]['project_name'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_dcmanager_tenant_name(self, mock):
        self.shell('--os-tenant-name default quota defaults')
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('default', params[1]['project_name'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_dcmanager_project_domain_name(self, mock):
        self.shell('--os-project-domain-name default quota defaults')
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('default', params[1]['project_domain_name'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_dcmanager_project_domain_id(self, mock):
        self.shell('--os-project-domain-id default quota defaults')
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('default', params[1]['project_domain_id'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_dcmanager_user_domain_name(self, mock):
        self.shell('--os-user-domain-name default quota defaults')
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('default', params[1]['user_domain_name'])

    @mock.patch('dcmanagerclient.api.client.client')
    def test_dcmanager_user_domain_id(self, mock):
        self.shell('--os-user-domain-id default quota defaults')
        self.assertTrue(mock.called)
        params = mock.call_args
        self.assertEqual('default', params[1]['user_domain_id'])