Merge pull request #18 from jirikuncar/autodiscovery-app-fix

Fix issue with app in ModuleAutoDiscoveryRegistry

Author: lnielsen

flask_registry/registries/modulediscovery.py

@@ -248,5 +248,7 @@ def __init__(self, module_name, app=None, registry_namespace=None,
             module_name, with_setup=with_setup, silent=silent,
             registry_namespace=registry_namespace
         )
+        if app is None and has_app_context():
+            app = current_app
         self.app = app
         self.discover(app=app)

tests/test_modulediscovery.py

@@ -217,3 +217,22 @@ def test_registration(self):
         assert len(self.app.extensions['registry']['myns']) == 1
         from flask_registry.registries import appdiscovery
         assert self.app.extensions['registry']['myns'][0] == appdiscovery
+
+    def test_registry_proxy(self):
+        Registry(app=self.app)
+
+        self.app.extensions['registry']['pathns'] = \
+            ImportPathRegistry(initial=['flask_registry.*'])
+
+        myns = RegistryProxy(
+            'myns', ModuleAutoDiscoveryRegistry, 'appdiscovery',
+            registry_namespace='pathns'
+        )
+
+        with self.app.app_context():
+            assert len(self.app.extensions['registry']['pathns']) == 2
+            assert len(list(myns)) == 1
+            from flask_registry.registries import appdiscovery
+            assert myns[0] == appdiscovery
+
+