• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• vanilla-master
  • .coveragerc
  • examples
    • ticktock.py
    • tcp-basic.py
    • tcp-echo.py
  • xx
    • react.py
    • pipe.py
    • router.py
    • notes.md
    • pipe2.py
    • stream.py
  • LICENSE
  • TODO.md
  • README.rst
  • setup.py
  • CHANGELOG.rst
  • .travis.yml
  • tests
    • integration
      • test_http.py
    • unit
      • test_poll.py
      • test_thread.py
      • test_process.py
      • test_udp.py
      • test_io.py
      • test_message.py
      • test_http.py
      • test_signal.py
      • test_tcp.py
      • test_core.py
  • requirements.txt
  • bench
    • heap.py
  • .gitignore
  • vanilla
    • signal.py
    • udp.py
    • http.py
    • tcp.py
    • poll.py
    • message.py
    • io.py
    • meta.py
    • __init__.py
    • exception.py
    • core.py
    • process.py
    • thread.py
  • docs
    • Makefile
    • api.rst
    • requirements.txt
    • index.rst
    • conf.py
    • _static
  • MANIFEST.in
  • tox.ini

import vanilla
import signal
import os

import pytest


class TestProcess(object):
    def test_basic(self):
        h = vanilla.Hub()

        child = h.process.execv(
            ['/usr/bin/env', 'grep', '--line-buffered', 'foo'])
        assert child.check_liveness()
        pytest.raises(vanilla.Timeout, child.done.recv, timeout=0)

        child.stdin.send('foo1\n')
        assert child.stdout.recv() == 'foo1\n'
        child.stdin.send('bar1\n')
        child.stdin.send('foo2\n')
        assert child.stdout.recv() == 'foo2\n'

        child.terminate()
        child.done.recv()
        assert not child.check_liveness()

        # spawn a new process to ensure the reaper watch is rebooted
        h.sleep(1)
        child = h.process.execv(
            ['/usr/bin/env', 'grep', '--line-buffered', 'foo'])
        assert child.check_liveness()
        pytest.raises(vanilla.Timeout, child.done.recv, timeout=0)

        child.stdin.send('foo1\n')
        assert child.stdout.recv() == 'foo1\n'
        child.stdin.send('bar1\n')
        child.stdin.send('foo2\n')
        assert child.stdout.recv() == 'foo2\n'

        child.terminate()
        child.done.recv()
        assert not child.check_liveness()

    def test_unexecutable(self):
        h = vanilla.Hub()
        child = h.process.execv(['foobar123'])
        child.done.recv()
        assert child.exitcode == 127
        child = h.process.execv(['/dev/null'])
        child.done.recv()
        assert child.exitcode == 126

    def test_stderr(self):
        h = vanilla.Hub()
        child = h.process.execv(['/usr/bin/env', 'grep', '-g'])
        assert child.stderr.recv()

    def test_stderrtoout(self):
        h = vanilla.Hub()
        child = h.process.execv(
            ['/usr/bin/env', 'grep', '-g'], stderrtoout=True)
        assert child.stdout.recv()

    def test_signal(self):
        h = vanilla.Hub()
        child = h.process.execv(
            ['/usr/bin/env', 'grep', '--line-buffered', 'foo'])
        child.signal(signal.SIGTERM)
        child.done.recv()
        assert not child.check_liveness()

    def test_env(self):
        h = vanilla.Hub()

        VAR1 = 'VANILLA_%s_VAR1' % os.getpid()
        VAR2 = 'VANILLA_%s_VAR2' % os.getpid()

        os.putenv(VAR1, 'VAR1')

        child = h.process.execv(
            ['/usr/bin/env', 'sh', '-c', 'echo $%s $%s' % (VAR1, VAR2)])
        assert child.stdout.recv() == 'VAR1\n'

        child = h.process.execv(
            ['/usr/bin/env', 'sh', '-c', 'echo $%s $%s' % (VAR1, VAR2)],
            env={VAR2: 'VAR2'})
        assert child.stdout.recv() == 'VAR2\n'

    def test_spawn(self):
        h = vanilla.Hub()

        def worker():
            import sys
            while True:
                line = sys.stdin.readline()
                sys.stdout.write('worker: %s' % line)

        child = h.process.spawn(worker)
        child.stdin.send('line1\n')
        assert child.stdout.recv_partition('\n') == 'worker: line1'
        child.stdin.send('line2\n')
        assert child.stdout.recv_partition('\n') == 'worker: line2'