import io
from testpath import assert_isfile

from nsist import commands, _rewrite_shebangs

cmds = {'acommand': {'entry_point': 'somemod:somefunc',
                     'extra_preamble': io.StringIO(u'import extra')}}

def test_prepare_bin_dir(tmpdir):
    commands.prepare_bin_directory(tmpdir, cmds)
    assert_isfile(str(tmpdir / 'acommand.exe'))
    script_file = tmpdir / 'acommand-script.py'
    assert_isfile(str(script_file))

    with script_file.open() as f:
        script_contents = f.read()
    assert script_contents.startswith("#!python")
    assert 'import extra' in script_contents
    assert 'somefunc()' in script_contents

    _rewrite_shebangs.main(['_rewrite_shebangs.py', str(tmpdir)])
    with script_file.open() as f:
        assert f.read().startswith('#!"')