Updated cli_test.py to validate against the updated help command conditions

Signed-off-by: Tony Witherspoon <tony.witherspoon@gmail.com>
This commit is contained in:
Tony Witherspoon 2016-04-12 11:47:15 -04:00
parent f5b80640fe
commit 276738f733
1 changed files with 7 additions and 0 deletions

View File

@ -5,6 +5,7 @@ from __future__ import unicode_literals
import os
import shutil
import tempfile
from StringIO import StringIO
import docker
import py
@ -82,6 +83,12 @@ class CLITestCase(unittest.TestCase):
self.assertTrue(project.client)
self.assertTrue(project.services)
def test_command_help(self):
with mock.patch('sys.stdout', new=StringIO()) as fake_stdout:
TopLevelCommand.help({'COMMAND': 'up'})
assert "Usage: up" in fake_stdout.getvalue()
def test_command_help_nonexistent(self):
with pytest.raises(NoSuchCommand):
TopLevelCommand.help({'COMMAND': 'nonexistent'})