abstract_test.py

"""Describe TestBlock class."""
# pylint: disable=attribute-defined-outside-init,unused-argument
# pylint: disable=dangerous-default-value,access-member-before-definition
# pylint: disable=bare-except,protected-access,too-many-instance-attributes
# pylint: disable=too-many-arguments,too-many-locals,broad-except,no-self-use
# pylint: disable=too-many-public-methods
import os
import sys
import unittest
from bdb import BdbQuit
from functools import wraps
from itertools import count

from ipdbugger import debug
from attrdict import AttrDict

from rotest.common.utils import get_class_fields
from rotest.core.models.case_data import TestOutcome
from rotest.management.base_resource import BaseResource
from rotest.management.client.manager import ResourceRequest
from rotest.management.client.manager import ClientResourceManager

request = ResourceRequest


class AbstractTest(unittest.TestCase):
    """Base class for all runnable Rotest tests.

    Attributes:
        resources (tuple): list of the required resources. each item is a
            tuple of (resource_name, resource type, parameters dictionary),
            you can use :func:`rotest.core..request` to create the tuple.
        identifier (number): unique id of the test.
        data (rotest.core.models._data.Data): contain information
            about a test  run.
        logger (logging.Logger): test logger.
        save_state (bool): a flag to determine if storing the states of
            resources is required.
        force_initialize (bool): a flag to determine if the resources will be
            initialized even if their validation succeeds.
        config (AttrDict): dictionary of configurations.
        enable_debug (bool): whether to enable entering ipdb debugging mode
            upon any exception in a test statement.
        skip_init (bool): True to skip resources initialize and validation.
        resource_manager (ClientResourceManager): client resource manager.
        TAGS (list): list of tags by which the test may be filtered.
        IS_COMPLEX (bool): if this test is complex (may contain sub-tests).
        TIMEOUT (number): timeout for flow run, None means no timeout.
    """
    SETUP_METHOD_NAME = 'setUp'
    TEARDOWN_METHOD_NAME = 'tearDown'

    TIMEOUT = 1800  # 30 minutes

    resources = ()

    TAGS = []
    IS_COMPLEX = False

    STATE_DIR_NAME = "state"

    def __init__(self, indexer=count(), methodName='runTest', save_state=True,
                 force_initialize=False, config=None, parent=None,
                 enable_debug=True, resource_manager=None, skip_init=False):

        if enable_debug:
            for method_name in (methodName, self.SETUP_METHOD_NAME,
                                self.TEARDOWN_METHOD_NAME):

                debug(getattr(self, method_name),
                      ignore_exceptions=[KeyboardInterrupt,
                                         unittest.SkipTest,
                                         BdbQuit])

        super(AbstractTest, self).__init__(methodName)

        self.result = None
        self.config = config
        self.parent = parent
        self.skip_init = skip_init
        self.save_state = save_state
        self.identifier = indexer.next()
        self.enable_debug = enable_debug
        self.force_initialize = force_initialize
        self.parents_count = self._get_parents_count()

        self.all_resources = AttrDict()
        self.locked_resources = AttrDict()

        self._is_client_local = False
        self.resource_manager = resource_manager

        if parent is not None:
            parent.addTest(self)

    def override_resource_loggers(self):
        """Replace the resources' logger with the test's logger."""
        for resource in self.all_resources.itervalues():
            resource.override_logger(self.logger)

    def release_resource_loggers(self):
        """Revert logger replacement."""
        for resource in self.all_resources.itervalues():
            resource.release_logger(self.logger)

    @classmethod
    def get_resource_requests(cls):
        """Return a list of all the resource requests this test makes.

        Resource requests can be done both by overriding the class's
        'resources' field and by declaring class fields that point to a
        BaseResource instance.

        Returns:
            list. resource requests of the test class.
        """
        all_requests = list(cls.resources)
        for (field_name, field) in get_class_fields(cls, BaseResource):
            new_request = request(field_name,
                                  field.__class__,
                                  **field.kwargs)

            if new_request not in all_requests:
                all_requests.append(new_request)

        return all_requests

    def create_resource_manager(self):
        """Create a new resource manager client instance.

        Returns:
            ClientResourceManager. new resource manager client.
        """
        return ClientResourceManager()

    def expect(self, expression, msg=None):
        """Check an expression and fail the test at the end if it's False.

        This does not raise an AssertionError like assertTrue, but instead
        updates the result of the test and appends the message to the saved
        traceback without stopping its flow.

        Args:
            expression (bool): value to validate.
            msg (str): failure message if the expression is False.

        Returns:
            bool. True if the validation passed, False otherwise.
        """
        if not expression:
            failure = AssertionError(msg)
            self.result.addFailure(self, (failure.__class__, failure, None))
            return False

        return True

    def add_resources(self, resources):
        """Register the resources to the case and set them as its attributes.

        Args:
            resources (dict): dictionary of attributes name to resources
                instance.
        """
        self.all_resources.update(resources)
        for name, resource in resources.iteritems():
            setattr(self, name, resource)

    def request_resources(self, resources_to_request, use_previous=False):
        """Lock the requested resources and prepare them for the test.

        Lock the required resources using the resource manager, then assign
        each resource to its requested name, and update the result of the
        chosen resources. This method can also be used to add resources to all
        the sibling blocks under the test-flow.

        Args:
            resources_to_request (list): list of resource requests to lock.
            use_previous (bool): whether to use previously locked resources and
                release the unused ones.
        """
        if len(resources_to_request) == 0:
            # No resources to requested
            return

        requested_resources = self.resource_manager.request_resources(
                                        config=self.config,
                                        skip_init=self.skip_init,
                                        use_previous=use_previous,
                                        base_work_dir=self.work_dir,
                                        requests=resources_to_request,
                                        enable_debug=self.enable_debug,
                                        force_initialize=self.force_initialize)

        self.add_resources(requested_resources)
        self.locked_resources.update(requested_resources)
        for resource in requested_resources.itervalues():
            resource.override_logger(self.logger)

        if self.result is not None:
            self.result.updateResources(self)

    def release_resources(self, resources=None, dirty=False,
                          force_release=True):
        """Release given resources using the client.

        Args:
            resources (list): resource names to release, leave None to release
                all locked resources.
            dirty (bool): True if the resource's integrity has been
                compromised, and it should be re-validated.
            force_release (bool): whether to always release to resources
                or enable saving them for next tests.
        """
        if resources is None:
            resources = self.locked_resources.keys()

        if len(resources) == 0:
            # No resources to release locked
            return

        resources_dict = {name: resource
                          for name, resource in self.locked_resources.items()
                          if name in resources}

        self.resource_manager.release_resources(resources_dict,
                                                dirty=dirty,
                                                force_release=force_release)

        # Remove the resources from the test's resource to avoid double release
        for resource in resources_dict.itervalues():
            self.locked_resources.pop(resource, None)

    def _get_parents_count(self):
        """Get the number of ancestors.

        Returns:
            number. number of ancestors.
        """
        if self.parent is None:
            return 0

        return self.parent.parents_count + 1

    def start(self):
        """Update the data that the test started."""
        self.data.start()

    def end(self, test_outcome, details=None):
        """Update the data that the test ended.

        Args:
            test_outcome (number): test outcome code (as defined in
                rotest.core.models.case_data.TestOutcome).
            details (str): details of the result (traceback/skip reason).
        """
        self.data.update_result(test_outcome, details)

    def _decorate_teardown(self, teardown_method, result):
        """Decorate the tearDown method to handle resource release.

        Args:
            teardown_method (function): the original tearDown method.
            result (rotest.core.result.result.Result): test result information.

        Returns:
            function. the wrapped tearDown method.
        """
        @wraps(teardown_method)
        def teardown_method_wrapper(*args, **kwargs):
            """tearDown method wrapper.

            * Executes the original tearDown method.
            * Releases the test resources.
            * Closes the client if needed
            """
            self.result.startTeardown(self)
            try:
                teardown_method(*args, **kwargs)

            except Exception:
                result.addError(self, sys.exc_info())

            finally:
                self.store_state()
                self.release_resources(
                       dirty=self.data.exception_type == TestOutcome.ERROR,
                       force_release=False)

                if (self._is_client_local and
                        self.resource_manager.is_connected()):
                    self.resource_manager.disconnect()

        return teardown_method_wrapper

    def store_state(self):
        """Store the state of the resources in the work dir."""
        status = self.data.exception_type
        if (not self.save_state or status is None or
                status in TestOutcome.POSITIVE_RESULTS):

            self.logger.debug("Skipping saving error state")
            return

        store_dir = os.path.join(self.work_dir, self.STATE_DIR_NAME)

        # In case a state dir already exists, create a new one.
        state_dir_index = 1
        while os.path.exists(store_dir):
            state_dir_index += 1
            store_dir = os.path.join(self.work_dir,
                                     self.STATE_DIR_NAME + str(
                                         state_dir_index))

        self.logger.debug("Creating state dir %r", store_dir)
        os.makedirs(store_dir)

        for resource in self.all_resources.itervalues():
            resource.store_state(store_dir)

    def _wrap_assert(self, assert_method, *args, **kwargs):
        try:
            assert_method(*args, **kwargs)

        except AssertionError as err:
            self.expect(False, str(err))

    def expectFalse(self, expr, msg=None):
        self._wrap_assert(self.assertFalse, expr, msg)

    def expectTrue(self, expr, msg=None):
        self._wrap_assert(self.assertTrue, expr, msg)

    def expectEqual(self, first, second, msg=None):
        self._wrap_assert(self.assertEqual, first, second, msg)

    def expectNotEqual(self, first, second, msg=None):
        self._wrap_assert(self.assertNotEqual, first, second, msg)

    def expectAlmostEqual(self, first, second, places=None,
                          msg=None, delta=None):

        self._wrap_assert(self.assertAlmostEqual, first, second, places,
                          msg, delta)

    def expectNotAlmostEqual(self, first, second, places=None,
                             msg=None, delta=None):

        self._wrap_assert(self.assertNotAlmostEqual, first, second, places,
                          msg, delta)

    expectEquals = expectEqual
    expectNotEquals = expectNotEqual
    expectAlmostEquals = expectAlmostEqual
    expectNotAlmostEquals = expectNotAlmostEqual

    def expectSequenceEqual(self, seq1, seq2, msg=None, seq_type=None):
        self._wrap_assert(self.assertSequenceEqual, seq1, seq2, msg, seq_type)

    def expectListEqual(self, list1, list2, msg=None):
        self._wrap_assert(self.assertListEqual, list1, list2, msg)

    def expectTupleEqual(self, tuple1, tuple2, msg=None):
        self._wrap_assert(self.assertTupleEqual, tuple1, tuple2, msg)

    def expectSetEqual(self, set1, set2, msg=None):
        self._wrap_assert(self.assertSetEqual, set1, set2, msg)

    def expectIn(self, member, container, msg=None):
        self._wrap_assert(self.assertIn, member, container, msg)

    def expectNotIn(self, member, container, msg=None):
        self._wrap_assert(self.assertNotIn, member, container, msg)

    def expectIs(self, expr1, expr2, msg=None):
        self._wrap_assert(self.assertIs, expr1, expr2, msg)

    def expectIsNot(self, expr1, expr2, msg=None):
        self._wrap_assert(self.assertIsNot, expr1, expr2, msg)

    def expectDictEqual(self, set1, set2, msg=None):
        self._wrap_assert(self.assertDictEqual, set1, set2, msg)

    def expectDictContainsSubset(self, expected, actual, msg=None):
        self._wrap_assert(self.assertDictContainsSubset, expected, actual, msg)

    def expectItemsEqual(self, expected_seq, actual_seq, msg=None):
        self._wrap_assert(self.assertItemsEqual, expected_seq, actual_seq, msg)

    def expectMultiLineEqual(self, first, second, msg=None):
        self._wrap_assert(self.assertMultiLineEqual, first, second, msg)

    def expectLess(self, a, b, msg=None):
        self._wrap_assert(self.assertLess, a, b, msg)

    def expectLessEqual(self, a, b, msg=None):
        self._wrap_assert(self.assertLessEqual, a, b, msg)

    def expectGreater(self, a, b, msg=None):
        self._wrap_assert(self.assertGreater, a, b, msg)

    def expectGreaterEqual(self, a, b, msg=None):
        self._wrap_assert(self.assertGreaterEqual, a, b, msg)

    def expectIsNone(self, obj, msg=None):
        self._wrap_assert(self.assertIsNone, obj, msg)

    def expectIsNotNone(self, obj, msg=None):
        self._wrap_assert(self.assertIsNotNone, obj, msg)

    def expectIsInstance(self, obj, msg=None):
        self._wrap_assert(self.assertIsInstance, obj, msg)

    def expectNotIsInstance(self, obj, msg=None):
        self._wrap_assert(self.assertNotIsInstance, obj, msg)

    def expectRegexpMatches(self, text, expected_regexp, msg=None):
        self._wrap_assert(self.assertRegexpMatches, text,
                          expected_regexp, msg)

    def expectNotRegexpMatches(self, text, unexpected_regexp, msg=None):
        self._wrap_assert(self.assertNotRegexpMatches, text,
                          unexpected_regexp, msg)

    class _ExpectRaisesContext(object):
        def __init__(self, assert_context, wrap_assert):
            self.assert_context = assert_context
            self.wrap_assert = wrap_assert

        def __enter__(self):
            return self

        def __exit__(self, exc_type, exc_value, tb):
            self.wrap_assert(self.assert_context.__exit__,
                             exc_type, exc_value, tb)

    def expectRaises(self, expected_exception, callable_obj=None,
                     *args, **kwargs):

        if callable_obj is None:
            return AbstractTest._ExpectRaisesContext(self.assertRaises(
                                                        expected_exception,
                                                        callable_obj,
                                                        *args, **kwargs),
                                                     self._wrap_assert)

        self._wrap_assert(self.assertRaises, expected_exception, callable_obj,
                          *args, **kwargs)

    def expectRaisesRegexp(self, expected_exception, expected_regexp,
                           callable_obj=None, *args, **kwargs):

        if callable_obj is None:
            return AbstractTest._ExpectRaisesContext(self.assertRaisesRegexp(
                                                        expected_exception,
                                                        expected_regexp,
                                                        callable_obj,
                                                        *args, **kwargs),
                                                     self._wrap_assert)

        self._wrap_assert(self.assertRaisesRegexp, expected_exception,
                          expected_regexp, callable_obj, *args, **kwargs)