# Copyright 2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"). You
# may not use this file except in compliance with the License. A copy of
# the License is located at
#
# http://aws.amazon.com/apache2.0/
#
# or in the "license" file accompanying this file. This file 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.
import os

import shutil
import sys
from tempfile import mkdtemp, mkstemp

import mock
import unittest

from cement.core import backend, foundation
from cement.utils.misc import rando

from ebcli.core import ebcore, ebrun, fileoperations
from ebcli.lib import aws

from tests.unit.integration import mockservice

ebrun.fix_path()


class TestApp(foundation.CementApp):

    """
    Basic CementApp for generic testing.

    """
    class Meta:
        label = "app-%s" % rando()[:12]
        config_files = []
        argv = []
        base_controller = None
        arguments = []
        exit_on_close = False


class CementTestCase(unittest.TestCase):
    """
    A copy of cement.utils.test.CementTestCase inorder to avoid external, unnecessary dependency
    of `nose`
    """
    app_class = TestApp

    def __init__(self, *args, **kw):
        super(CementTestCase, self).__init__(*args, **kw)

    def setUp(self):
        self.app = self.make_app()
        _, self.tmp_file = mkstemp()
        self.tmp_dir = mkdtemp()

    def make_app(self, *args, **kw):
        self.reset_backend()
        return self.app_class(*args, **kw)

    def reset_backend(self):
        for _handler in backend.__handlers__.copy():
            del backend.__handlers__[_handler]
        for _hook in backend.__hooks__.copy():
            del backend.__hooks__[_hook]

    def ok(self, expr, msg=None):
        if not expr:
            raise AssertionError(msg)

    def eq(self, a, b, msg=None):
        if not a == b:
            raise AssertionError(msg or "%r != %r" % (a, b))


class BaseIntegrationTest(CementTestCase):
    app_class = ebcore.EB

    def setUp(self):
        super(BaseIntegrationTest, self).setUp()
        aws._flush()
        aws.set_region('us-east-1')
        self.reset_backend()
        self.patcher_input = mock.patch('ebcli.core.io.get_input')
        self.patcher_output = mock.patch('ebcli.core.io.echo')
        self.patcher_warning = mock.patch('ebcli.core.io.log_warning')
        self.mock_input = self.patcher_input.start()
        self.mock_output = self.patcher_output.start()
        self.mock_warning = self.patcher_warning.start()

        self.patcher_endpoint = mock.patch('botocore.endpoint.Endpoint')
        self.mock_endpoint = self.patcher_endpoint.start()
        instance = self.mock_endpoint.return_value
        instance.make_request = mockservice.handle_response
        instance.host = 'http://someurl.test/something'

        if not os.path.exists('testDir/'):
            os.makedirs('testDir/')
        os.chdir('testDir')

        if not os.path.exists(fileoperations.beanstalk_directory):
            os.makedirs(fileoperations.beanstalk_directory)

        fileoperations.default_section = 'ebcli_test_default'

        if not os.path.exists('home'):
            os.makedirs('home')

        fileoperations.aws_config_folder = 'home' + os.path.sep
        fileoperations.aws_config_location \
            = fileoperations.aws_config_folder + 'config'

    def run_command(self, *args):
        self.app = ebcore.EB(argv=list(args))
        self.app.setup()
        self.app.run()
        self.app.close()

    def tearDown(self):
        self.patcher_input.stop()
        self.patcher_output.stop()
        self.patcher_endpoint.stop()
        self.patcher_warning.stop()

        os.chdir(os.path.pardir)
        if os.path.exists('testDir'):
            if sys.platform.startswith('win'):
                os.system('rmdir /S /Q testDir')
            else:
                shutil.rmtree('testDir')
        mockservice.reset()


class BaseOperationsTest(CementTestCase):
    module_name = 'base'

    def setUp(self):
        super(BaseOperationsTest, self).setUp()
        self.reset_backend()
        self.patcher_input = mock.patch('ebcli.core.io.get_input')
        self.patcher_eb = mock.patch('ebcli.operations.' + self.module_name + '.elasticbeanstalk')
        self.patcher_output = mock.patch('ebcli.core.io.echo')
        self.patcher_file = mock.patch('ebcli.operations.' + self.module_name + '.fileoperations')

        self.mock_input = self.patcher_input.start()
        self.mock_elasticbeanstalk = self.patcher_eb.start()
        self.mock_output = self.patcher_output.start()

    def tearDown(self):
        self.patcher_eb.stop()
        self.patcher_input.stop()
        self.patcher_output.stop()