# Copyright 2012 Managed I.T.
#
# Author: Kiall Mac Innes <kiall@managedit.ie>
#
# 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.
import sys
import unittest
import mox
from moniker.openstack.common import cfg
from moniker.openstack.common import log as logging
from moniker.openstack.common.context import RequestContext, get_admin_context
from moniker import storage

LOG = logging.getLogger(__name__)


class TestCase(unittest.TestCase):
    def setUp(self):
        super(TestCase, self).setUp()

        self.mox = mox.Mox()
        self.config(database_connection='sqlite://',
                    rpc_backend='moniker.openstack.common.rpc.impl_fake',
                    notification_driver=[])
        storage.setup_schema()

    def tearDown(self):
        storage.teardown_schema()
        cfg.CONF.reset()
        self.mox.UnsetStubs()

        super(TestCase, self).tearDown()

    def config(self, **kwargs):
        group = kwargs.pop('group', None)
        for k, v in kwargs.iteritems():
            cfg.CONF.set_override(k, v, group)

    def get_context(self, **kwargs):
        return RequestContext(**kwargs)

    def get_admin_context(self):
        return get_admin_context()


if sys.version_info < (2, 7):
    # Add in some of the nicer methods not present in 2.6
    from contextlib import contextmanager

    def assertIsNone(self, expr, msg=None):
        return self.assertEqual(expr, None, msg)

    TestCase.assertIsNone = assertIsNone

    def assertIsNotNone(self, expr, msg=None):
        return self.assertNotEqual(expr, None, msg)

    TestCase.assertIsNotNone = assertIsNotNone

    def assertRaises(self, excClass, callableObj=None, *args, **kwargs):
        @contextmanager
        def context():
            raised = None
            try:
                yield
            except Exception, e:
                raised = e
            finally:
                if not isinstance(raised, excClass):
                    raise self.failureException(
                        "%s not raised" % str(excClass))

        if callableObj is None:
            return context()
        with context:
            callableObj(*args, **kwargs)

    TestCase.assertRaises = assertRaises