Mirrors/sherlock
2
0
Fork 0
mirror of https://github.com/sherlock-project/sherlock synced 2025-02-16 12:38:27 +00:00
Code Issues Projects Releases Packages Wiki Activity

Change tests so that all sites that share a common username are executed in parallel. Just the little bit of extra processing halves the time taken for the tests (~15s verus ~34s). This will only get worse as the test coverage improves...

Browse source
This commit is contained in:
Christopher K. Hoadley 2019-02-03 11:14:45 -06:00
parent 5eb06e3502
commit dea8c29351
1 changed files with 20 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
tests/base.py
View file

@ -128,6 +128,10 @@ class SherlockBaseTest(unittest.TestCase):
existence state.
"""
#Dictionary of sites that should be tested for having a username.
#This will allow us to test sites with a common username in parallel.
sites_by_username = {}
for site, site_data in self.site_data_all.items():
if (
(site_data["errorType"] != detect_type) or
@ -143,12 +147,22 @@ class SherlockBaseTest(unittest.TestCase):
# Figure out which type of user
if exist_check:
username_list = [site_data.get("username_claimed")]
username = site_data.get("username_claimed")
else:
username_list = [site_data.get("username_unclaimed")]
self.username_check(username_list,
[site],
exist_check=exist_check
)
username = site_data.get("username_unclaimed")
# Add this site to the list of sites corresponding to this
# username.
if username in sites_by_username:
sites_by_username[username].append(site)
else:
sites_by_username[username] = [site]
# Check on the username availability against all of the sites.
for username, site_list in sites_by_username.items():
self.username_check([username],
site_list,
exist_check=exist_check
)
return