From ee809a44e6576ed662f5c9704e2cbad4b7590698 Mon Sep 17 00:00:00 2001
From: Roman Mogylatov <rmogilatov@gmail.com>
Date: Tue, 20 Oct 2020 13:22:13 -0400
Subject: [PATCH] Add test

---
 tests/unit/wiring/package/__init__.py | 8 ++++++++
 tests/unit/wiring/test_wiring_py36.py | 5 +++++
 2 files changed, 13 insertions(+)

diff --git a/tests/unit/wiring/package/__init__.py b/tests/unit/wiring/package/__init__.py
index e69de29b..83d62e58 100644
--- a/tests/unit/wiring/package/__init__.py
+++ b/tests/unit/wiring/package/__init__.py
@@ -0,0 +1,8 @@
+from dependency_injector.wiring import Provide
+
+from ..container import Container
+from ..service import Service
+
+
+def test_package_function(service: Service = Provide[Container.service]):
+    return service
diff --git a/tests/unit/wiring/test_wiring_py36.py b/tests/unit/wiring/test_wiring_py36.py
index 862a8386..8f755391 100644
--- a/tests/unit/wiring/test_wiring_py36.py
+++ b/tests/unit/wiring/test_wiring_py36.py
@@ -21,6 +21,11 @@ class WiringTest(unittest.TestCase):
         self.addCleanup(self.container.unwire)
 
     def test_package_lookup(self):
+        from .package import test_package_function
+        service = test_package_function()
+        self.assertIsInstance(service, Service)
+
+    def test_package_submodule_lookup(self):
         from .package.subpackage.submodule import test_function
         service = test_function()
         self.assertIsInstance(service, Service)