From f3f62232a44bbbb3eb2d52af373d7a1456f23bd8 Mon Sep 17 00:00:00 2001
From: L1ghtn1ng <jay@cybermon.uk>
Date: Wed, 30 Jun 2021 01:15:30 +0100
Subject: [PATCH] Add bufferover unit test

---
 tests/discovery/test_bufferover.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 tests/discovery/test_bufferover.py

diff --git a/tests/discovery/test_bufferover.py b/tests/discovery/test_bufferover.py
new file mode 100644
index 00000000..f251c12f
--- /dev/null
+++ b/tests/discovery/test_bufferover.py
@@ -0,0 +1,30 @@
+#!/usr/bin/env python3
+# coding=utf-8
+from theHarvester.lib.core import *
+from theHarvester.discovery import bufferoverun
+import requests
+import pytest
+
+pytestmark = pytest.mark.asyncio
+
+
+class TestBufferover(object):
+    @staticmethod
+    def domain() -> str:
+        return 'uber.com'
+
+    async def test_api(self):
+        base_url = f'https://dns.bufferover.run/dns?q={TestBufferover.domain()}'
+        headers = {'User-Agent': Core.get_user_agent()}
+        request = requests.get(base_url, headers=headers)
+        assert request.status_code == 200
+
+    async def test_do_search(self):
+        search = bufferoverun.SearchBufferover(TestBufferover.domain())
+        await search.process()
+        assert isinstance(await search.get_hostnames(), set)
+        assert isinstance(await search.get_ips(), set)
+
+
+if __name__ == '__main__':
+    pytest.main()