Merge pull request docker-library#17446 from infosiftr/test-setuptools-wheel

Test that setuptools and wheel are installed (or not)

Author: yosifkit

test/tests/python-imports/container.py

@@ -1,8 +1,8 @@
-import platform
+import platform, sys
 
 isWindows = platform.system() == 'Windows'
 isNotPypy = platform.python_implementation() != 'PyPy'
-isCaveman = platform.python_version_tuple()[0] == '2'
+isCaveman = sys.version_info[0] == 2
 
 if not isWindows:
     import curses
@@ -23,3 +23,11 @@
 if not isCaveman:
     import lzma
     assert(lzma.decompress(lzma.compress(b'IT WORKS IT WORKS IT WORKS')) == b'IT WORKS IT WORKS IT WORKS')
+
+    # https://github.com/docker-library/python/pull/954
+    shouldHaveSetuptoolsAndWheel = sys.version_info[0] == 3 and sys.version_info[1] < 12
+    import importlib.util
+    hasSetuptools = importlib.util.find_spec('setuptools') is not None
+    hasWheel = importlib.util.find_spec('wheel') is not None
+    assert(hasSetuptools == shouldHaveSetuptoolsAndWheel)
+    assert(hasWheel == shouldHaveSetuptoolsAndWheel)