[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/
_build/
# vscode
.vscode

View File

@ -22,6 +22,16 @@ def test_cli_lattice():
assert result.exit_code == 0
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():
with TemporaryDirectory() as tempdir:
@ -31,4 +41,12 @@ def test_cli_stream():
result = runner.invoke(cli, ['--format', 'csv', '--output', outfile,
'stream', infile])
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