From 18324abb38343c2db441614e1b30f87bf0aa30c8 Mon Sep 17 00:00:00 2001
From: Rajat Gupta <35985127+rjt-gupta@users.noreply.github.com>
Date: Mon, 10 Jun 2019 13:01:35 +0530
Subject: [PATCH] web-server test (#331)
---
tanner/tests/test_web_server.py | 126 ++++++++++++++++++++++++++++++++
1 file changed, 126 insertions(+)
create mode 100644 tanner/tests/test_web_server.py
diff --git a/tanner/tests/test_web_server.py b/tanner/tests/test_web_server.py
new file mode 100644
index 00000000..0c4f2bda
--- /dev/null
+++ b/tanner/tests/test_web_server.py
@@ -0,0 +1,126 @@
+import asyncio
+
+from unittest import mock
+from tanner.api.api import Api
+from aiohttp.test_utils import AioHTTPTestCase, unittest_run_loop
+from tanner.utils.asyncmock import AsyncMock
+from tanner.web.server import TannerWebServer
+
+
+class TestWebServer(AioHTTPTestCase):
+ def setUp(self):
+ self.loop = asyncio.new_event_loop()
+ redis = mock.Mock()
+ redis.close = mock.Mock()
+
+ self.handler = TannerWebServer()
+ self.handler.redis_client = redis
+ self.handler.api = Api(self.handler.redis_client)
+
+ self.returned_content = None
+ self.expected_content = None
+
+ super(TestWebServer, self).setUp()
+
+ def get_app(self):
+ app = self.handler.create_app(loop=self.loop)
+ return app
+
+ @unittest_run_loop
+ async def test_handle_index(self):
+ response = await self.client.request('GET', '/')
+ self.returned_content = await response.text()
+
+ self.expected_content = '\n\n\tHome - Tanner Web\n\t'
+
+ self.assertIn(self.expected_content, self.returned_content)
+
+ @unittest_run_loop
+ async def test_handle_snares(self):
+ self.handler.api.return_snares = AsyncMock(return_value=['9a631aee-2b52-4108-9831-b495ac8afa80'])
+
+ response = await self.client.request('GET', '/snares')
+ self.returned_content = await response.text()
+
+ self.expected_content = '' \
+ '9a631aee-2b52-4108-9831-b495ac8afa80'
+ self.assertIn(self.expected_content, self.returned_content)
+
+ @unittest_run_loop
+ async def test_handle_snare(self):
+
+ response = await self.client.request('GET', '/snare/9a631aee-2b52-4108-9831-b495ac8afa80')
+
+ self.returned_content = await response.text()
+
+ self.expected_content = 'Snare(9a631aee-2b52-4108-9831-b495ac8afa80) - Tanner Web'
+ self.assertIn(self.expected_content, self.returned_content)
+
+ @unittest_run_loop
+ async def test_handle_snare_stats(self):
+
+ content = {
+ "attack_frequency": {
+ 'cmd_exec': 1, 'lfi': 2, 'rfi': 1, 'sqli': 0, 'xss': 1
+ }
+ }
+
+ self.handler.api.return_snare_stats = AsyncMock(return_value=content)
+
+ response = await self.client.request('GET', '/snare-stats/9a631aee-2b52-4108-9831-b495ac8afa80')
+ self.returned_content = await response.text()
+
+ self.expected_content = '\n | Attack Frequency | \n \n \n cmd_exec : 1
' \
+ '\n \n lfi : 2
\n \n rfi : 1
\n \n sqli : 0
\n ' \
+ ' \n xss : 1
\n \n | \n
\n'
+
+ self.assertIn(self.expected_content, self.returned_content)
+
+ @unittest_run_loop
+ async def test_handle_sessions(self):
+
+ async def mock_return_sessions(filters):
+ if filters['peer_ip'] == "127.0.0.1" and filters['start_time'] == 1497890400 and \
+ filters['user_agent'] == 'Mozilla/5.0':
+
+ return [{"sess_uuid": "f387d46eaeb1454cadf0713a4a55be49"},
+ {"sess_uuid": "e85ae767b0bb4b1f91b421b3a28082ef"}]
+
+ self.handler.api.return_sessions = mock_return_sessions
+
+ response = await self.client.request('GET',
+ '/8fa6aa98-4283-4085-bfb9-a1cd3a9e56e4/sessions/page/2?filters=peer_ip:127.0.0.1 start_time:1497890400 user_agent:Mozilla/5.0') # noqa
+ self.returned_content = await response.text()
+
+ self.expected_content = 'Session-uuid | \n IP | \n Owner | \n \n ' \
+ '\n\n
\n