def test_egginst_script_untouched(self):
"""
Ensure we don't touch a script which has already been written by
egginst.
"""
simple_script = """\
#!/home/davidc/src/enthought/enstaller/.env/bin/python
# This script was created by egginst when installing:
#
# enstaller-4.6.3.dev1-py2.7.egg
#
if __name__ == '__main__':
import sys
from enstaller.patch import main
sys.exit(main())
"""
with mkdtemp() as d:
path = op.join(d, "script")
with open(path, "wt") as fp:
fp.write(simple_script)
fix_script(path)
with open(path, "rt") as fp:
self.assertEqual(fp.read(), simple_script)
def test_egginst_script_untouched(self):
"""
Ensure we don't touch a script which has already been written by
egginst.
"""
simple_script = """\
#!/home/davidc/src/enthought/enstaller/.env/bin/python
# This script was created by egginst when installing:
#
# enstaller-4.6.3.dev1-py2.7.egg
#
if __name__ == '__main__':
import sys
from enstaller.patch import main
sys.exit(main())
"""
with mkdtemp() as d:
path = os.path.join(d, "script")
with open(path, "wt") as fp:
fp.write(simple_script)
fix_script(path)
with open(path, "rt") as fp:
self.assertEqual(fp.read(), simple_script)
def test_setuptools_script_fixed(self):
"""
Ensure a script generated by setuptools is fixed.
"""
setuptools_script = """\
#!/dummy_path/.env/bin/python
# EASY-INSTALL-ENTRY-SCRIPT: 'enstaller==4.6.3.dev1','console_scripts','enpkg'
__requires__ = 'enstaller==4.6.3.dev1'
import sys
from pkg_resources import load_entry_point
if __name__ == '__main__':
sys.exit(
load_entry_point('enstaller==4.6.3.dev1', 'console_scripts', 'enpkg')()
)
"""
if sys.platform == "win32":
quoted_executable = '"' + sys.executable + '"'
else:
quoted_executable = sys.executable
r_egginst_script = """\
#!{executable}
# EASY-INSTALL-ENTRY-SCRIPT: 'enstaller==4.6.3.dev1','console_scripts','enpkg'
__requires__ = 'enstaller==4.6.3.dev1'
import sys
from pkg_resources import load_entry_point
if __name__ == '__main__':
sys.exit(
load_entry_point('enstaller==4.6.3.dev1', 'console_scripts', 'enpkg')()
)
""".format(executable=quoted_executable)
with mkdtemp() as d:
path = os.path.join(d, "script")
with open(path, "wt") as fp:
fp.write(setuptools_script)
fix_script(path, sys.executable)
with open(path, "rt") as fp:
self.assertMultiLineEqual(fp.read(), r_egginst_script)
def test_setuptools_script_fixed(self):
"""
Ensure a script generated by setuptools is fixed.
"""
setuptools_script = """\
#!/dummy_path/.env/bin/python
# EASY-INSTALL-ENTRY-SCRIPT: 'enstaller==4.6.3.dev1','console_scripts','enpkg'
__requires__ = 'enstaller==4.6.3.dev1'
import sys
from pkg_resources import load_entry_point
if __name__ == '__main__':
sys.exit(
load_entry_point('enstaller==4.6.3.dev1', 'console_scripts', 'enpkg')()
)
"""
if sys.platform == "win32":
executable = '"' + sys.executable + '"'
else:
executable = sys.executable
r_egginst_script = """\
#!{executable}
# EASY-INSTALL-ENTRY-SCRIPT: 'enstaller==4.6.3.dev1','console_scripts','enpkg'
__requires__ = 'enstaller==4.6.3.dev1'
import sys
from pkg_resources import load_entry_point
if __name__ == '__main__':
sys.exit(
load_entry_point('enstaller==4.6.3.dev1', 'console_scripts', 'enpkg')()
)
""".format(executable=executable)
with mkdtemp() as d:
path = os.path.join(d, "script")
with open(path, "wt") as fp:
fp.write(setuptools_script)
fix_script(path)
with open(path, "rt") as fp:
self.assertMultiLineEqual(fp.read(), r_egginst_script)
