#
# Copyright (c) 2013-2019 Wind River Systems, Inc.
#
# SPDX-License-Identifier: Apache-2.0
#

import fixtures
import httplib2
import mock
import re
from six.moves import cStringIO as StringIO
import sys
from testtools import matchers

import keystoneauth1

from cgtsclient import exc
from cgtsclient import shell as cgts_shell
from cgtsclient.tests import utils
from cgtsclient.v1.ihost import ihost

FAKE_ENV = {'OS_USERNAME': 'username',
            'OS_PASSWORD': 'password',
            'OS_PROJECT_NAME': 'project',
            'OS_REGION_NAME': 'region',
            'OS_TENANT_NAME': 'tenant_name',
            'OS_AUTH_URL': 'http://no.where'}


class ShellTest(utils.BaseTestCase):
    re_options = re.DOTALL | re.MULTILINE

    mock_endpoint_patcher = mock.patch.object(keystoneauth1.session.Session,
                                              'get_endpoint')

    # Patch os.environ to avoid required auth info.
    def make_env(self, exclude=None):
        env = dict((k, v) for k, v in FAKE_ENV.items() if k != exclude)
        self.useFixture(fixtures.MonkeyPatch('os.environ', env))

    def setUp(self):
        super(ShellTest, self).setUp()
        self.mock_endpoint = self.mock_endpoint_patcher.start()

    def tearDown(self):
        super(ShellTest, self).tearDown()
        self.mock_endpoint_patcher.stop()

    def shell(self, argstr):
        orig = sys.stdout
        try:
            sys.stdout = StringIO()
            _shell = cgts_shell.CgtsShell()
            _shell.main(argstr.split())
        except SystemExit:
            exc_type, exc_value, exc_traceback = sys.exc_info()
            self.assertEqual(exc_value.code, 0)
        finally:
            out = sys.stdout.getvalue()
            sys.stdout.close()
            sys.stdout = orig

        return out

    def test_help_unknown_command(self):
        self.assertRaises(exc.CommandError, self.shell, 'help foofoo')

    def test_debug(self):
        httplib2.debuglevel = 0
        self.shell('--debug help')
        self.assertEqual(httplib2.debuglevel, 1)

    def test_help(self):
        required = [
            '.*?^usage: system',
            '.*?^See "system help COMMAND" '
            'for help on a specific command',
        ]
        for argstr in ['--help', 'help']:
            help_text = self.shell(argstr)
            for r in required:
                self.assertThat(help_text,
                                matchers.MatchesRegex(r,
                                                      self.re_options))

    def test_help_on_subcommand(self):
        required = [
            r'.*?^usage: system host-show \[--column COLUMN\] \[--format {table,yaml,value}\]'
            r'.*?<hostname or id>'
            r'',
            r".*?^Show host attributes.",
            r'',
            r".*?^Positional arguments:",
            r".*?<hostname or id>      Name or ID of host",
        ]
        argstrings = [
            'help host-show',
        ]
        for argstr in argstrings:
            help_text = self.shell(argstr)
            for r in required:
                self.assertThat(help_text,
                                matchers.MatchesRegex(r, self.re_options))

    def test_auth_param(self):
        self.make_env(exclude='OS_USERNAME')
        self.test_help()

    @mock.patch('cgtsclient.v1.ihost.ihostManager.list')
    def test_host_list(self, mock_list):
        # This unit test mocks returning a single controller-0 host through host-list
        fake_controller = {'id': '0',
                           'hostname': 'controller-0',
                           'personality': 'controller',
                           'administrative': 'unlocked',
                           'operational': 'enabled',
                           'availability': 'available'}
        mock_list.return_value = [ihost(None, fake_controller, True)]
        self.make_env()
        host_results = self.shell("host-list")
        self.assertIn('controller-0', host_results)
        self.assertNotIn('controller-1', host_results)

    def test_help_on_network_addrpool_subcommand(self):
        self.shell("help network-addrpool-list")
        self.shell("help network-addrpool-show")
        self.shell("help network-addrpool-assign")
        self.shell("help network-addrpool-remove")