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

Use SitesInformation() object in tests. For now, use the new SitesInformation() object to calculate the original JSON dictionary: the rest of the code will be updated in the future.

Browse source
This commit is contained in:
Christopher K. Hoadley 2019-12-31 15:33:48 -06:00
parent 8f6938ecb1
commit f48a2980f5
1 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
sherlock/tests/base.py
View file

@ -9,6 +9,7 @@ import unittest
import sherlock import sherlock
from result import QueryStatus from result import QueryStatus
from result import QueryResult from result import QueryResult
from sites import SitesInformation
import warnings import warnings
@ -29,10 +30,16 @@ class SherlockBaseTest(unittest.TestCase):
#TODO: Figure out how to fix the code so this is not needed. #TODO: Figure out how to fix the code so this is not needed.
warnings.simplefilter("ignore", ResourceWarning) warnings.simplefilter("ignore", ResourceWarning)
# Load the data file with all site information. #Create object with all information about sites we are aware of.
data_file_path = os.path.join(os.path.dirname(os.path.realpath(sherlock.__file__)), "resources/data.json") sites = SitesInformation()
with open(data_file_path, "r", encoding="utf-8") as raw:
self.site_data_all = json.load(raw) #Create original dictionary from SitesInformation() object.
#Eventually, the rest of the code will be updated to use the new object
#directly, but this will glue the two pieces together.
site_data_all = {}
for site in sites:
site_data_all[site.name] = site.information
self.site_data_all = site_data_all
# Load excluded sites list, if any # Load excluded sites list, if any
excluded_sites_path = os.path.join(os.path.dirname(os.path.realpath(sherlock.__file__)), "tests/.excluded_sites") excluded_sites_path = os.path.join(os.path.dirname(os.path.realpath(sherlock.__file__)), "tests/.excluded_sites")
@ -116,7 +123,7 @@ class SherlockBaseTest(unittest.TestCase):
with self.subTest(f"Checking Username '{username}' " with self.subTest(f"Checking Username '{username}' "
f"{check_type_text} on Site '{site}'" f"{check_type_text} on Site '{site}'"
): ):
self.assertEqual(result['status'].status, self.assertEqual(result['status'].status,
exist_result_desired) exist_result_desired)
return return