test_ok2/bin-for-dist/test_install.py

import py
import subprocess
import os, sys

execnet = py.test.importorskip("execnet")


#
# experimental funcargs for venv/install-tests
#

pytest_plugins = 'pytest_pytester',

def pytest_funcarg__venv(request):
    p = request.config.mktemp(request.function.__name__, numbered=True)
    venv = VirtualEnv(str(p)) 
    return venv 
   
def pytest_funcarg__py_setup(request):
    testdir = request.getfuncargvalue('testdir')
    rootdir = py.path.local(py.__file__).dirpath().dirpath()
    setup = rootdir.join('setup.py')
    if not setup.check():
        py.test.skip("not found: %r" % setup)
    return SetupBuilder(setup, testdir.tmpdir)

class SetupBuilder:
    def __init__(self, setup_path, tmpdir):
        self.setup_path = setup_path
        self.tmpdir = tmpdir
        assert setup_path.check()

    def make_sdist(self, destdir=None):
        temp = self.tmpdir.mkdir('dist')
        args = ['python', 'setup.py', 'sdist', '--dist-dir', str(temp)]
        old = self.setup_path.dirpath().chdir()
        try:
            subcall(args)
        finally:
            old.chdir()
        l = temp.listdir('py-*')
        assert len(l) == 1
        sdist = l[0]
        if destdir is None:
            destdir = self.setup_path.dirpath('build')
            assert destdir.check()
        else:
            destdir = py.path.local(destdir)
        target = destdir.join(sdist.basename)
        sdist.copy(target)
        return target 

def subcall(args):
    if hasattr(subprocess, 'check_call'):
        subprocess.check_call(args)
    else:
        subprocess.call(args)
# code taken from Ronny Pfannenschmidt's virtualenvmanager 

class VirtualEnv(object):
    def __init__(self, path):
        #XXX: supply the python executable
        self.path = path

    def __repr__(self):
        return "<VirtualEnv at %r>" %(self.path)

    def _cmd(self, name):
        if sys.platform == "win32":
            return os.path.join(self.path, 'Scripts', name)
        else:
            return os.path.join(self.path, 'bin', name)

    def ensure(self):
        if not os.path.exists(self._cmd('python')):
            self.create()

    def create(self, sitepackages=False):
        args = ['virtualenv', self.path]
        if not sitepackages:
            args.append('--no-site-packages')
        subcall(args)

    def makegateway(self):
        python = self._cmd('python')
        return execnet.makegateway("popen//python=%s" %(python,))

    def pcall(self, cmd, *args, **kw):
        self.ensure()
        return subprocess.call([
                self._cmd(cmd)
            ] + list(args),
            **kw)

    def pytest_getouterr(self, *args):
        self.ensure()
        args = [self._cmd("py.test")] + list(args)
        popen = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
        out, err = popen.communicate()
        return out

    def setup_develop(self):
        self.ensure()
        return self.pcall("python", "setup.py", "develop")

    def easy_install(self, *packages, **kw):
        args = []
        if 'index' in kw:
            index = kw['index']
            if isinstance(index, (list, tuple)):
                for i in index:
                    args.extend(['-i', i])
            else:
                args.extend(['-i', index])

        args.extend(packages)
        self.pcall('easy_install', *args)


def test_make_sdist_and_run_it(py_setup, venv):
    sdist = py_setup.make_sdist(venv.path)
    venv.easy_install(str(sdist)) 
    gw = venv.makegateway()
    ch = gw.remote_exec("import py ; channel.send(py.__version__)")
    version = ch.receive()
    assert version == py.__version__

def test_plugin_setuptools_entry_point_integration(py_setup, venv, tmpdir):
    sdist = py_setup.make_sdist(venv.path)
    venv.easy_install(str(sdist)) 
    # create a sample plugin
    basedir = tmpdir.mkdir("testplugin")
    basedir.join("setup.py").write("""if 1:
        from setuptools import setup
        setup(name="testplugin",
            entry_points = {'pytest11': ['testplugin=tp1']},
            py_modules = ['tp1'],
        )
    """)
    basedir.join("tp1.py").write(py.code.Source("""
        def pytest_addoption(parser):
            parser.addoption("--testpluginopt", action="store_true")
    """))
    basedir.chdir()
    print ("created sample plugin in %s" %basedir)
    venv.setup_develop()
    out = venv.pytest_getouterr("-h")
    assert "testpluginopt" in out

def test_cmdline_entrypoints(monkeypatch):
    monkeypatch.syspath_prepend(py.path.local(__file__).dirpath().dirpath())
    from setup import cmdline_entrypoints
    versioned_scripts = ['py.test', 'py.which']
    unversioned_scripts = versioned_scripts + [ 'py.cleanup', 
        'py.convert_unittest', 'py.countloc', 'py.lookup', 'py.svnwcrevert']
    for ver in [(2,4,0), (2,5,0), (2,6,0), (2,7,0), (3,0,1), (3,1,1)]:
        for platform in ('posix', 'win32'):
            points = cmdline_entrypoints(ver, "posix", 'python')
            for script in versioned_scripts:
                script_ver = script + "-%s.%s" % ver[:2]
                assert script_ver in points
            for script in unversioned_scripts:
                assert script in points
    points = cmdline_entrypoints((2,5,1), "java1.6.123", 'jython')
    for script in versioned_scripts:
        expected = "%s-jython" % script
        assert expected in points
    for script in unversioned_scripts:
        assert script not in points

    points = cmdline_entrypoints((2,5,1), "xyz", 'pypy-c-XYZ')
    for script in versioned_scripts:
        expected = "%s-pypy-c-XYZ" % script
        assert expected in points
    for script in unversioned_scripts:
        assert script in points

def test_slave_popen_needs_no_pylib(testdir, venv, pytestconfig):
    pytestconfig.pluginmanager.skipifmissing("xdist")
    venv.ensure()
    #xxx execnet optimizes popen
    #ch = venv.makegateway().remote_exec("import execnet")
    #py.test.raises(ch.RemoteError, ch.waitclose)
    python = venv._cmd("python")
    p = testdir.makepyfile("""
        import py
        def test_func():
            pass
     """)
    result = testdir.runpytest(p, '--rsyncdir=%s' % str(p), 
            '--dist=each', '--tx=popen//python=%s' % python)
    result.stdout.fnmatch_lines([
        "*1 passed*"
    ])

def test_slave_needs_no_execnet(testdir, sshhost, pytestconfig):
    pytestconfig.pluginmanager.skipifmissing("xdist")
    xspec = "ssh=%s" % sshhost
    gw = execnet.makegateway("ssh=%s" % sshhost)
    ch = gw.remote_exec("""
        import os, subprocess
        subprocess.call(["virtualenv", "--no-site-packages", "subdir"])
        channel.send(os.path.join(os.path.abspath("subdir"), 'bin', 'python'))
        channel.send(os.path.join(os.path.abspath("subdir")))
    """)
    try:
        path = ch.receive()
        chdir = ch.receive()
    except ch.RemoteError:
        e = sys.exc_info()[1]
        py.test.skip("could not prepare ssh slave:%s" % str(e))
    gw.exit()
    newspec = "%s//python=%s//chdir=%s" % (xspec, path, chdir)
    gw = execnet.makegateway(newspec)
    ch = gw.remote_exec("import execnet")
    py.test.raises(ch.RemoteError, ch.waitclose)
    gw.exit()
    
    p = testdir.makepyfile("""
        import py
        def test_func():
            pass
     """)
    result = testdir.runpytest(p, '--rsyncdir=%s' % str(p), 
            '--dist=each', '--tx=%s' % newspec)
    result.stdout.fnmatch_lines([
        "*1 passed*"
    ])