python-dependency-injector/tests/test_utils.py

"""Dependency injector utils unittests."""

import unittest2 as unittest
import dependency_injector as di


class IsProviderTests(unittest.TestCase):
    """`is_provider()` test cases."""

    def test_with_instance(self):
        """Test with instance."""
        self.assertTrue(di.is_provider(di.Provider()))

    def test_with_class(self):
        """Test with class."""
        self.assertFalse(di.is_provider(di.Provider))

    def test_with_string(self):
        """Test with string."""
        self.assertFalse(di.is_provider('some_string'))

    def test_with_object(self):
        """Test with object."""
        self.assertFalse(di.is_provider(object()))

    def test_with_subclass_instance(self):
        """Test with subclass of provider instance."""
        class SomeProvider(di.Provider):
            """Some provider for test."""

        self.assertTrue(di.is_provider(SomeProvider()))

    def test_with_class_with_getattr(self):
        """Test with class that has __getattr__() method implementation."""
        class SomeClass(object):
            """Some test class with __getattr__() method implementation."""

            def __getattr__(self, _):
                """Test implementation that just returns False."""
                return False

        self.assertFalse(di.is_provider(SomeClass()))


class EnsureIsProviderTests(unittest.TestCase):
    """`ensure_is_provider` test cases."""

    def test_with_instance(self):
        """Test with instance."""
        provider = di.Provider()
        self.assertIs(di.ensure_is_provider(provider), provider)

    def test_with_class(self):
        """Test with class."""
        self.assertRaises(di.Error, di.ensure_is_provider, di.Provider)

    def test_with_string(self):
        """Test with string."""
        self.assertRaises(di.Error, di.ensure_is_provider, 'some_string')

    def test_with_object(self):
        """Test with object."""
        self.assertRaises(di.Error, di.ensure_is_provider, object())


class IsInjectionTests(unittest.TestCase):
    """`is_injection()` test cases."""

    def test_with_instance(self):
        """Test with instance."""
        self.assertTrue(di.is_injection(di.Injection('value')))

    def test_with_subclass_instances(self):
        """Test with subclass instances."""
        self.assertTrue(di.is_injection(di.Arg('value')))
        self.assertTrue(di.is_injection(di.KwArg('name', 'value')))
        self.assertTrue(di.is_injection(di.Attribute('name', 'value')))
        self.assertTrue(di.is_injection(di.Method('name', 'value')))

    def test_with_class(self):
        """Test with class."""
        self.assertFalse(di.is_injection(di.Injection))

    def test_with_string(self):
        """Test with string."""
        self.assertFalse(di.is_injection('some_string'))

    def test_with_object(self):
        """Test with object."""
        self.assertFalse(di.is_injection(object()))


class EnsureIsInjectionTests(unittest.TestCase):
    """`ensure_is_injection` test cases."""

    def test_with_instance(self):
        """Test with instance."""
        injection = di.Injection('value')
        self.assertIs(di.ensure_is_injection(injection), injection)

    def test_with_class(self):
        """Test with class."""
        self.assertRaises(di.Error, di.ensure_is_injection, di.Injection)

    def test_with_string(self):
        """Test with string."""
        self.assertRaises(di.Error, di.ensure_is_injection, 'some_string')

    def test_with_object(self):
        """Test with object."""
        self.assertRaises(di.Error, di.ensure_is_injection, object())


class IsArgInjectionTests(unittest.TestCase):
    """`is_arg_injection()` test cases."""

    def test_with_instance(self):
        """Test with instance."""
        self.assertTrue(di.is_arg_injection(di.Arg('value')))

    def test_with_class(self):
        """Test with class."""
        self.assertFalse(di.is_arg_injection(di.Arg))

    def test_with_parent_class(self):
        """Test with parent class."""
        self.assertFalse(di.is_arg_injection(di.Injection))

    def test_with_string(self):
        """Test with string."""
        self.assertFalse(di.is_arg_injection('some_string'))

    def test_with_object(self):
        """Test with object."""
        self.assertFalse(di.is_arg_injection(object()))


class IsKwArgInjectionTests(unittest.TestCase):
    """`is_kwarg_injection()` test cases."""

    def test_with_instance(self):
        """Test with instance."""
        self.assertTrue(di.is_kwarg_injection(di.KwArg('name', 'value')))

    def test_with_class(self):
        """Test with class."""
        self.assertFalse(di.is_kwarg_injection(di.KwArg))

    def test_with_parent_class(self):
        """Test with parent class."""
        self.assertFalse(di.is_kwarg_injection(di.Injection))

    def test_with_string(self):
        """Test with string."""
        self.assertFalse(di.is_kwarg_injection('some_string'))

    def test_with_object(self):
        """Test with object."""
        self.assertFalse(di.is_kwarg_injection(object()))


class IsAttributeInjectionTests(unittest.TestCase):
    """`is_attribute_injection()` test cases."""

    def test_with_instance(self):
        """Test with instance."""
        self.assertTrue(di.is_attribute_injection(di.Attribute('name',
                                                               'value')))

    def test_with_class(self):
        """Test with class."""
        self.assertFalse(di.is_attribute_injection(di.Attribute))

    def test_with_parent_class(self):
        """Test with parent class."""
        self.assertFalse(di.is_attribute_injection(di.Injection))

    def test_with_string(self):
        """Test with string."""
        self.assertFalse(di.is_attribute_injection('some_string'))

    def test_with_object(self):
        """Test with object."""
        self.assertFalse(di.is_attribute_injection(object()))


class IsMethodInjectionTests(unittest.TestCase):
    """`is_method_injection()` test cases."""

    def test_with_instance(self):
        """Test with instance."""
        self.assertTrue(di.is_method_injection(di.Method('name', 'value')))

    def test_with_class(self):
        """Test with class."""
        self.assertFalse(di.is_method_injection(di.Method))

    def test_with_parent_class(self):
        """Test with parent class."""
        self.assertFalse(di.is_method_injection(di.Injection))

    def test_with_string(self):
        """Test with string."""
        self.assertFalse(di.is_method_injection('some_string'))

    def test_with_object(self):
        """Test with object."""
        self.assertFalse(di.is_method_injection(object()))


class IsCatalogTests(unittest.TestCase):
    """`is_catalog()` test cases."""

    def test_with_declarative_catalog(self):
        """Test with class."""
        self.assertTrue(di.is_catalog(di.DeclarativeCatalog))

    def test_with_dynamic_catalog(self):
        """Test with class."""
        self.assertTrue(di.is_catalog(di.DynamicCatalog('TestCatalog')))

    def test_with_child_class(self):
        """Test with parent class."""
        class Catalog(di.AbstractCatalog):
            """Example catalog child class."""

        self.assertTrue(di.is_catalog(Catalog))

    def test_with_string(self):
        """Test with string."""
        self.assertFalse(di.is_catalog('some_string'))

    def test_with_object(self):
        """Test with object."""
        self.assertFalse(di.is_catalog(object()))


class IsDynamicCatalogTests(unittest.TestCase):
    """`is_dynamic_catalog()` test cases."""

    def test_with_declarative_catalog(self):
        """Test with declarative catalog."""
        self.assertFalse(di.is_dynamic_catalog(di.DeclarativeCatalog))

    def test_with_dynamic_catalog(self):
        """Test with dynamic catalog."""
        self.assertTrue(di.is_dynamic_catalog(di.DynamicCatalog(
            'TestCatalog')))


class IsDeclarativeCatalogTests(unittest.TestCase):
    """`is_declarative_catalog()` test cases."""

    def test_with_declarative_catalog(self):
        """Test with declarative catalog."""
        self.assertTrue(di.is_declarative_catalog(di.DeclarativeCatalog))

    def test_with_dynamic_catalog(self):
        """Test with dynamic catalog."""
        self.assertFalse(di.is_declarative_catalog(di.DynamicCatalog(
            'TestCatalog')))


class IsCatalogBundleTests(unittest.TestCase):
    """`is_catalog_bundle()` test cases."""

    def test_with_instance(self):
        """Test with instance."""
        self.assertTrue(di.is_catalog_bundle(di.CatalogBundle()))

    def test_with_cls(self):
        """Test with class."""
        self.assertFalse(di.is_catalog_bundle(di.CatalogBundle))

    def test_with_string(self):
        """Test with string."""
        self.assertFalse(di.is_catalog_bundle('some_string'))

    def test_with_object(self):
        """Test with object."""
        self.assertFalse(di.is_catalog_bundle(object()))


class EnsureIsCatalogBundleTests(unittest.TestCase):
    """`ensure_is_catalog_bundle` test cases."""

    def test_with_instance(self):
        """Test with instance."""
        bundle = di.CatalogBundle()
        self.assertIs(di.ensure_is_catalog_bundle(bundle), bundle)

    def test_with_class(self):
        """Test with class."""
        self.assertRaises(di.Error, di.ensure_is_catalog_bundle,
                          di.CatalogBundle)

    def test_with_string(self):
        """Test with string."""
        self.assertRaises(di.Error, di.ensure_is_catalog_bundle, 'some_string')

    def test_with_object(self):
        """Test with object."""
        self.assertRaises(di.Error, di.ensure_is_catalog_bundle, object())