bumblebee-status/tests/util/test_cli.py

import unittest

import util.cli

class cli(unittest.TestCase):
    def setUp(self):
        self.nonExistentCommand = 'i-do-not-exist'
        self.validCommand = 'echo test'
        self.validCommandOutput = 'test\n'
        self.utfCommand = 'echo ÖPmŧß'
        self.utfCommandOutput = 'ÖPmŧß\n'

    def test_valid_command(self):
        rv = util.cli.execute(self.validCommand)
        self.assertEqual(self.validCommandOutput, rv)

    def test_utf_command(self):
        rv = util.cli.execute(self.utfCommand)
        self.assertEqual(self.utfCommandOutput, rv)

    def test_invalid_command(self):
        with self.assertRaises(RuntimeError):
            util.cli.execute(self.nonExistentCommand)

    def test_command_exit_code(self):
        with self.assertRaises(RuntimeError):
            util.cli.execute('cat {}'.format(self.nonExistentCommand))

    def test_command_exit_code_no_error(self):
        try:
            util.cli.execute('cat {}'.format(self.nonExistentCommand), ignore_errors=True)
        except Exception:
            self.fail('exception was thrown')

    def test_async(self):
        rv = util.cli.execute(self.validCommand, wait=False)
        self.assertEqual('', rv)

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4