forked from adysec/ARL
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_findvhost.py
29 lines (22 loc) · 985 Bytes
/
test_findvhost.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import unittest
from app.services.findVhost import find_vhost, brute_vhost
class TestFindVhost(unittest.TestCase):
def test_brute_vhost(self):
ip = "43.254.45.30"
domains = ["www.vulbox.com", "account.vulbox.com", "ysrc.vulbox.com"]
results = brute_vhost(ip, (domains, 'https'))
self.assertTrue(len(results) == 2)
results = brute_vhost(ip, (domains, 'http'))
self.assertTrue(len(results) == 0)
def test_find_vhost(self):
ips = ["43.254.45.30", "1.1.1.1"]
domains = ["www.vulbox.com", "account.vulbox.com", "ysrc.vulbox.com"]
results = find_vhost(ips, domains)
self.assertTrue(len(results) == 2)
def test_vhost_uniq(self):
ips = ["59.63.235.44", "1.1.1.1", "180.96.32.95", "114.106.160.20"]
domains = ["www.tencent.com", "m.tencent.com"]
results = find_vhost(ips, domains)
self.assertTrue(len(results) == 1)
if __name__ == '__main__':
unittest.main()