Add test for old namespace packages protocol.

25 files changed, 37 insertions, 0 deletions

diff --git a/astroid/tests/testdata/python2/data/path_pkg_resources_1/package/__init__.py b/astroid/tests/testdata/python2/data/path_pkg_resources_1/package/__init__.py
new file mode 100644
index 00000000..b0d64337
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkg_resources_1/package/__init__.py
@@ -0,0 +1 @@
+__import__('pkg_resources').declare_namespace(__name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python2/data/path_pkg_resources_1/package/foo.py b/astroid/tests/testdata/python2/data/path_pkg_resources_1/package/foo.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkg_resources_1/package/foo.py
diff --git a/astroid/tests/testdata/python2/data/path_pkg_resources_2/package/__init__.py b/astroid/tests/testdata/python2/data/path_pkg_resources_2/package/__init__.py
new file mode 100644
index 00000000..b0d64337
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkg_resources_2/package/__init__.py
@@ -0,0 +1 @@
+__import__('pkg_resources').declare_namespace(__name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python2/data/path_pkg_resources_2/package/bar.py b/astroid/tests/testdata/python2/data/path_pkg_resources_2/package/bar.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkg_resources_2/package/bar.py
diff --git a/astroid/tests/testdata/python2/data/path_pkg_resources_3/package/__init__.py b/astroid/tests/testdata/python2/data/path_pkg_resources_3/package/__init__.py
new file mode 100644
index 00000000..b0d64337
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkg_resources_3/package/__init__.py
@@ -0,0 +1 @@
+__import__('pkg_resources').declare_namespace(__name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python2/data/path_pkg_resources_3/package/baz.py b/astroid/tests/testdata/python2/data/path_pkg_resources_3/package/baz.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkg_resources_3/package/baz.py
diff --git a/astroid/tests/testdata/python2/data/path_pkgutil_1/package/__init__.py b/astroid/tests/testdata/python2/data/path_pkgutil_1/package/__init__.py
new file mode 100644
index 00000000..0bfb5a62
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkgutil_1/package/__init__.py
@@ -0,0 +1,2 @@
+from pkgutil import extend_path
+__path__ = extend_path(__path__, __name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python2/data/path_pkgutil_1/package/foo.py b/astroid/tests/testdata/python2/data/path_pkgutil_1/package/foo.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkgutil_1/package/foo.py
diff --git a/astroid/tests/testdata/python2/data/path_pkgutil_2/package/__init__.py b/astroid/tests/testdata/python2/data/path_pkgutil_2/package/__init__.py
new file mode 100644
index 00000000..0bfb5a62
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkgutil_2/package/__init__.py
@@ -0,0 +1,2 @@
+from pkgutil import extend_path
+__path__ = extend_path(__path__, __name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python2/data/path_pkgutil_2/package/bar.py b/astroid/tests/testdata/python2/data/path_pkgutil_2/package/bar.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkgutil_2/package/bar.py
diff --git a/astroid/tests/testdata/python2/data/path_pkgutil_3/package/__init__.py b/astroid/tests/testdata/python2/data/path_pkgutil_3/package/__init__.py
new file mode 100644
index 00000000..0bfb5a62
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkgutil_3/package/__init__.py
@@ -0,0 +1,2 @@
+from pkgutil import extend_path
+__path__ = extend_path(__path__, __name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python2/data/path_pkgutil_3/package/baz.py b/astroid/tests/testdata/python2/data/path_pkgutil_3/package/baz.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python2/data/path_pkgutil_3/package/baz.py
diff --git a/astroid/tests/testdata/python3/data/path_pkg_resources_1/package/__init__.py b/astroid/tests/testdata/python3/data/path_pkg_resources_1/package/__init__.py
new file mode 100644
index 00000000..b0d64337
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkg_resources_1/package/__init__.py
@@ -0,0 +1 @@
+__import__('pkg_resources').declare_namespace(__name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python3/data/path_pkg_resources_1/package/foo.py b/astroid/tests/testdata/python3/data/path_pkg_resources_1/package/foo.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkg_resources_1/package/foo.py
diff --git a/astroid/tests/testdata/python3/data/path_pkg_resources_2/package/__init__.py b/astroid/tests/testdata/python3/data/path_pkg_resources_2/package/__init__.py
new file mode 100644
index 00000000..b0d64337
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkg_resources_2/package/__init__.py
@@ -0,0 +1 @@
+__import__('pkg_resources').declare_namespace(__name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python3/data/path_pkg_resources_2/package/bar.py b/astroid/tests/testdata/python3/data/path_pkg_resources_2/package/bar.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkg_resources_2/package/bar.py
diff --git a/astroid/tests/testdata/python3/data/path_pkg_resources_3/package/__init__.py b/astroid/tests/testdata/python3/data/path_pkg_resources_3/package/__init__.py
new file mode 100644
index 00000000..b0d64337
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkg_resources_3/package/__init__.py
@@ -0,0 +1 @@
+__import__('pkg_resources').declare_namespace(__name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python3/data/path_pkg_resources_3/package/baz.py b/astroid/tests/testdata/python3/data/path_pkg_resources_3/package/baz.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkg_resources_3/package/baz.py
diff --git a/astroid/tests/testdata/python3/data/path_pkgutil_1/package/__init__.py b/astroid/tests/testdata/python3/data/path_pkgutil_1/package/__init__.py
new file mode 100644
index 00000000..0bfb5a62
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkgutil_1/package/__init__.py
@@ -0,0 +1,2 @@
+from pkgutil import extend_path
+__path__ = extend_path(__path__, __name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python3/data/path_pkgutil_1/package/foo.py b/astroid/tests/testdata/python3/data/path_pkgutil_1/package/foo.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkgutil_1/package/foo.py
diff --git a/astroid/tests/testdata/python3/data/path_pkgutil_2/package/__init__.py b/astroid/tests/testdata/python3/data/path_pkgutil_2/package/__init__.py
new file mode 100644
index 00000000..0bfb5a62
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkgutil_2/package/__init__.py
@@ -0,0 +1,2 @@
+from pkgutil import extend_path
+__path__ = extend_path(__path__, __name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python3/data/path_pkgutil_2/package/bar.py b/astroid/tests/testdata/python3/data/path_pkgutil_2/package/bar.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkgutil_2/package/bar.py
diff --git a/astroid/tests/testdata/python3/data/path_pkgutil_3/package/__init__.py b/astroid/tests/testdata/python3/data/path_pkgutil_3/package/__init__.py
new file mode 100644
index 00000000..0bfb5a62
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkgutil_3/package/__init__.py
@@ -0,0 +1,2 @@
+from pkgutil import extend_path
+__path__ = extend_path(__path__, __name__)
\ No newline at end of file
diff --git a/astroid/tests/testdata/python3/data/path_pkgutil_3/package/baz.py b/astroid/tests/testdata/python3/data/path_pkgutil_3/package/baz.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/astroid/tests/testdata/python3/data/path_pkgutil_3/package/baz.py
diff --git a/astroid/tests/unittest_manager.py b/astroid/tests/unittest_manager.py
index 12674516..aa1fd400 100644
--- a/astroid/tests/unittest_manager.py
+++ b/astroid/tests/unittest_manager.py
@@ -22,6 +22,7 @@ import unittest
 
 import six
 
+import astroid
 from astroid import exceptions
 from astroid import manager
 from astroid.tests import resources
@@ -94,6 +95,24 @@ class AstroidManagerTest(resources.SysPathSetup,
                           self.manager.ast_from_module_name,
                           'unhandledModule')
 
+    def _test_ast_from_old_namespace_package_protocol(self, root):
+        origpath = sys.path[:]
+        paths = [resources.find('data/path_{}_{}'.format(root, index))
+                 for index in range(1, 4)]
+        sys.path.extend(paths)
+        try:
+            for name in ('foo', 'bar', 'baz'):
+                module = self.manager.ast_from_module_name('package.' + name)
+                self.assertIsInstance(module, astroid.Module)
+        finally:
+            sys.path = origpath
+
+    def test_ast_from_namespace_pkgutil(self):
+        self._test_ast_from_old_namespace_package_protocol('pkgutil')
+
+    def test_ast_from_namespace_pkg_resources(self):
+        self._test_ast_from_old_namespace_package_protocol('pkg_resources')
+
     def _test_ast_from_zip(self, archive):
         origpath = sys.path[:]
         sys.modules.pop('mypypa', None)