theHarvester/tests/test_myparser.py

#!/usr/bin/env python3
# coding=utf-8

import pytest

from theHarvester.parsers import myparser


class TestMyParser(object):
    @pytest.mark.asyncio
    @pytest.mark.parametrize(
        ('word', 'expected'),
        [
            ('Example.COM.', {'admin@example.com', 'lead@sub.example.com', 'owner@www.example.com'}),
            ('WWW.Example.COM.', {'owner@www.example.com'}),
        ],
    )
    async def test_emails_respect_target_label_boundaries(self, word: str, expected: set[str]) -> None:
        results = 'Admin@Example.COM***admin@notexample.com***.Lead@Sub.Example.COM.***owner@WWW.Example.COM***other@outside.test'
        parse = myparser.Parser(results, word)
        assert await parse.emails() == expected

    @pytest.mark.asyncio
    @pytest.mark.parametrize(
        ('word', 'expected'),
        [
            ('Example.COM.', {'api.example.com', 'example.com', 'sub.example.com', 'www.example.com'}),
            ('WWW.Example.COM.', {'www.example.com'}),
        ],
    )
    async def test_hostnames_respect_target_label_boundaries(self, word: str, expected: set[str]) -> None:
        results = 'Example.COM. API.Example.COM. badexample.com outside.test sub.example.com WWW.Example.COM.'
        parse = myparser.Parser(results, word)
        assert set(await parse.hostnames()) == expected

    @pytest.mark.asyncio
    async def test_hostnames_remove_uppercase_encoded_slash(self) -> None:
        parse = myparser.Parser('%2Fencrypted.google.com', 'google.com')

        assert await parse.hostnames() == ['encrypted.google.com']

    @pytest.mark.asyncio
    async def test_empty_target_fails_closed(self) -> None:
        parse = myparser.Parser('admin@example.com api.example.com', '')

        assert await parse.emails() == set()
        assert await parse.hostnames() == []


if __name__ == "__main__":
    pytest.main()