mirror of https://github.com/ConsenSys/mythril
and start a unit test for options processing.pull/414/head
parent
a5a0e716fb
commit
4974a886a5
@ -0,0 +1,25 @@ |
||||
from mythril.interfaces.cli import main |
||||
import pytest |
||||
import json |
||||
|
||||
import sys |
||||
|
||||
def test_version_opt(capsys): |
||||
# Check that "myth --version" returns a string with the word |
||||
# "version" in it |
||||
sys.argv = ['mythril', '--version'] |
||||
with pytest.raises(SystemExit) as pytest_wrapped_e: |
||||
main() |
||||
assert pytest_wrapped_e.type == SystemExit |
||||
captured = capsys.readouterr() |
||||
assert captured.out.find(' version ') >= 1 |
||||
|
||||
# Check that "myth --version -o json" returns a JSON object |
||||
sys.argv = ['mythril', '--version', '-o', 'json'] |
||||
with pytest.raises(SystemExit) as pytest_wrapped_e: |
||||
main() |
||||
assert pytest_wrapped_e.type == SystemExit |
||||
captured = capsys.readouterr() |
||||
d = json.loads(captured.out) |
||||
assert isinstance(d, dict) |
||||
assert d['version_str'] |
Loading…
Reference in new issue