From 6de05139bfb6dfbd3532bbe85db95810bbfee982 Mon Sep 17 00:00:00 2001 From: Jay Townsend Date: Sun, 22 Sep 2019 17:03:53 +0100 Subject: [PATCH] Add otx tests Signed-off-by: Jay Townsend --- tests/discovery/test_otx.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 tests/discovery/test_otx.py diff --git a/tests/discovery/test_otx.py b/tests/discovery/test_otx.py new file mode 100644 index 00000000..67d54d10 --- /dev/null +++ b/tests/discovery/test_otx.py @@ -0,0 +1,32 @@ +#!/usr/bin/env python3 +# coding=utf-8 +from theHarvester.lib.core import * +from theHarvester.discovery import otxsearch +import requests +import pytest + + +class TestOtx(object): + @staticmethod + def domain() -> str: + return 'metasploit.com' + + def test_api(self): + base_url = f'https://otx.alienvault.com/api/v1/indicators/domain/{TestOtx.domain()}/passive_dns' + headers = {'User-Agent': Core.get_user_agent()} + request = requests.get(base_url, headers=headers) + assert request.status_code == 200 + + def test_search(self): + search = otxsearch.SearchOtx(TestOtx.domain()) + search.process() + assert type(search.get_hostnames()) == list + + def test_search_no_results(self): + search = otxsearch.SearchOtx('radiant.eu') + search.process() + assert len(search.get_hostnames()) == 0 + + +if __name__ == '__main__': + pytest.main()