[MRG + 1] Test UsageError for CLI (#122)

* add .vscode

* add tests for UsageError

* fix pep8
pull/2/head
Vaibhav Mule 2018-10-04 22:01:20 +05:30 committed by Vinayak Mehta
parent 9d00937ec7
commit 58eddd0804
2 changed files with 22 additions and 1 deletions

3
.gitignore vendored
View File

@ -11,3 +11,6 @@ coverage.xml
.pytest_cache/ .pytest_cache/
_build/ _build/
# vscode
.vscode

View File

@ -22,6 +22,16 @@ def test_cli_lattice():
assert result.exit_code == 0 assert result.exit_code == 0
assert result.output == 'Found 1 tables\n' assert result.output == 'Found 1 tables\n'
result = runner.invoke(cli, ['--format', 'csv',
'lattice', infile])
output_error = 'Error: Please specify output file path using --output'
assert output_error in result.output
result = runner.invoke(cli, ['--output', outfile,
'lattice', infile])
format_error = 'Please specify output file format using --format'
assert format_error in result.output
def test_cli_stream(): def test_cli_stream():
with TemporaryDirectory() as tempdir: with TemporaryDirectory() as tempdir:
@ -32,3 +42,11 @@ def test_cli_stream():
'stream', infile]) 'stream', infile])
assert result.exit_code == 0 assert result.exit_code == 0
assert result.output == 'Found 1 tables\n' assert result.output == 'Found 1 tables\n'
result = runner.invoke(cli, ['--format', 'csv', 'stream', infile])
output_error = 'Error: Please specify output file path using --output'
assert output_error in result.output
result = runner.invoke(cli, ['--output', outfile, 'stream', infile])
format_error = 'Please specify output file format using --format'
assert format_error in result.output