Make a few assertions about the entry point script rather than keeping a fully-rendered copy.

1 files changed, 7 insertions, 31 deletions

diff --git a/setuptools/tests/test_easy_install.py b/setuptools/tests/test_easy_install.py
index 345823cf..8611dc16 100644
--- a/setuptools/tests/test_easy_install.py
+++ b/setuptools/tests/test_easy_install.py
@@ -16,6 +16,7 @@ import io
 import zipfile
 import mock
 import time
+import re
 
 from setuptools.extern import six
 
@@ -71,40 +72,15 @@ class TestEasyInstallTest:
 
     def test_get_script_args(self):
         header = ei.CommandSpec.best().from_environment().as_header()
-        if sys.version_info >= (3, 8):
-            expected = header + DALS(r"""
-                # EASY-INSTALL-ENTRY-SCRIPT: 'spec','console_scripts','name'
-                import re
-                import sys
-                from importlib.metadata import distribution
-
-                if __name__ == '__main__':
-                    sys.argv[0] = re.sub(r'(-script\.pyw?|\.exe)?$', '', sys.argv[0])
-                    for entry_point in distribution('spec').entry_points:
-                        if entry_point.group == 'console_scripts' and entry_point.name == 'name':
-                            sys.exit(entry_point.load()())
-                """)  # noqa: E501
-        else:
-            expected = header + DALS(r"""
-                # EASY-INSTALL-ENTRY-SCRIPT: 'spec','console_scripts','name'
-                __requires__ = 'spec'
-                import re
-                import sys
-                from pkg_resources import load_entry_point
-
-                if __name__ == '__main__':
-                    sys.argv[0] = re.sub(r'(-script\.pyw?|\.exe)?$', '', sys.argv[0])
-                    sys.exit(
-                        load_entry_point('spec', 'console_scripts', 'name')()
-                    )
-                """)  # noqa: E501
-
         dist = FakeDist()
-
         args = next(ei.ScriptWriter.get_args(dist))
         name, script = itertools.islice(args, 2)
-
-        assert script == expected
+        assert script.startswith(header)
+        assert "'spec'" in script
+        assert "'console_scripts'" in script
+        assert "'name'" in script
+        assert re.search(
+            '^# EASY-INSTALL-ENTRY-SCRIPT', script, flags=re.MULTILINE)
 
     def test_no_find_links(self):
         # new option '--no-find-links', that blocks find-links added at