theHarvester/tests/discovery/test_linkedin_links.py

#!/usr/bin/env python3
# coding=utf-8
from theHarvester.discovery import linkedinsearch
from theHarvester.discovery.constants import splitter
import os
import re
import pytest

pytestmark = pytest.mark.asyncio


class TestGetLinks(object):

    async def test_splitter(self):
        results = [
            'https://www.linkedin.com/in/don-draper-b1045618',
            'https://www.linkedin.com/in/don-draper-b59210a',
            'https://www.linkedin.com/in/don-draper-b5bb50b3',
            'https://www.linkedin.com/in/don-draper-b83ba26',
            'https://www.linkedin.com/in/don-draper-b854a51'
        ]
        filtered_results = await splitter(results)
        assert len(filtered_results) == 1

    async def test_get_links(self):
        search = linkedinsearch.SearchLinkedin("facebook.com", '100')
        await search.process()
        links = await search.get_links()
        assert isinstance(links, list)

    async def test_links_linkedin(self):
        dir_path = os.path.dirname(os.path.realpath(__file__))
        mock_response = open(dir_path + "/test_linkedin_links.txt")
        mock_response_content = mock_response.read()
        mock_response.close()
        reg_links = re.compile(r"url=https:\/\/www\.linkedin.com(.*?)&")
        temp = reg_links.findall(mock_response_content)
        resul = []
        for regex_item in temp:
            stripped_url = regex_item.replace("url=", "")
            resul.append("https://www.linkedin.com" + stripped_url)
        assert set(resul)


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