Refactor injections

This commit is contained in:
Roman Mogilatov 2016-05-29 16:39:39 +03:00
parent a1da758e6b
commit 99b6e27ab4
15 changed files with 287 additions and 802 deletions

View File

@ -134,10 +134,10 @@ IoC containers from previous example could look like these:
"""Catalog of platform service providers.""" """Catalog of platform service providers."""
database = providers.Singleton(sqlite3.connect) \ database = providers.Singleton(sqlite3.connect) \
.args(':memory:') .add_args(':memory:')
s3 = providers.Singleton(boto.s3.connection.S3Connection) \ s3 = providers.Singleton(boto.s3.connection.S3Connection) \
.kwargs(aws_access_key_id='KEY', .add_kwargs(aws_access_key_id='KEY',
aws_secret_access_key='SECRET') aws_secret_access_key='SECRET')
@ -145,14 +145,14 @@ IoC containers from previous example could look like these:
"""Catalog of business service providers.""" """Catalog of business service providers."""
users = providers.Factory(example.services.Users) \ users = providers.Factory(example.services.Users) \
.kwargs(db=Platform.database) .add_kwargs(db=Platform.database)
photos = providers.Factory(example.services.Photos) \ photos = providers.Factory(example.services.Photos) \
.kwargs(db=Platform.database, .add_kwargs(db=Platform.database,
s3=Platform.s3) s3=Platform.s3)
auth = providers.Factory(example.services.Auth) \ auth = providers.Factory(example.services.Auth) \
.kwargs(db=Platform.database, .add_kwargs(db=Platform.database,
token_ttl=3600) token_ttl=3600)
or like this these: or like this these:
@ -163,10 +163,10 @@ or like this these:
"""Catalog of platform service providers.""" """Catalog of platform service providers."""
database = providers.Singleton(sqlite3.connect) database = providers.Singleton(sqlite3.connect)
database.args(':memory:') database.add_args(':memory:')
s3 = providers.Singleton(boto.s3.connection.S3Connection) s3 = providers.Singleton(boto.s3.connection.S3Connection)
s3.kwargs(aws_access_key_id='KEY', s3.add_kwargs(aws_access_key_id='KEY',
aws_secret_access_key='SECRET') aws_secret_access_key='SECRET')
@ -174,14 +174,14 @@ or like this these:
"""Catalog of business service providers.""" """Catalog of business service providers."""
users = providers.Factory(example.services.Users) users = providers.Factory(example.services.Users)
users.kwargs(db=Platform.database) users.add_kwargs(db=Platform.database)
photos = providers.Factory(example.services.Photos) photos = providers.Factory(example.services.Photos)
photos.kwargs(db=Platform.database, photos.add_kwargs(db=Platform.database,
s3=Platform.s3) s3=Platform.s3)
auth = providers.Factory(example.services.Auth) auth = providers.Factory(example.services.Auth)
auth.kwargs(db=Platform.database, auth.add_kwargs(db=Platform.database,
token_ttl=3600) token_ttl=3600)
You can get more *Dependency Injector* examples in ``/examples`` directory on You can get more *Dependency Injector* examples in ``/examples`` directory on

View File

@ -1,148 +1,13 @@
"""Injections module.""" """Injections module."""
import itertools
import six import six
from dependency_injector.utils import ( from dependency_injector.providers.base import (
is_provider, _parse_positional_injections,
is_injection, _parse_keyword_injections,
is_arg_injection,
is_kwarg_injection,
is_delegated_provider,
fetch_cls_init,
) )
from dependency_injector import utils
from dependency_injector.errors import Error from dependency_injector import errors
@six.python_2_unicode_compatible
class Injection(object):
"""Base injection class.
All injections extend this class.
.. py:attribute:: injectable
Injectable value, could be provider or any other object.
:type: object | :py:class:`dependency_injector.providers.Provider`
.. py:attribute:: call_injectable
Flag that is set to ``True`` if it is needed to call injectable.
Injectable needs to be called if it is not delegated provider.
:type: bool
"""
__IS_INJECTION__ = True
__slots__ = ('injectable', 'get_value')
def __init__(self, injectable):
"""Initializer.
:param injectable: Injectable value, could be provider or any
other object.
:type injectable: object |
:py:class:`dependency_injector.providers.Provider`
"""
self.injectable = injectable
if not is_provider(injectable) or is_delegated_provider(injectable):
def injectable():
return self.injectable
self.get_value = injectable
super(Injection, self).__init__()
def __str__(self):
"""Return string representation of provider.
:rtype: str
"""
return '<{injection}({injectable}) at {address}>'.format(
injection='.'.join((self.__class__.__module__,
self.__class__.__name__)),
injectable=repr(self.injectable),
address=hex(id(self)))
__repr__ = __str__
class Arg(Injection):
"""Positional argument injection."""
__IS_ARG_INJECTION__ = True
@six.python_2_unicode_compatible
class _NamedInjection(Injection):
"""Base class of named injections.
.. py:attribute:: name
Injection target's name (keyword argument, attribute).
:type: str
"""
__slots__ = ('name',)
def __init__(self, name, injectable):
"""Initializer.
:param name: Injection target's name.
:type name: str
:param injectable: Injectable value, could be provider or any
other object.
:type injectable: object |
:py:class:`dependency_injector.providers.Provider`
"""
self.name = name
super(_NamedInjection, self).__init__(injectable)
def __str__(self):
"""Return string representation of provider.
:rtype: str
"""
return '<{injection}({name}, {injectable}) at {address}>'.format(
name=repr(self.name),
injection='.'.join((self.__class__.__module__,
self.__class__.__name__)),
injectable=repr(self.injectable),
address=hex(id(self)))
__repr__ = __str__
class KwArg(_NamedInjection):
"""Keyword argument injection.
.. py:attribute:: name
Keyword argument's name.
:type: str
"""
__IS_KWARG_INJECTION__ = True
class Attribute(_NamedInjection):
"""Attribute injection.
.. py:attribute:: name
Attribute's name.
:type: str
"""
__IS_ATTRIBUTE_INJECTION__ = True
def inject(*args, **kwargs): def inject(*args, **kwargs):
@ -157,24 +22,18 @@ def inject(*args, **kwargs):
.. code-block:: python .. code-block:: python
# Positional arguments injections (simplified syntax): # Positional arguments injections:
@inject(1, 2) @inject(1, 2)
def some_function(arg1, arg2): def some_function(arg1, arg2):
pass pass
# Keyword arguments injections (simplified syntax): # Keyword arguments injections:
@inject(arg1=1) @inject(arg1=1)
@inject(arg2=2) @inject(arg2=2)
def some_function(arg1, arg2): def some_function(arg1, arg2):
pass pass
# Keyword arguments injections (extended (full) syntax): # Keyword arguments injections into class init:
@inject(KwArg('arg1', 1))
@inject(KwArg('arg2', 2))
def some_function(arg1, arg2):
pass
# Keyword arguments injections into class init (simplified syntax):
@inject(arg1=1) @inject(arg1=1)
@inject(arg2=2) @inject(arg2=2)
class SomeClass(object): class SomeClass(object):
@ -191,16 +50,16 @@ def inject(*args, **kwargs):
:return: Class / callable decorator :return: Class / callable decorator
:rtype: (callable) -> (type | callable) :rtype: (callable) -> (type | callable)
""" """
arg_injections = _parse_args_injections(args) arg_injections = _parse_positional_injections(args)
kwarg_injections = _parse_kwargs_injections(args, kwargs) kwarg_injections = _parse_keyword_injections(kwargs)
def decorator(callback_or_cls): def decorator(callback_or_cls):
"""Dependency injection decorator.""" """Dependency injection decorator."""
if isinstance(callback_or_cls, six.class_types): if isinstance(callback_or_cls, six.class_types):
cls = callback_or_cls cls = callback_or_cls
cls_init = fetch_cls_init(cls) cls_init = utils.fetch_cls_init(cls)
if not cls_init: if not cls_init:
raise Error( raise errors.Error(
'Class {0}.{1} has no __init__() '.format(cls.__module__, 'Class {0}.{1} has no __init__() '.format(cls.__module__,
cls.__name__) + cls.__name__) +
'method and could not be decorated with @inject decorator') 'method and could not be decorated with @inject decorator')
@ -211,19 +70,18 @@ def inject(*args, **kwargs):
if hasattr(callback, '__INJECT_DECORATED__'): if hasattr(callback, '__INJECT_DECORATED__'):
callback.args += arg_injections callback.args += arg_injections
callback.kwargs += kwarg_injections callback.kwargs.update(kwarg_injections)
callback.injections += arg_injections + kwarg_injections
return callback return callback
@six.wraps(callback) @six.wraps(callback)
def decorated(*args, **kwargs): def decorated(*args, **kwargs):
"""Decorated with dependency injection callback.""" """Decorated with dependency injection callback."""
if decorated.args: if decorated.args:
args = tuple(arg.get_value() for arg in decorated.args) + args args = tuple(arg.inject() for arg in decorated.args) + args
for kwarg in decorated.kwargs: for name, arg in six.iteritems(decorated.kwargs):
if kwarg.name not in kwargs: if name not in kwargs:
kwargs[kwarg.name] = kwarg.get_value() kwargs[name] = arg.inject()
return callback(*args, **kwargs) return callback(*args, **kwargs)
@ -231,23 +89,6 @@ def inject(*args, **kwargs):
decorated.origin = callback decorated.origin = callback
decorated.args = arg_injections decorated.args = arg_injections
decorated.kwargs = kwarg_injections decorated.kwargs = kwarg_injections
decorated.injections = arg_injections + kwarg_injections
return decorated return decorated
return decorator return decorator
def _parse_args_injections(args):
return tuple(Arg(arg) if not is_injection(arg) else arg
for arg in args
if not is_injection(arg) or is_arg_injection(arg))
def _parse_kwargs_injections(args, kwargs):
kwarg_injections = tuple(injection
for injection in args
if is_kwarg_injection(injection))
if kwargs:
kwarg_injections += tuple(itertools.starmap(KwArg,
six.iteritems(kwargs)))
return kwarg_injections

View File

@ -5,6 +5,8 @@ from dependency_injector.providers.base import (
Delegate, Delegate,
Object, Object,
ExternalDependency, ExternalDependency,
OverridingContext,
override,
) )
from dependency_injector.providers.callable import ( from dependency_injector.providers.callable import (
Callable, Callable,
@ -16,10 +18,6 @@ from dependency_injector.providers.creational import (
Singleton, Singleton,
DelegatedSingleton, DelegatedSingleton,
) )
from dependency_injector.providers.utils import (
OverridingContext,
override,
)
__all__ = ( __all__ = (
@ -28,6 +26,9 @@ __all__ = (
'Object', 'Object',
'ExternalDependency', 'ExternalDependency',
'OverridingContext',
'override',
'Callable', 'Callable',
'DelegatedCallable', 'DelegatedCallable',
@ -35,7 +36,4 @@ __all__ = (
'DelegatedFactory', 'DelegatedFactory',
'Singleton', 'Singleton',
'DelegatedSingleton', 'DelegatedSingleton',
'OverridingContext',
'override',
) )

View File

@ -2,7 +2,6 @@
import six import six
from dependency_injector.providers.utils import OverridingContext
from dependency_injector.errors import Error from dependency_injector.errors import Error
from dependency_injector.utils import ( from dependency_injector.utils import (
is_provider, is_provider,
@ -84,6 +83,13 @@ class Provider(object):
"""Call last overriding provider and return result.""" """Call last overriding provider and return result."""
return self.last_overriding(*args, **kwargs) return self.last_overriding(*args, **kwargs)
def inject(self):
"""Injection strategy implementation.
:rtype: object
"""
return self.provide()
@property @property
def is_overridden(self): def is_overridden(self):
"""Read-only property that is set to ``True`` if provider is overridden. """Read-only property that is set to ``True`` if provider is overridden.
@ -349,3 +355,77 @@ class ExternalDependency(Provider):
return represent_provider(provider=self, provides=self.instance_of) return represent_provider(provider=self, provides=self.instance_of)
__repr__ = __str__ __repr__ = __str__
class OverridingContext(object):
"""Provider overriding context.
:py:class:`OverridingContext` is used by :py:meth:`Provider.override` for
implemeting ``with`` contexts. When :py:class:`OverridingContext` is
closed, overriding that was created in this context is dropped also.
.. code-block:: python
with provider.override(another_provider):
assert provider.is_overridden
assert not provider.is_overridden
"""
def __init__(self, overridden, overriding):
"""Initializer.
:param overridden: Overridden provider.
:type overridden: :py:class:`Provider`
:param overriding: Overriding provider.
:type overriding: :py:class:`Provider`
"""
self.overridden = overridden
self.overriding = overriding
def __enter__(self):
"""Do nothing."""
return self.overriding
def __exit__(self, *_):
"""Exit overriding context."""
self.overridden.reset_last_overriding()
def override(overridden):
"""Decorator for overriding providers.
This decorator overrides ``overridden`` provider by decorated one.
.. code-block:: python
@Factory
class SomeClass(object):
pass
@override(SomeClass)
@Factory
class ExtendedSomeClass(SomeClass.cls):
pass
:param overridden: Provider that should be overridden.
:type overridden: :py:class:`Provider`
:return: Overriding provider.
:rtype: :py:class:`Provider`
"""
def decorator(overriding):
overridden.override(overriding)
return overriding
return decorator
def _parse_positional_injections(args):
return tuple(arg if is_provider(arg) else Object(arg)
for arg in args)
def _parse_keyword_injections(kwargs):
return dict((name, arg if is_provider(arg) else Object(arg))
for name, arg in six.iteritems(kwargs))

View File

@ -2,8 +2,11 @@
import six import six
from dependency_injector.providers.base import Provider from dependency_injector.providers.base import (
from dependency_injector.injections import Arg, KwArg Provider,
_parse_positional_injections,
_parse_keyword_injections,
)
from dependency_injector.utils import represent_provider from dependency_injector.utils import represent_provider
from dependency_injector.errors import Error from dependency_injector.errors import Error
@ -26,17 +29,17 @@ class Callable(Provider):
# or # or
some_function = Callable(some_function) \ some_function = Callable(some_function) \
.args('positional_arg1', 'positional_arg2') \ .add_args('positional_arg1', 'positional_arg2') \
.kwargs(keyword_argument1=3, keyword_argument=4) .add_kwargs(keyword_argument1=3, keyword_argument=4)
# or # or
some_function = Callable(some_function) some_function = Callable(some_function)
some_function.args('positional_arg1', 'positional_arg2') some_function.add_args('positional_arg1', 'positional_arg2')
some_function.kwargs(keyword_argument1=3, keyword_argument=4) some_function.add_kwargs(keyword_argument1=3, keyword_argument=4)
""" """
__slots__ = ('_provides', '_args', '_kwargs') __slots__ = ('provides', 'args', 'kwargs')
def __init__(self, provides, *args, **kwargs): def __init__(self, provides, *args, **kwargs):
"""Initializer. """Initializer.
@ -50,24 +53,17 @@ class Callable(Provider):
self.__class__.__name__)), self.__class__.__name__)),
provides)) provides))
self._provides = provides self.provides = provides
self._args = tuple()
self._kwargs = tuple()
self.args(*args) self.args = tuple()
self.kwargs(**kwargs) self.kwargs = dict()
self.add_args(*args)
self.add_kwargs(**kwargs)
super(Callable, self).__init__() super(Callable, self).__init__()
@property def add_args(self, *args):
def injections(self):
"""Read-only tuple of all injections.
:rtype: tuple[:py:class:`dependency_injector.injections.Injection`]
"""
return self._args + self._kwargs
def args(self, *args):
"""Add postional argument injections. """Add postional argument injections.
:param args: Tuple of injections. :param args: Tuple of injections.
@ -75,10 +71,10 @@ class Callable(Provider):
:return: Reference ``self`` :return: Reference ``self``
""" """
self._args += tuple(Arg(value) for value in args) self.args += _parse_positional_injections(args)
return self return self
def kwargs(self, **kwargs): def add_kwargs(self, **kwargs):
"""Add keyword argument injections. """Add keyword argument injections.
:param kwargs: Dictionary of injections. :param kwargs: Dictionary of injections.
@ -86,8 +82,7 @@ class Callable(Provider):
:return: Reference ``self`` :return: Reference ``self``
""" """
self._kwargs += tuple(KwArg(name, value) self.kwargs.update(_parse_keyword_injections(kwargs))
for name, value in six.iteritems(kwargs))
return self return self
def _provide(self, *args, **kwargs): def _provide(self, *args, **kwargs):
@ -101,21 +96,21 @@ class Callable(Provider):
:rtype: object :rtype: object
""" """
if self._args: if self.args:
args = tuple(arg.get_value() for arg in self._args) + args args = tuple(arg.inject() for arg in self.args) + args
for kwarg in self._kwargs: for name, arg in six.iteritems(self.kwargs):
if kwarg.name not in kwargs: if name not in kwargs:
kwargs[kwarg.name] = kwarg.get_value() kwargs[name] = arg.inject()
return self._provides(*args, **kwargs) return self.provides(*args, **kwargs)
def __str__(self): def __str__(self):
"""Return string representation of provider. """Return string representation of provider.
:rtype: str :rtype: str
""" """
return represent_provider(provider=self, provides=self._provides) return represent_provider(provider=self, provides=self.provides)
__repr__ = __str__ __repr__ = __str__
@ -127,4 +122,9 @@ class DelegatedCallable(Callable):
"as is". "as is".
""" """
__IS_DELEGATED__ = True def inject(self):
"""Injection strategy implementation.
:rtype: object
"""
return self

View File

@ -3,7 +3,7 @@
import six import six
from dependency_injector.providers.callable import Callable from dependency_injector.providers.callable import Callable
from dependency_injector.injections import Attribute from dependency_injector.providers.base import _parse_keyword_injections
from dependency_injector.utils import GLOBAL_LOCK from dependency_injector.utils import GLOBAL_LOCK
from dependency_injector.errors import Error from dependency_injector.errors import Error
@ -25,14 +25,14 @@ class Factory(Callable):
# or # or
factory = Factory(SomeClass) \ factory = Factory(SomeClass) \
.args('positional_arg1', 'positional_arg2') \ .add_args('positional_arg1', 'positional_arg2') \
.kwargs(keyword_argument1=3, keyword_argument=4) .add_kwargs(keyword_argument1=3, keyword_argument=4)
# or # or
factory = Factory(SomeClass) factory = Factory(SomeClass)
factory.args('positional_arg1', 'positional_arg2') factory.add_args('positional_arg1', 'positional_arg2')
factory.kwargs(keyword_argument1=3, keyword_argument=4) factory.add_kwargs(keyword_argument1=3, keyword_argument=4)
Attribute injections are defined by using :py:meth:`Factory.attributes`: Attribute injections are defined by using :py:meth:`Factory.attributes`:
@ -40,7 +40,7 @@ class Factory(Callable):
.. code-block:: python .. code-block:: python
factory = Factory(SomeClass) \ factory = Factory(SomeClass) \
.attributes(attribute1=1, attribute2=2) .add_attributes(attribute1=1, attribute2=2)
Retrieving of provided instance can be performed via calling Retrieving of provided instance can be performed via calling
:py:class:`Factory` object: :py:class:`Factory` object:
@ -68,7 +68,7 @@ class Factory(Callable):
provided_type = None provided_type = None
__slots__ = ('cls', '_attributes') __slots__ = ('cls', 'attributes')
def __init__(self, provides, *args, **kwargs): def __init__(self, provides, *args, **kwargs):
"""Initializer. """Initializer.
@ -82,21 +82,13 @@ class Factory(Callable):
raise Error('{0} can provide only {1} instances'.format( raise Error('{0} can provide only {1} instances'.format(
self.__class__, self.__class__.provided_type)) self.__class__, self.__class__.provided_type))
self._attributes = tuple() self.attributes = dict()
super(Factory, self).__init__(provides, *args, **kwargs) super(Factory, self).__init__(provides, *args, **kwargs)
self.cls = self._provides self.cls = self.provides
@property def add_attributes(self, **kwargs):
def injections(self):
"""Read-only tuple of all injections.
:rtype: tuple[:py:class:`dependency_injector.injections.Injection`]
"""
return super(Factory, self).injections + self._attributes
def attributes(self, **kwargs):
"""Add attribute injections. """Add attribute injections.
:param kwargs: Dictionary of injections. :param kwargs: Dictionary of injections.
@ -104,8 +96,7 @@ class Factory(Callable):
:return: Reference ``self`` :return: Reference ``self``
""" """
self._attributes += tuple(Attribute(name, value) self.attributes.update(_parse_keyword_injections(kwargs))
for name, value in six.iteritems(kwargs))
return self return self
def _provide(self, *args, **kwargs): def _provide(self, *args, **kwargs):
@ -119,17 +110,10 @@ class Factory(Callable):
:rtype: object :rtype: object
""" """
if self._args: instance = super(Factory, self)._provide(*args, **kwargs)
args = tuple(arg.get_value() for arg in self._args) + args
for kwarg in self._kwargs: for name, arg in six.iteritems(self.attributes):
if kwarg.name not in kwargs: setattr(instance, name, arg.inject())
kwargs[kwarg.name] = kwarg.get_value()
instance = self._provides(*args, **kwargs)
for attribute in self._attributes:
setattr(instance, attribute.name, attribute.get_value())
return instance return instance
@ -156,7 +140,12 @@ class DelegatedFactory(Factory):
:type: type :type: type
""" """
__IS_DELEGATED__ = True def inject(self):
"""Injection strategy implementation.
:rtype: object
"""
return self
class Singleton(Factory): class Singleton(Factory):
@ -193,7 +182,7 @@ class Singleton(Factory):
:type: type :type: type
""" """
__slots__ = ('_instance',) __slots__ = ('instance',)
def __init__(self, provides, *args, **kwargs): def __init__(self, provides, *args, **kwargs):
"""Initializer. """Initializer.
@ -202,7 +191,7 @@ class Singleton(Factory):
for creation. for creation.
:type provides: type | callable :type provides: type | callable
""" """
self._instance = None self.instance = None
super(Singleton, self).__init__(provides, *args, **kwargs) super(Singleton, self).__init__(provides, *args, **kwargs)
def reset(self): def reset(self):
@ -210,7 +199,7 @@ class Singleton(Factory):
:rtype: None :rtype: None
""" """
self._instance = None self.instance = None
def _provide(self, *args, **kwargs): def _provide(self, *args, **kwargs):
"""Return provided instance. """Return provided instance.
@ -223,13 +212,13 @@ class Singleton(Factory):
:rtype: object :rtype: object
""" """
if self._instance: if self.instance:
return self._instance return self.instance
with GLOBAL_LOCK: with GLOBAL_LOCK:
self._instance = super(Singleton, self)._provide(*args, **kwargs) self.instance = super(Singleton, self)._provide(*args, **kwargs)
return self._instance return self.instance
class DelegatedSingleton(Singleton): class DelegatedSingleton(Singleton):
@ -254,4 +243,9 @@ class DelegatedSingleton(Singleton):
:type: type :type: type
""" """
__IS_DELEGATED__ = True def inject(self):
"""Injection strategy implementation.
:rtype: object
"""
return self

View File

@ -1,65 +0,0 @@
"""Dependency injector provider utils."""
class OverridingContext(object):
"""Provider overriding context.
:py:class:`OverridingContext` is used by :py:meth:`Provider.override` for
implemeting ``with`` contexts. When :py:class:`OverridingContext` is
closed, overriding that was created in this context is dropped also.
.. code-block:: python
with provider.override(another_provider):
assert provider.is_overridden
assert not provider.is_overridden
"""
def __init__(self, overridden, overriding):
"""Initializer.
:param overridden: Overridden provider.
:type overridden: :py:class:`Provider`
:param overriding: Overriding provider.
:type overriding: :py:class:`Provider`
"""
self.overridden = overridden
self.overriding = overriding
def __enter__(self):
"""Do nothing."""
return self.overriding
def __exit__(self, *_):
"""Exit overriding context."""
self.overridden.reset_last_overriding()
def override(overridden):
"""Decorator for overriding providers.
This decorator overrides ``overridden`` provider by decorated one.
.. code-block:: python
@Factory
class SomeClass(object):
pass
@override(SomeClass)
@Factory
class ExtendedSomeClass(SomeClass.cls):
pass
:param overridden: Provider that should be overridden.
:type overridden: :py:class:`Provider`
:return: Overriding provider.
:rtype: :py:class:`Provider`
"""
def decorator(overriding):
overridden.override(overriding)
return overriding
return decorator

View File

@ -7,7 +7,7 @@ import threading
import six import six
from dependency_injector.errors import Error from dependency_injector import errors
GLOBAL_LOCK = threading.RLock() GLOBAL_LOCK = threading.RLock()
@ -54,93 +54,11 @@ def ensure_is_provider(instance):
:rtype: :py:class:`dependency_injector.providers.Provider` :rtype: :py:class:`dependency_injector.providers.Provider`
""" """
if not is_provider(instance): if not is_provider(instance):
raise Error('Expected provider instance, ' raise errors.Error('Expected provider instance, '
'got {0}'.format(str(instance))) 'got {0}'.format(str(instance)))
return instance return instance
def is_delegated_provider(instance):
"""Check if instance is delegated provider instance.
:param instance: Instance to be checked.
:type instance: object
:rtype: bool
"""
return (is_provider(instance) and
hasattr(instance, '__IS_DELEGATED__') and
getattr(instance, '__IS_DELEGATED__') is True)
def is_injection(instance):
"""Check if instance is injection instance.
:param instance: Instance to be checked.
:type instance: object
:rtype: bool
"""
return (not isinstance(instance, six.class_types) and
hasattr(instance, '__IS_INJECTION__') and
getattr(instance, '__IS_INJECTION__') is True)
def ensure_is_injection(instance):
"""Check if instance is injection instance and return it.
:param instance: Instance to be checked.
:type instance: object
:raise: :py:exc:`dependency_injector.errors.Error` if provided instance is
not injection.
:rtype: :py:class:`dependency_injector.injections.Injection`
"""
if not is_injection(instance):
raise Error('Expected injection instance, '
'got {0}'.format(str(instance)))
return instance
def is_arg_injection(instance):
"""Check if instance is positional argument injection instance.
:param instance: Instance to be checked.
:type instance: object
:rtype: bool
"""
return (not isinstance(instance, six.class_types) and
hasattr(instance, '__IS_ARG_INJECTION__') and
getattr(instance, '__IS_ARG_INJECTION__', False) is True)
def is_kwarg_injection(instance):
"""Check if instance is keyword argument injection instance.
:param instance: Instance to be checked.
:type instance: object
:rtype: bool
"""
return (not isinstance(instance, six.class_types) and
hasattr(instance, '__IS_KWARG_INJECTION__') and
getattr(instance, '__IS_KWARG_INJECTION__', False) is True)
def is_attribute_injection(instance):
"""Check if instance is attribute injection instance.
:param instance: Instance to be checked.
:type instance: object
:rtype: bool
"""
return (not isinstance(instance, six.class_types) and
hasattr(instance, '__IS_ATTRIBUTE_INJECTION__') and
getattr(instance, '__IS_ATTRIBUTE_INJECTION__', False) is True)
def is_catalog(instance): def is_catalog(instance):
"""Check if instance is catalog instance. """Check if instance is catalog instance.

View File

@ -15,10 +15,10 @@ class Platform(catalogs.DeclarativeCatalog):
"""Catalog of platform service providers.""" """Catalog of platform service providers."""
database = providers.Singleton(sqlite3.connect) \ database = providers.Singleton(sqlite3.connect) \
.args(':memory:') .add_args(':memory:')
s3 = providers.Singleton(boto.s3.connection.S3Connection) \ s3 = providers.Singleton(boto.s3.connection.S3Connection) \
.kwargs(aws_access_key_id='KEY', .add_kwargs(aws_access_key_id='KEY',
aws_secret_access_key='SECRET') aws_secret_access_key='SECRET')
@ -26,12 +26,12 @@ class Services(catalogs.DeclarativeCatalog):
"""Catalog of business service providers.""" """Catalog of business service providers."""
users = providers.Factory(example.services.Users) \ users = providers.Factory(example.services.Users) \
.kwargs(db=Platform.database) .add_kwargs(db=Platform.database)
photos = providers.Factory(example.services.Photos) \ photos = providers.Factory(example.services.Photos) \
.kwargs(db=Platform.database, .add_kwargs(db=Platform.database,
s3=Platform.s3) s3=Platform.s3)
auth = providers.Factory(example.services.Auth) \ auth = providers.Factory(example.services.Auth) \
.kwargs(db=Platform.database, .add_kwargs(db=Platform.database,
token_ttl=3600) token_ttl=3600)

View File

@ -15,10 +15,10 @@ class Platform(catalogs.DeclarativeCatalog):
"""Catalog of platform service providers.""" """Catalog of platform service providers."""
database = providers.Singleton(sqlite3.connect) database = providers.Singleton(sqlite3.connect)
database.args(':memory:') database.add_args(':memory:')
s3 = providers.Singleton(boto.s3.connection.S3Connection) s3 = providers.Singleton(boto.s3.connection.S3Connection)
s3.kwargs(aws_access_key_id='KEY', s3.add_kwargs(aws_access_key_id='KEY',
aws_secret_access_key='SECRET') aws_secret_access_key='SECRET')
@ -26,12 +26,12 @@ class Services(catalogs.DeclarativeCatalog):
"""Catalog of business service providers.""" """Catalog of business service providers."""
users = providers.Factory(example.services.Users) users = providers.Factory(example.services.Users)
users.kwargs(db=Platform.database) users.add_kwargs(db=Platform.database)
photos = providers.Factory(example.services.Photos) photos = providers.Factory(example.services.Photos)
photos.kwargs(db=Platform.database, photos.add_kwargs(db=Platform.database,
s3=Platform.s3) s3=Platform.s3)
auth = providers.Factory(example.services.Auth) auth = providers.Factory(example.services.Auth)
auth.kwargs(db=Platform.database, auth.add_kwargs(db=Platform.database,
token_ttl=3600) token_ttl=3600)

View File

@ -383,7 +383,7 @@ class CopyingTests(unittest.TestCase):
class CatalogA(catalogs.DeclarativeCatalog): class CatalogA(catalogs.DeclarativeCatalog):
p11 = providers.Object(0) p11 = providers.Object(0)
p12 = providers.Factory(dict) \ p12 = providers.Factory(dict) \
.kwargs(p11=p11) .add_kwargs(p11=p11)
@catalogs.copy(CatalogA) @catalogs.copy(CatalogA)
class CatalogA1(CatalogA): class CatalogA1(CatalogA):
@ -404,9 +404,9 @@ class CopyingTests(unittest.TestCase):
self.assertIsNot(CatalogA1.p11, CatalogA2.p11) self.assertIsNot(CatalogA1.p11, CatalogA2.p11)
self.assertIsNot(CatalogA1.p12, CatalogA2.p12) self.assertIsNot(CatalogA1.p12, CatalogA2.p12)
self.assertIs(CatalogA.p12.injections[0].injectable, CatalogA.p11) self.assertIs(CatalogA.p12.kwargs['p11'], CatalogA.p11)
self.assertIs(CatalogA1.p12.injections[0].injectable, CatalogA1.p11) self.assertIs(CatalogA1.p12.kwargs['p11'], CatalogA1.p11)
self.assertIs(CatalogA2.p12.injections[0].injectable, CatalogA2.p11) self.assertIs(CatalogA2.p12.kwargs['p11'], CatalogA2.p11)
self.assertEqual(CatalogA.p12(), dict(p11=0)) self.assertEqual(CatalogA.p12(), dict(p11=0))
self.assertEqual(CatalogA1.p12(), dict(p11=1)) self.assertEqual(CatalogA1.p12(), dict(p11=1))

View File

@ -30,47 +30,52 @@ class CallableTests(unittest.TestCase):
New simplified syntax. New simplified syntax.
""" """
provider = providers.Callable(self.example) \ provider = providers.Callable(self.example,
.args(1, 2, 3, 4) 1, 2, 3, 4)
self.assertTupleEqual(provider(), (1, 2, 3, 4)) self.assertTupleEqual(provider(), (1, 2, 3, 4))
def test_call_with_keyword_args(self): def test_call_with_keyword_args(self):
"""Test call with keyword args.""" """Test call with keyword args."""
provider = providers.Callable(self.example) \ provider = providers.Callable(self.example,
.kwargs(arg1=1, arg2=2, arg3=3, arg4=4) arg1=1, arg2=2, arg3=3, arg4=4)
self.assertTupleEqual(provider(), (1, 2, 3, 4)) self.assertTupleEqual(provider(), (1, 2, 3, 4))
def test_call_with_positional_and_keyword_args(self): def test_call_with_positional_and_keyword_args(self):
"""Test call with positional and keyword args.""" """Test call with positional and keyword args."""
provider = providers.Callable(self.example) \ provider = providers.Callable(self.example,
.args(1, 2) \ 1, 2,
.kwargs(arg3=3, arg4=4) arg3=3, arg4=4)
self.assertTupleEqual(provider(), (1, 2, 3, 4)) self.assertTupleEqual(provider(), (1, 2, 3, 4))
def test_call_with_context_args(self): def test_call_with_context_args(self):
"""Test call with context args.""" """Test call with context args."""
provider = providers.Callable(self.example) \ provider = providers.Callable(self.example, 1, 2)
.args(1, 2)
self.assertTupleEqual(provider(3, 4), (1, 2, 3, 4)) self.assertTupleEqual(provider(3, 4), (1, 2, 3, 4))
def test_call_with_context_kwargs(self): def test_call_with_context_kwargs(self):
"""Test call with context kwargs.""" """Test call with context kwargs."""
provider = providers.Callable(self.example) \ provider = providers.Callable(self.example, arg1=1)
.kwargs(arg1=1)
self.assertTupleEqual(provider(arg2=2, arg3=3, arg4=4), (1, 2, 3, 4)) self.assertTupleEqual(provider(arg2=2, arg3=3, arg4=4), (1, 2, 3, 4))
def test_call_with_context_args_and_kwargs(self): def test_call_with_context_args_and_kwargs(self):
"""Test call with context args and kwargs.""" """Test call with context args and kwargs."""
provider = providers.Callable(self.example) \ provider = providers.Callable(self.example, 1)
.args(1)
self.assertTupleEqual(provider(2, arg3=3, arg4=4), (1, 2, 3, 4)) self.assertTupleEqual(provider(2, arg3=3, arg4=4), (1, 2, 3, 4))
def test_fluent_interface(self):
"""Test injections definition with fluent interface."""
provider = providers.Singleton(self.example) \
.add_args(1, 2) \
.add_kwargs(arg3=3, arg4=4) \
self.assertTupleEqual(provider(), (1, 2, 3, 4))
def test_call_overridden(self): def test_call_overridden(self):
"""Test creation of new instances on overridden provider.""" """Test creation of new instances on overridden provider."""
provider = providers.Callable(self.example) provider = providers.Callable(self.example)
@ -80,21 +85,9 @@ class CallableTests(unittest.TestCase):
self.assertTupleEqual(provider(), (1, 2, 3, 4)) self.assertTupleEqual(provider(), (1, 2, 3, 4))
def test_injections(self):
"""Test getting a full list of injections using injections property."""
provider = providers.Callable(self.example) \
.args(1, 2) \
.kwargs(arg3=3, arg4=4)
self.assertEquals(len(provider.injections), 4)
def test_repr(self): def test_repr(self):
"""Test representation of provider.""" """Test representation of provider."""
provider = providers.Callable(self.example) \ provider = providers.Callable(self.example)
.kwargs(arg1=providers.Factory(dict),
arg2=providers.Factory(list),
arg3=providers.Factory(set),
arg4=providers.Factory(tuple))
self.assertEqual(repr(provider), self.assertEqual(repr(provider),
'<dependency_injector.providers.callable.' '<dependency_injector.providers.callable.'
@ -117,6 +110,5 @@ class DelegatedCallableTests(unittest.TestCase):
def test_is_delegated_provider(self): def test_is_delegated_provider(self):
"""Test is_delegated_provider.""" """Test is_delegated_provider."""
self.assertTrue(utils.is_delegated_provider( provider = providers.DelegatedCallable(len)
providers.DelegatedCallable(len))) self.assertIs(provider.inject(), provider)
self.assertFalse(utils.is_delegated_provider(providers.Callable(len)))

View File

@ -42,8 +42,7 @@ class FactoryTests(unittest.TestCase):
provided_type = Example provided_type = Example
example_provider = ExampleProvider(Example) \ example_provider = ExampleProvider(Example, 1, 2)
.args(1, 2)
self.assertIsInstance(example_provider(), Example) self.assertIsInstance(example_provider(), Example)
@ -57,8 +56,7 @@ class FactoryTests(unittest.TestCase):
class NewExampe(Example): class NewExampe(Example):
"""Example class subclass.""" """Example class subclass."""
example_provider = ExampleProvider(NewExampe) \ example_provider = ExampleProvider(NewExampe, 1, 2)
.args(1, 2)
self.assertIsInstance(example_provider(), NewExampe) self.assertIsInstance(example_provider(), NewExampe)
@ -85,8 +83,7 @@ class FactoryTests(unittest.TestCase):
def test_call_with_init_positional_args(self): def test_call_with_init_positional_args(self):
"""Test call with init positional args.""" """Test call with init positional args."""
provider = providers.Factory(Example) \ provider = providers.Factory(Example, 'i1', 'i2')
.args('i1', 'i2')
instance1 = provider() instance1 = provider()
instance2 = provider() instance2 = provider()
@ -103,8 +100,7 @@ class FactoryTests(unittest.TestCase):
def test_call_with_init_keyword_args(self): def test_call_with_init_keyword_args(self):
"""Test call with init keyword args.""" """Test call with init keyword args."""
provider = providers.Factory(Example) \ provider = providers.Factory(Example, init_arg1='i1', init_arg2='i2')
.kwargs(init_arg1='i1', init_arg2='i2')
instance1 = provider() instance1 = provider()
instance2 = provider() instance2 = provider()
@ -121,9 +117,7 @@ class FactoryTests(unittest.TestCase):
def test_call_with_init_positional_and_keyword_args(self): def test_call_with_init_positional_and_keyword_args(self):
"""Test call with init positional and keyword args.""" """Test call with init positional and keyword args."""
provider = providers.Factory(Example) \ provider = providers.Factory(Example, 'i1', init_arg2='i2')
.args('i1') \
.kwargs(init_arg2='i2')
instance1 = provider() instance1 = provider()
instance2 = provider() instance2 = provider()
@ -140,8 +134,8 @@ class FactoryTests(unittest.TestCase):
def test_call_with_attributes(self): def test_call_with_attributes(self):
"""Test call with attribute injections.""" """Test call with attribute injections."""
provider = providers.Factory(Example) \ provider = providers.Factory(Example)
.attributes(attribute1='a1', attribute2='a2') provider.add_attributes(attribute1='a1', attribute2='a2')
instance1 = provider() instance1 = provider()
instance2 = provider() instance2 = provider()
@ -158,8 +152,7 @@ class FactoryTests(unittest.TestCase):
def test_call_with_context_args(self): def test_call_with_context_args(self):
"""Test call with context args.""" """Test call with context args."""
provider = providers.Factory(Example) \ provider = providers.Factory(Example, 11, 22)
.args(11, 22)
instance = provider(33, 44) instance = provider(33, 44)
@ -170,8 +163,7 @@ class FactoryTests(unittest.TestCase):
def test_call_with_context_kwargs(self): def test_call_with_context_kwargs(self):
"""Test call with context kwargs.""" """Test call with context kwargs."""
provider = providers.Factory(Example) \ provider = providers.Factory(Example, init_arg1=1)
.kwargs(init_arg1=1)
instance1 = provider(init_arg2=22) instance1 = provider(init_arg2=22)
self.assertEqual(instance1.init_arg1, 1) self.assertEqual(instance1.init_arg1, 1)
@ -183,8 +175,7 @@ class FactoryTests(unittest.TestCase):
def test_call_with_context_args_and_kwargs(self): def test_call_with_context_args_and_kwargs(self):
"""Test call with context args and kwargs.""" """Test call with context args and kwargs."""
provider = providers.Factory(Example) \ provider = providers.Factory(Example, 11)
.args(11)
instance = provider(22, init_arg3=33, init_arg4=44) instance = provider(22, init_arg3=33, init_arg4=44)
@ -193,6 +184,22 @@ class FactoryTests(unittest.TestCase):
self.assertEqual(instance.init_arg3, 33) self.assertEqual(instance.init_arg3, 33)
self.assertEqual(instance.init_arg4, 44) self.assertEqual(instance.init_arg4, 44)
def test_fluent_interface(self):
"""Test injections definition with fluent interface."""
provider = providers.Factory(Example) \
.add_args(1, 2) \
.add_kwargs(init_arg3=3, init_arg4=4) \
.add_attributes(attribute1=5, attribute2=6)
instance = provider()
self.assertEqual(instance.init_arg1, 1)
self.assertEqual(instance.init_arg2, 2)
self.assertEqual(instance.init_arg3, 3)
self.assertEqual(instance.init_arg4, 4)
self.assertEqual(instance.attribute1, 5)
self.assertEqual(instance.attribute2, 6)
def test_call_overridden(self): def test_call_overridden(self):
"""Test call on overridden provider.""" """Test call on overridden provider."""
provider = providers.Factory(Example) provider = providers.Factory(Example)
@ -209,20 +216,9 @@ class FactoryTests(unittest.TestCase):
self.assertIsInstance(instance1, list) self.assertIsInstance(instance1, list)
self.assertIsInstance(instance2, list) self.assertIsInstance(instance2, list)
def test_injections(self):
"""Test getting a full list of injections using injections property."""
provider = providers.Factory(Example) \
.args(1) \
.kwargs(init_arg2=2) \
.attributes(attribute1=3, attribute2=4)
self.assertEquals(len(provider.injections), 4)
def test_repr(self): def test_repr(self):
"""Test representation of provider.""" """Test representation of provider."""
provider = providers.Factory(Example) \ provider = providers.Factory(Example)
.kwargs(init_arg1=providers.Factory(dict),
init_arg2=providers.Factory(list))
self.assertEqual(repr(provider), self.assertEqual(repr(provider),
'<dependency_injector.providers.creational.' '<dependency_injector.providers.creational.'
@ -245,10 +241,8 @@ class DelegatedFactoryTests(unittest.TestCase):
def test_is_delegated_provider(self): def test_is_delegated_provider(self):
"""Test is_delegated_provider.""" """Test is_delegated_provider."""
self.assertTrue(utils.is_delegated_provider( provider = providers.DelegatedFactory(object)
providers.DelegatedFactory(object))) self.assertIs(provider.inject(), provider)
self.assertFalse(utils.is_delegated_provider(
providers.Factory(object)))
class SingletonTests(unittest.TestCase): class SingletonTests(unittest.TestCase):
@ -273,8 +267,7 @@ class SingletonTests(unittest.TestCase):
provided_type = Example provided_type = Example
example_provider = ExampleProvider(Example) \ example_provider = ExampleProvider(Example, 1, 2)
.args(1, 2)
self.assertIsInstance(example_provider(), Example) self.assertIsInstance(example_provider(), Example)
@ -288,8 +281,7 @@ class SingletonTests(unittest.TestCase):
class NewExampe(Example): class NewExampe(Example):
"""Example class subclass.""" """Example class subclass."""
example_provider = ExampleProvider(NewExampe) \ example_provider = ExampleProvider(NewExampe, 1, 2)
.args(1, 2)
self.assertIsInstance(example_provider(), NewExampe) self.assertIsInstance(example_provider(), NewExampe)
@ -316,8 +308,7 @@ class SingletonTests(unittest.TestCase):
def test_call_with_init_positional_args(self): def test_call_with_init_positional_args(self):
"""Test getting of instances with init positional args.""" """Test getting of instances with init positional args."""
provider = providers.Singleton(Example) \ provider = providers.Singleton(Example, 'i1', 'i2')
.args('i1', 'i2')
instance1 = provider() instance1 = provider()
instance2 = provider() instance2 = provider()
@ -334,8 +325,7 @@ class SingletonTests(unittest.TestCase):
def test_call_with_init_keyword_args(self): def test_call_with_init_keyword_args(self):
"""Test getting of instances with init keyword args.""" """Test getting of instances with init keyword args."""
provider = providers.Singleton(Example) \ provider = providers.Singleton(Example, init_arg1='i1', init_arg2='i2')
.kwargs(init_arg1='i1', init_arg2='i2')
instance1 = provider() instance1 = provider()
instance2 = provider() instance2 = provider()
@ -352,9 +342,7 @@ class SingletonTests(unittest.TestCase):
def test_call_with_init_positional_and_keyword_args(self): def test_call_with_init_positional_and_keyword_args(self):
"""Test getting of instances with init positional and keyword args.""" """Test getting of instances with init positional and keyword args."""
provider = providers.Singleton(Example) \ provider = providers.Singleton(Example, 'i1', init_arg2='i2')
.args('i1') \
.kwargs(init_arg2='i2')
instance1 = provider() instance1 = provider()
instance2 = provider() instance2 = provider()
@ -371,8 +359,8 @@ class SingletonTests(unittest.TestCase):
def test_call_with_attributes(self): def test_call_with_attributes(self):
"""Test getting of instances with attribute injections.""" """Test getting of instances with attribute injections."""
provider = providers.Singleton(Example) \ provider = providers.Singleton(Example)
.attributes(attribute1='a1', attribute2='a2') provider.add_attributes(attribute1='a1', attribute2='a2')
instance1 = provider() instance1 = provider()
instance2 = provider() instance2 = provider()
@ -398,8 +386,7 @@ class SingletonTests(unittest.TestCase):
def test_call_with_context_kwargs(self): def test_call_with_context_kwargs(self):
"""Test getting of instances with context kwargs.""" """Test getting of instances with context kwargs."""
provider = providers.Singleton(Example) \ provider = providers.Singleton(Example, init_arg1=1)
.kwargs(init_arg1=1)
instance1 = provider(init_arg2=22) instance1 = provider(init_arg2=22)
self.assertEqual(instance1.init_arg1, 1) self.assertEqual(instance1.init_arg1, 1)
@ -412,8 +399,8 @@ class SingletonTests(unittest.TestCase):
def test_call_with_context_args_and_kwargs(self): def test_call_with_context_args_and_kwargs(self):
"""Test getting of instances with context args and kwargs.""" """Test getting of instances with context args and kwargs."""
provider = providers.Singleton(Example) \ provider = providers.Singleton(Example, 11)
.args(11)
instance = provider(22, init_arg3=33, init_arg4=44) instance = provider(22, init_arg3=33, init_arg4=44)
self.assertEqual(instance.init_arg1, 11) self.assertEqual(instance.init_arg1, 11)
@ -421,6 +408,22 @@ class SingletonTests(unittest.TestCase):
self.assertEqual(instance.init_arg3, 33) self.assertEqual(instance.init_arg3, 33)
self.assertEqual(instance.init_arg4, 44) self.assertEqual(instance.init_arg4, 44)
def test_fluent_interface(self):
"""Test injections definition with fluent interface."""
provider = providers.Singleton(Example) \
.add_args(1, 2) \
.add_kwargs(init_arg3=3, init_arg4=4) \
.add_attributes(attribute1=5, attribute2=6)
instance = provider()
self.assertEqual(instance.init_arg1, 1)
self.assertEqual(instance.init_arg2, 2)
self.assertEqual(instance.init_arg3, 3)
self.assertEqual(instance.init_arg4, 4)
self.assertEqual(instance.attribute1, 5)
self.assertEqual(instance.attribute2, 6)
def test_call_overridden(self): def test_call_overridden(self):
"""Test getting of instances on overridden provider.""" """Test getting of instances on overridden provider."""
provider = providers.Singleton(Example) provider = providers.Singleton(Example)
@ -437,15 +440,6 @@ class SingletonTests(unittest.TestCase):
self.assertIsInstance(instance1, object) self.assertIsInstance(instance1, object)
self.assertIsInstance(instance2, object) self.assertIsInstance(instance2, object)
def test_injections(self):
"""Test getting a full list of injections using injections property."""
provider = providers.Singleton(Example) \
.args(1) \
.kwargs(init_arg2=2) \
.attributes(attribute1=3, attribute2=4)
self.assertEquals(len(provider.injections), 4)
def test_reset(self): def test_reset(self):
"""Test creation and reset of single object.""" """Test creation and reset of single object."""
provider = providers.Singleton(object) provider = providers.Singleton(object)
@ -462,9 +456,7 @@ class SingletonTests(unittest.TestCase):
def test_repr(self): def test_repr(self):
"""Test representation of provider.""" """Test representation of provider."""
provider = providers.Singleton(Example) \ provider = providers.Singleton(Example)
.kwargs(init_arg1=providers.Factory(dict),
init_arg2=providers.Factory(list))
self.assertEqual(repr(provider), self.assertEqual(repr(provider),
'<dependency_injector.providers.creational.' '<dependency_injector.providers.creational.'
@ -488,10 +480,8 @@ class DelegatedSingletonTests(unittest.TestCase):
def test_is_delegated_provider(self): def test_is_delegated_provider(self):
"""Test is_delegated_provider.""" """Test is_delegated_provider."""
self.assertTrue(utils.is_delegated_provider( provider = providers.DelegatedSingleton(object)
providers.DelegatedSingleton(object))) self.assertIs(provider.inject(), provider)
self.assertFalse(utils.is_delegated_provider(
providers.Singleton(object)))
class FactoryAsDecoratorTests(unittest.TestCase): class FactoryAsDecoratorTests(unittest.TestCase):

View File

@ -7,97 +7,6 @@ from dependency_injector import providers
from dependency_injector import errors from dependency_injector import errors
class InjectionTests(unittest.TestCase):
"""Injection test cases."""
def test_init(self):
"""Test Injection creation and initialization."""
injection = injections.Injection('some_value')
self.assertEqual(injection.injectable, 'some_value')
def test_value_with_scalar_injectable(self):
"""Test Injection value property with scalar value."""
injection = injections.Injection('some_value')
self.assertEqual(injection.get_value(), 'some_value')
def test_value_with_provider_injectable(self):
"""Test Injection value property with provider."""
injection = injections.Injection(providers.Factory(object))
self.assertIsInstance(injection.get_value(), object)
def test_repr(self):
"""Test Injection representation."""
provider = providers.Factory(object)
injection = injections.Injection(provider)
self.assertEqual(
repr(injection),
'<dependency_injector.injections.Injection({0}) at {1}>'.format(
repr(provider),
hex(id(injection))))
class ArgTests(unittest.TestCase):
"""Positional arg injection test cases."""
def test_init(self):
"""Test Arg creation and initialization."""
injection = injections.Arg('some_value')
self.assertEqual(injection.injectable, 'some_value')
def test_repr(self):
"""Test Arg representation."""
provider = providers.Factory(object)
injection = injections.Arg(provider)
self.assertEqual(
repr(injection),
'<dependency_injector.injections.Arg({0}) at {1}>'.format(
repr(provider),
hex(id(injection))))
class KwArgTests(unittest.TestCase):
"""Keyword arg injection test cases."""
def test_init(self):
"""Test KwArg creation and initialization."""
injection = injections.KwArg('some_arg_name', 'some_value')
self.assertEqual(injection.name, 'some_arg_name')
self.assertEqual(injection.injectable, 'some_value')
def test_repr(self):
"""Test KwArg representation."""
provider = providers.Factory(object)
injection = injections.KwArg('name', provider)
self.assertEqual(
repr(injection),
'<dependency_injector.injections.KwArg({0}, {1}) at {2}>'.format(
repr('name'),
repr(provider),
hex(id(injection))))
class AttributeTests(unittest.TestCase):
"""Attribute injection test cases."""
def test_init(self):
"""Test Attribute creation and initialization."""
injection = injections.Attribute('some_arg_name', 'some_value')
self.assertEqual(injection.name, 'some_arg_name')
self.assertEqual(injection.injectable, 'some_value')
def test_repr(self):
"""Test Attribute representation."""
provider = providers.Factory(object)
injection = injections.Attribute('name', provider)
self.assertEqual(
repr(injection),
'<dependency_injector.injections.Attribute({0}, {1}) '
'at {2}>'.format(
repr('name'),
repr(provider),
hex(id(injection))))
class InjectTests(unittest.TestCase): class InjectTests(unittest.TestCase):
"""Inject decorator test cases.""" """Inject decorator test cases."""
@ -121,27 +30,6 @@ class InjectTests(unittest.TestCase):
self.assertIsInstance(b2, list) self.assertIsInstance(b2, list)
self.assertIsNot(b1, b2) self.assertIsNot(b1, b2)
def test_decorated_args_extended_syntax(self):
"""Test `inject()` decoration with args."""
provider1 = providers.Factory(object)
provider2 = providers.Factory(list)
@injections.inject(injections.Arg(provider1),
injections.Arg(provider2))
def test(a, b):
return a, b
a1, b1 = test()
a2, b2 = test()
self.assertIsInstance(a1, object)
self.assertIsInstance(a2, object)
self.assertIsNot(a1, a2)
self.assertIsInstance(b1, list)
self.assertIsInstance(b2, list)
self.assertIsNot(b1, b2)
def test_decorated_args_several_times(self): def test_decorated_args_several_times(self):
"""Test `inject()` decoration with args several times.""" """Test `inject()` decoration with args several times."""
provider1 = providers.Factory(object) provider1 = providers.Factory(object)
@ -248,27 +136,6 @@ class InjectTests(unittest.TestCase):
self.assertIsInstance(b2, list) self.assertIsInstance(b2, list)
self.assertIsNot(b1, b2) self.assertIsNot(b1, b2)
def test_injection_kwarg_syntax(self):
"""Test `inject()` decorated callback with "old" style using KwArg."""
provider = providers.Factory(list)
object_a = object()
@injections.inject(injections.KwArg('b', provider))
def test(a, b):
return a, b
a1, b1 = test(object_a)
a2, b2 = test(object_a)
self.assertIsInstance(a1, object)
self.assertIsInstance(a2, object)
self.assertIs(a1, object_a)
self.assertIs(a2, object_a)
self.assertIsInstance(b1, list)
self.assertIsInstance(b2, list)
self.assertIsNot(b1, b2)
def test_decorate_class_method(self): def test_decorate_class_method(self):
"""Test `inject()` decorator with class method.""" """Test `inject()` decorator with class method."""
class Test(object): class Test(object):

View File

@ -4,7 +4,6 @@ import unittest2 as unittest
from dependency_injector import utils from dependency_injector import utils
from dependency_injector import providers from dependency_injector import providers
from dependency_injector import injections
from dependency_injector import catalogs from dependency_injector import catalogs
from dependency_injector import errors from dependency_injector import errors
@ -72,135 +71,6 @@ class EnsureIsProviderTests(unittest.TestCase):
self.assertRaises(errors.Error, utils.ensure_is_provider, object()) self.assertRaises(errors.Error, utils.ensure_is_provider, object())
class IsInjectionTests(unittest.TestCase):
"""`is_injection()` test cases."""
def test_with_instance(self):
"""Test with instance."""
self.assertTrue(utils.is_injection(injections.Injection('value')))
def test_with_subclass_instances(self):
"""Test with subclass instances."""
self.assertTrue(utils.is_injection(injections.Arg('value')))
self.assertTrue(utils.is_injection(injections.KwArg('name',
'value')))
self.assertTrue(utils.is_injection(injections.Attribute('name',
'value')))
def test_with_class(self):
"""Test with class."""
self.assertFalse(utils.is_injection(injections.Injection))
def test_with_string(self):
"""Test with string."""
self.assertFalse(utils.is_injection('some_string'))
def test_with_object(self):
"""Test with object."""
self.assertFalse(utils.is_injection(object()))
class EnsureIsInjectionTests(unittest.TestCase):
"""`ensure_is_injection` test cases."""
def test_with_instance(self):
"""Test with instance."""
injection = injections.Injection('value')
self.assertIs(utils.ensure_is_injection(injection), injection)
def test_with_class(self):
"""Test with class."""
self.assertRaises(errors.Error,
utils.ensure_is_injection,
injections.Injection)
def test_with_string(self):
"""Test with string."""
self.assertRaises(errors.Error,
utils.ensure_is_injection,
'some_string')
def test_with_object(self):
"""Test with object."""
self.assertRaises(errors.Error,
utils.ensure_is_injection,
object())
class IsArgInjectionTests(unittest.TestCase):
"""`is_arg_injection()` test cases."""
def test_with_instance(self):
"""Test with instance."""
self.assertTrue(utils.is_arg_injection(injections.Arg('value')))
def test_with_class(self):
"""Test with class."""
self.assertFalse(utils.is_arg_injection(injections.Arg))
def test_with_parent_class(self):
"""Test with parent class."""
self.assertFalse(utils.is_arg_injection(injections.Injection))
def test_with_string(self):
"""Test with string."""
self.assertFalse(utils.is_arg_injection('some_string'))
def test_with_object(self):
"""Test with object."""
self.assertFalse(utils.is_arg_injection(object()))
class IsKwArgInjectionTests(unittest.TestCase):
"""`is_kwarg_injection()` test cases."""
def test_with_instance(self):
"""Test with instance."""
self.assertTrue(utils.is_kwarg_injection(injections.KwArg('name',
'value')))
def test_with_class(self):
"""Test with class."""
self.assertFalse(utils.is_kwarg_injection(injections.KwArg))
def test_with_parent_class(self):
"""Test with parent class."""
self.assertFalse(utils.is_kwarg_injection(injections.Injection))
def test_with_string(self):
"""Test with string."""
self.assertFalse(utils.is_kwarg_injection('some_string'))
def test_with_object(self):
"""Test with object."""
self.assertFalse(utils.is_kwarg_injection(object()))
class IsAttributeInjectionTests(unittest.TestCase):
"""`is_attribute_injection()` test cases."""
def test_with_instance(self):
"""Test with instance."""
self.assertTrue(utils.is_attribute_injection(
injections.Attribute('name', 'value')))
def test_with_class(self):
"""Test with class."""
self.assertFalse(utils.is_attribute_injection(injections.Attribute))
def test_with_parent_class(self):
"""Test with parent class."""
self.assertFalse(utils.is_attribute_injection(injections.Injection))
def test_with_string(self):
"""Test with string."""
self.assertFalse(utils.is_attribute_injection('some_string'))
def test_with_object(self):
"""Test with object."""
self.assertFalse(utils.is_attribute_injection(object()))
class IsCatalogTests(unittest.TestCase): class IsCatalogTests(unittest.TestCase):
"""`is_catalog()` test cases.""" """`is_catalog()` test cases."""