# Copyright (c) 2017 Cable Television Laboratories, Inc. and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
#
# http://www.apache.org/licenses/LICENSE-2.0

# pylint: disable=missing-docstring

import logging
import unittest

import mock
from snaps.openstack.os_credentials import OSCreds
from xtesting.core import testcase

from functest.opnfv_tests.openstack.snaps import api_check
from functest.opnfv_tests.openstack.snaps import health_check
from functest.opnfv_tests.openstack.snaps import smoke


class APICheckTesting(unittest.TestCase):
    """
    Ensures the ApiCheck class can run in Functest. This test does not
    actually connect with an OpenStack pod.
    """

    def setUp(self):
        self.os_creds = OSCreds(
            username='user', password='pass',
            auth_url='http://foo.com:5000/v3', project_name='bar')
        with mock.patch('os_client_config.get_config') as mock_get_config, \
                mock.patch('shade.OpenStackCloud') as mock_shade, \
                mock.patch('functest.core.tenantnetwork.NewProject') \
                as mock_new_project:
            self.api_check = api_check.ApiCheck(
                os_creds=self.os_creds, ext_net_name='foo')
        mock_get_config.assert_called()
        mock_shade.assert_called()
        mock_new_project.assert_called()

    @mock.patch('xtesting.core.unit.Suite.generate_html')
    @mock.patch('xtesting.core.unit.Suite.generate_xunit')
    @mock.patch('xtesting.core.unit.Suite.generate_stats')
    @mock.patch('os.path.isdir', return_value=True)
    @mock.patch('unittest.TestLoader')
    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_suite_builder.'
                'add_openstack_client_tests')
    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_suite_builder.'
                'add_openstack_api_tests')
    @mock.patch('subunit.run.SubunitTestRunner.run')
    def test_run_success(self, *args):
        args[0].return_value = mock.Mock(
            decorated=mock.Mock(
                testsRun=100, errors=[], failures=[]))
        with mock.patch('six.moves.builtins.open', mock.mock_open()):
            self.assertEqual(testcase.TestCase.EX_OK, self.api_check.run())
        self.assertEqual(
            testcase.TestCase.EX_OK, self.api_check.is_successful())
        args[0].assert_called_with(mock.ANY)
        args[1].assert_called_with(
            ext_net_name='foo', image_metadata=mock.ANY,
            os_creds=self.os_creds, suite=mock.ANY, use_keystone=True)

    @mock.patch('xtesting.core.unit.Suite.generate_html')
    @mock.patch('xtesting.core.unit.Suite.generate_xunit')
    @mock.patch('xtesting.core.unit.Suite.generate_stats')
    @mock.patch('os.path.isdir', return_value=True)
    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_suite_builder.'
                'add_openstack_client_tests')
    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_suite_builder.'
                'add_openstack_api_tests')
    @mock.patch('subunit.run.SubunitTestRunner.run')
    def test_run_1_of_100_ko(self, *args):
        args[0].return_value = mock.Mock(
            decorated=mock.Mock(
                testsRun=100, errors=[], failures=['foo']))
        with mock.patch('six.moves.builtins.open', mock.mock_open()):
            self.assertEqual(testcase.TestCase.EX_OK, self.api_check.run())
        self.assertEqual(
            testcase.TestCase.EX_TESTCASE_FAILED,
            self.api_check.is_successful())
        args[0].assert_called_with(mock.ANY)
        args[1].assert_called_with(
            ext_net_name='foo', image_metadata=mock.ANY,
            os_creds=self.os_creds, suite=mock.ANY, use_keystone=True)

    @mock.patch('xtesting.core.unit.Suite.generate_html')
    @mock.patch('xtesting.core.unit.Suite.generate_xunit')
    @mock.patch('xtesting.core.unit.Suite.generate_stats')
    @mock.patch('os.path.isdir', return_value=True)
    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_suite_builder.'
                'add_openstack_client_tests')
    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_suite_builder.'
                'add_openstack_api_tests')
    @mock.patch('subunit.run.SubunitTestRunner.run')
    def test_run_1_of_100_ko_criteria(self, *args):
        self.api_check.criteria = 90
        args[0].return_value = mock.Mock(
            decorated=mock.Mock(
                testsRun=100, errors=[], failures=['foo']))
        with mock.patch('six.moves.builtins.open', mock.mock_open()):
            self.assertEqual(testcase.TestCase.EX_OK, self.api_check.run())
        self.assertEqual(
            testcase.TestCase.EX_OK, self.api_check.is_successful())
        args[0].assert_called_with(mock.ANY)
        args[1].assert_called_with(
            ext_net_name='foo', image_metadata=mock.ANY,
            os_creds=self.os_creds, suite=mock.ANY, use_keystone=True)


class HealthCheckTesting(unittest.TestCase):
    """
    Ensures the HealthCheck class can run in Functest. This test does not
    actually connect with an OpenStack pod.
    """

    def setUp(self):
        self.os_creds = OSCreds(
            username='user', password='pass',
            auth_url='http://foo.com:5000/v3', project_name='bar')
        with mock.patch('os_client_config.get_config') as mock_get_config, \
                mock.patch('shade.OpenStackCloud') as mock_shade, \
                mock.patch('functest.core.tenantnetwork.NewProject') \
                as mock_new_project:
            self.health_check = health_check.HealthCheck(
                os_creds=self.os_creds, ext_net_name='foo')
        mock_get_config.assert_called()
        mock_shade.assert_called()
        mock_new_project.assert_called()

    @mock.patch('xtesting.core.unit.Suite.generate_html')
    @mock.patch('xtesting.core.unit.Suite.generate_xunit')
    @mock.patch('xtesting.core.unit.Suite.generate_stats')
    @mock.patch('os.path.isdir', return_value=True)
    @mock.patch('snaps.openstack.tests.os_source_file_test.'
                'OSIntegrationTestCase.parameterize')
    @mock.patch('subunit.run.SubunitTestRunner.run')
    def test_run_success(self, *args):
        args[0].return_value = mock.Mock(
            decorated=mock.Mock(
                testsRun=100, errors=[], failures=[]))
        with mock.patch('six.moves.builtins.open', mock.mock_open()):
            self.assertEqual(testcase.TestCase.EX_OK, self.health_check.run())
        self.assertEqual(
            testcase.TestCase.EX_OK, self.health_check.is_successful())
        args[0].assert_called_with(mock.ANY)
        args[1].assert_called_with(
            mock.ANY, ext_net_name='foo', flavor_metadata=mock.ANY,
            image_metadata=mock.ANY, netconf_override=None,
            os_creds=self.os_creds, use_keystone=True)

    @mock.patch('xtesting.core.unit.Suite.generate_html')
    @mock.patch('xtesting.core.unit.Suite.generate_xunit')
    @mock.patch('xtesting.core.unit.Suite.generate_stats')
    @mock.patch('os.path.isdir', return_value=True)
    @mock.patch('snaps.openstack.tests.os_source_file_test.'
                'OSIntegrationTestCase.parameterize')
    @mock.patch('subunit.run.SubunitTestRunner.run')
    def test_run_1_of_100_ko(self, *args):
        args[0].return_value = mock.Mock(
            decorated=mock.Mock(
                testsRun=100, errors=[], failures=['foo']))
        with mock.patch('six.moves.builtins.open', mock.mock_open()):
            self.assertEqual(testcase.TestCase.EX_OK, self.health_check.run())
        self.assertEqual(
            testcase.TestCase.EX_TESTCASE_FAILED,
            self.health_check.is_successful())
        args[0].assert_called_with(mock.ANY)
        args[1].assert_called_with(
            mock.ANY, ext_net_name='foo', flavor_metadata=mock.ANY,
            image_metadata=mock.ANY, netconf_override=None,
            os_creds=self.os_creds, use_keystone=True)

    @mock.patch('xtesting.core.unit.Suite.generate_html')
    @mock.patch('xtesting.core.unit.Suite.generate_xunit')
    @mock.patch('xtesting.core.unit.Suite.generate_stats')
    @mock.patch('os.path.isdir', return_value=True)
    @mock.patch('snaps.openstack.tests.os_source_file_test.'
                'OSIntegrationTestCase.parameterize')
    @mock.patch('subunit.run.SubunitTestRunner.run')
    def test_run_1_of_100_ko_criteria(self, *args):
        self.health_check.criteria = 90
        args[0].return_value = mock.Mock(
            decorated=mock.Mock(
                testsRun=100, errors=[], failures=['foo']))
        with mock.patch('six.moves.builtins.open', mock.mock_open()):
            self.assertEqual(testcase.TestCase.EX_OK, self.health_check.run())
        self.assertEqual(
            testcase.TestCase.EX_OK, self.health_check.is_successful())
        args[0].assert_called_with(mock.ANY)
        args[1].assert_called_with(
            mock.ANY, ext_net_name='foo', flavor_metadata=mock.ANY,
            image_metadata=mock.ANY, netconf_override=None,
            os_creds=self.os_creds, use_keystone=True)


class SmokeTesting(unittest.TestCase):
    """
    Ensures the SnapsSmoke class can run in Functest. This test does not
    actually connect with an OpenStack pod.
    """

    def setUp(self):
        self.os_creds = OSCreds(
            username='user', password='pass',
            auth_url='http://foo.com:5000/v3', project_name='bar')
        with mock.patch('os_client_config.get_config') as mock_get_config, \
                mock.patch('shade.OpenStackCloud') as mock_shade, \
                mock.patch('functest.core.tenantnetwork.NewProject') \
                as mock_new_project:
            self.smoke = smoke.SnapsSmoke(
                os_creds=self.os_creds, ext_net_name='foo')
        mock_get_config.assert_called()
        mock_shade.assert_called()
        mock_new_project.assert_called()

    @mock.patch('xtesting.core.unit.Suite.generate_html')
    @mock.patch('xtesting.core.unit.Suite.generate_xunit')
    @mock.patch('xtesting.core.unit.Suite.generate_stats')
    @mock.patch('os.path.isdir', return_value=True)
    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_suite_builder.'
                'add_openstack_integration_tests')
    @mock.patch('subunit.run.SubunitTestRunner.run')
    def test_run_success(self, *args):
        args[0].return_value = mock.Mock(
            decorated=mock.Mock(
                testsRun=100, errors=[], failures=[]))
        with mock.patch('six.moves.builtins.open', mock.mock_open()):
            self.assertEqual(testcase.TestCase.EX_OK, self.smoke.run())
        self.assertEqual(testcase.TestCase.EX_OK, self.smoke.is_successful())
        args[0].assert_called_with(mock.ANY)
        args[1].assert_called_with(
            ext_net_name='foo', flavor_metadata=mock.ANY,
            image_metadata=mock.ANY, netconf_override=None,
            os_creds=self.os_creds, suite=mock.ANY, use_floating_ips=True,
            use_keystone=True)

    @mock.patch('xtesting.core.unit.Suite.generate_html')
    @mock.patch('xtesting.core.unit.Suite.generate_xunit')
    @mock.patch('xtesting.core.unit.Suite.generate_stats')
    @mock.patch('os.path.isdir', return_value=True)
    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_suite_builder.'
                'add_openstack_integration_tests')
    @mock.patch('subunit.run.SubunitTestRunner.run')
    def test_run_1_of_100_ko(self, *args):
        args[0].return_value = mock.Mock(
            decorated=mock.Mock(
                testsRun=100, errors=[], failures=['foo']))
        with mock.patch('six.moves.builtins.open', mock.mock_open()):
            self.assertEqual(testcase.TestCase.EX_OK, self.smoke.run())
        self.assertEqual(
            testcase.TestCase.EX_TESTCASE_FAILED, self.smoke.is_successful())
        args[0].assert_called_with(mock.ANY)
        args[1].assert_called_with(
            ext_net_name='foo', flavor_metadata=mock.ANY,
            image_metadata=mock.ANY, netconf_override=mock.ANY,
            os_creds=self.os_creds, suite=mock.ANY, use_floating_ips=True,
            use_keystone=True)

    @mock.patch('xtesting.core.unit.Suite.generate_html')
    @mock.patch('xtesting.core.unit.Suite.generate_xunit')
    @mock.patch('xtesting.core.unit.Suite.generate_stats')
    @mock.patch('os.path.isdir', return_value=True)
    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_suite_builder.'
                'add_openstack_integration_tests')
    @mock.patch('subunit.run.SubunitTestRunner.run')
    def test_run_1_of_100_ko_criteria(self, *args):
        self.smoke.criteria = 90
        args[0].return_value = mock.Mock(
            decorated=mock.Mock(
                testsRun=100, errors=[], failures=['foo']))
        with mock.patch('six.moves.builtins.open', mock.mock_open()):
            self.assertEqual(testcase.TestCase.EX_OK, self.smoke.run())
        self.assertEqual(
            testcase.TestCase.EX_OK, self.smoke.is_successful())
        args[0].assert_called_with(mock.ANY)
        args[1].assert_called_with(
            ext_net_name='foo', flavor_metadata=mock.ANY,
            image_metadata=mock.ANY, netconf_override=None,
            os_creds=self.os_creds, suite=mock.ANY, use_floating_ips=True,
            use_keystone=True)


if __name__ == "__main__":
    logging.disable(logging.CRITICAL)
    unittest.main(verbosity=2)