ets-labs/python-dependency-injector
1
1
Fork 0
mirror of https://github.com/ets-labs/python-dependency-injector.git synced 2024-11-22 01:26:51 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add tests for @providers.override decorator

Browse Source
This commit is contained in:
Roman Mogilatov 2016-03-01 16:28:37 +02:00
parent 86ef17c75a
commit 37fde8e97c
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_providers.py
View File

@ -1113,3 +1113,18 @@ class FactoryAsDecoratorTests(unittest.TestCase):
self.assertIsInstance(users_service, UsersService.cls)
self.assertIsInstance(users_service.auth_service, AuthService.cls)
def test_decoration_and_overriding(self):
"""Test decoration of some class with Factory provider."""
@providers.Factory
class AuthService(object):
"""Auth service."""
@providers.override(AuthService)
@providers.Factory
class ExtAuthService(AuthService.cls):
"""Extended auth service."""
auth_service = AuthService()
self.assertIsInstance(auth_service, ExtAuthService.cls)