def test_blank_command_line():
"""Test the command line usage of this project."""
with pytest.raises(docopt.DocoptExit) as exc_info:
cli.main()
# get the error message
error_message = exc_info.value
# make sure the error message contains the expected usage output
assert "Usage:" in str(error_message)
def test_pdns_reverse_query(command_line_args):
"""Test the pdns reverse query."""
output = cli.main({'--pdns-reverse': "8.8.8.8"})
assert len(output) > 1
assert output[0]['rrdata'] == command_line_args['ip']
def test_pdns_forward_query(command_line_args):
"""Test the pdns forward query."""
output = cli.main({'--pdns-forward': "example.com"})
assert len(output) > 1
assert output[0]['rrname'] == command_line_args['host']
def test_as_query():
"""Test the asn query."""
output = cli.main({'--as': "15169"})
assert output['status'] == "ok"
def test_ip_query():
"""Test the command line usage of this project."""
output = cli.main({'--ip': "8.8.8.8"})
assert output['status'] == "ok"
assert output['city'] == "Mountain View"
