#!/usr/bin/env python
# encoding: utf-8

import unittest
import envoy

class SimpleTest(unittest.TestCase):

    def testInput(self):
        r = envoy.run("sed s/i/I/g", "Hi")
        self.assertEqual(r.std_out, "HI")
        self.assertEqual(r.status_code, 0)

    def testPipe(self):
        r = envoy.run("echo -n 'hi'| tr [:lower:] [:upper:]")
        self.assertEqual(r.std_out, "HI")
        self.assertEqual(r.status_code, 0)

    def testTimeout(self):
        r = envoy.run("yes | head", timeout=1)
        self.assertEqual(r.std_out, "y\ny\ny\ny\ny\ny\ny\ny\ny\ny\n")
        self.assertEqual(r.status_code, 0)

    def test_quoted_args(self):
        sentinel = 'quoted_args' * 3
        r = envoy.run("python -c 'print \"%s\"'" % sentinel)
        self.assertEqual(r.std_out.rstrip(), sentinel)
        self.assertEqual(r.status_code, 0)

if __name__ == '__main__':
    unittest.main()



#!/usr/bin/env python
# encoding: utf-8

import unittest
import envoy
import time

class SimpleTest(unittest.TestCase):

    def testInput(self):
        r = envoy.run("sed s/i/I/g", "Hi")
        self.assertEqual(r.std_out, "HI")
        self.assertEqual(r.status_code, 0)

    def testPipe(self):
        r = envoy.run("echo -n 'hi'| tr [:lower:] [:upper:]")
        self.assertEqual(r.std_out, "HI")
        self.assertEqual(r.status_code, 0)

    def testTimeout(self):
        r = envoy.run("yes | head", timeout=1)
        self.assertEqual(r.std_out, "y\ny\ny\ny\ny\ny\ny\ny\ny\ny\n")
        self.assertEqual(r.status_code, 0)

    # THIS TEST FAILS BECAUSE expand_args DOESN'T HANDLE QUOTES PROPERLY
    def test_quoted_args(self):
        sentinel = 'quoted_args' * 3
        r = envoy.run("python -c 'print \"%s\"'" % sentinel)
        self.assertEqual(r.std_out.rstrip(), sentinel)
        self.assertEqual(r.status_code, 0)

class ConnectedCommandTests(unittest.TestCase):

    def test_status_code_none(self):
        c = envoy.connect("sleep 5")
        self.assertEqual(c.status_code, None)

    def test_status_code_success(self):
        c = envoy.connect("sleep 1")
        time.sleep(2)
        self.assertEqual(c.status_code, 0)

    def test_status_code_failure(self):
        c = envoy.connect("sleep 1")
        self.assertEqual(c.status_code, 127)

    def test_input(self):
        test_string = 'asdfQWER'
        r = envoy.connect("cat | tr [:lower:] [:upper:]")
        r.send(test_string)
        self.assertEqual(c.std_out, test_string.upper())
        self.assertEqual(c.status_code, 0)

if __name__ == '__main__':
    unittest.main()



Update test_envoy.py