maltego_test.py 文件源码

def test_valid_calls(endpoint, details):
    """Test all endpoints with valid calls."""
    full_url = MALTEGO_SERVER + endpoint
    logging.debug("Requesting %s" % full_url)
    response = requests.post(
        full_url,
        data=build_maltego_response(details),
        auth=(API_USERNAME, API_KEY),
        verify=False
    )
    logging.debug("Response %s" % response.content)
    assert response.status_code == 200
    xmldoc = minidom.parseString(response.content)
    messages = xmldoc.getElementsByTagName('UIMessage')
    for message in messages:
        value = message.attributes['MessageType'].value
        assert value != 'FatalError'