def test_all_checks_by_name(): """Run all checks and get CDN name.""" dom_in = Domain("login.gov", list(), list(), list(), list(), list(), list(), list()) check = cdnCheck() check.all_checks(dom_in, timeout=TIMEOUT, agent=USER_AGENT) assert (".cloudfront.net" in dom_in.cdns), "the ip for dns.google.com should be 8.8.8.8"
def test_cname(): """Test the CNAME resolving feature.""" dom_in = Domain("www.asu.edu", list(), list(), list(), list(), list(), list(), list()) check = cdnCheck() check.cname(dom_in, timeout=TIMEOUT) assert ( "www.asu.edu.cdn.cloudflare.net." in dom_in.cnames ), "www.asu.edu should have www.asu.edu.cdn.cloudflare.net. as a cname"
def test_ip(): """Test the IP resolving feature.""" dns.resolver.default_resolver = dns.resolver.Resolver() dns.resolver.default_resolver.nameservers = ["1.1.1.1", "8.8.8.8"] dom_in = Domain("dns.google.com", list(), list(), list(), list(), list(), list(), list()) check = cdnCheck() check.ip(dom_in) assert "8.8.8.8" in dom_in.ip, "the ip for dns.google.com should be 8.8.8.8"
def test_whois(): """Test the whois resolving feature.""" dom_in = Domain("google.com", list(), list(), list(), list(), list(), list(), list()) check = cdnCheck() check.ip(dom_in) check.whois(dom_in, interactive=False, verbose=False) assert ("GOOGLE" in dom_in.whois_data ), "google.com should return GOOGLE in the whois_data"
def test_https_lookup(): """Test the header resolving feature.""" dom_in = Domain("google.com", list(), list(), list(), list(), list(), list(), list()) check = cdnCheck() check.https_lookup(dom_in, timeout=TIMEOUT, agent=USER_AGENT, interactive=False, verbose=False) assert "gws" in dom_in.headers, "google.com should have gws as a header"
def test_broken_cname(): """Test a non-working domain CNAME resolving feature.""" dom_in = Domain( "notarealdomain.fakedomaindne.com", list(), list(), list(), list(), list(), list(), list(), ) check = cdnCheck() return_code = check.cname(dom_in, timeout=TIMEOUT) assert return_code != 0, "This fake site should return a non 0 code."
def test_broken_whois(): """Test a non-working domain whois resolving feature.""" dom_in = Domain( "notarealdomain.fakedomaindne.com", list(), list(), list(), list(), list(), list(), list(), ) check = cdnCheck() check.ip(dom_in) return_code = check.whois(dom_in, interactive=False, verbose=False) assert return_code != 0, "This fake site should return a non 0 code."
def test_broken_ip(): """Test a non-working domain IP resolving feature.""" dns.resolver.default_resolver = dns.resolver.Resolver() dns.resolver.default_resolver.nameservers = ["1.1.1.1", "8.8.8.8"] dom_in = Domain( "notarealdomain.fakedomaindne.com", list(), list(), list(), list(), list(), list(), list(), ) check = cdnCheck() return_code = check.ip(dom_in) assert return_code != 0, "This fake site should return a non 0 code."
def test_broken_https_lookup(): """Test a non-working domain header resolving feature.""" dom_in = Domain( "notarealdomain.fakedomaindne.com", list(), list(), list(), list(), list(), list(), list(), ) check = cdnCheck() check.https_lookup(dom_in, timeout=TIMEOUT, agent=USER_AGENT, interactive=False, verbose=False) assert len(dom_in.headers) <= 0, "There should be no response."
def test_all_checks_bad(): """Test fake domain and ensure it dosen't break anything.""" dns.resolver.default_resolver = dns.resolver.Resolver() dns.resolver.default_resolver.nameservers = ["1.1.1.1", "8.8.8.8"] dom = Domain( "super.definitelynot.notarealdomain.fakedomaindne.com", list(), list(), list(), list(), list(), list(), list(), ) print(dom.url, dom.cdns, dom.cnames, dom.headers, dom.whois_data, dom.ip) check = cdnCheck() return_code = check.all_checks(dom, timeout=TIMEOUT, agent=USER_AGENT) print(return_code) print(dom.url, dom.cdns, dom.cnames, dom.headers, dom.whois_data, dom.ip) assert return_code != 0, "This fake site should return a non 0 code."