コード例 #1
0
ファイル: scanner_test.py プロジェクト: nikoheikkila/pwnedapi
def test_export_as(tempfile):
    # set up a match response for each password to be scanned
    for word in PASSWORDS:
        password = Password(word)
        responses.add(
            responses.GET,
            url=Password.API_URL + password.hashed_password_prefix(),
            body="{}:1\r\n".format(password.hashed_password_suffix()),
            status=200,
        )
    scanner = Scanner()
    results = scanner.scan(tempfile).data

    # three matches found
    assert results.height == 3
    # password and count
    assert results.width == 2

    export_file = "test.json"

    scanner.export_as(export_file)
    data = json.load(open(export_file))

    for d in data:
        assert d[scanner.get_headers()[0]] in PASSWORDS
        assert d[scanner.get_headers()[1]] > 0

    os.remove(export_file)
コード例 #2
0
def test_export_as(tempfile):
    scanner = Scanner()
    results = scanner.scan(tempfile).data

    assert results.height == 3
    assert results.width == 2

    export_file = "test.json"

    scanner.export_as(export_file)
    data = json.load(open(export_file))

    for d in data:
        assert d[scanner.get_headers()[0]] in PASSWORDS
        assert d[scanner.get_headers()[1]] > 0

    os.remove(export_file)