Skip to content

U
url-checker
Commit b8bd2d34 authored by Romain Courteaud's avatar Romain Courteaud
Browse files
Options

Lint

parent 072a67f9
Hide whitespace changes
Inline Side-by-side
Showing with 120 additions and 46 deletions
test_urlchecker_dns.py
View file @ b8bd2d34
......@@ -23,28 +23,34 @@ class UrlCheckerDNSTestCase(unittest.TestCase):
def test_logDnsQuery_insertFirst(self):
domain = "http://example.org"
resolver_ip = "127.0.0.1"
rdtype = 'foo'
answer_list = ['4.3.2.1', '1.2.3.4']
rdtype = "foo"
answer_list = ["4.3.2.1", "1.2.3.4"]
status_id = logStatus(self.db, "foo")
result = logDnsQuery(self.db, status_id, resolver_ip, domain, rdtype, answer_list)
result = logDnsQuery(
self.db, status_id, resolver_ip, domain, rdtype, answer_list
)
assert self.db.DnsChange.select().count() == 1
assert self.db.DnsChange.get().resolver_ip == resolver_ip
assert self.db.DnsChange.get().domain == domain
assert self.db.DnsChange.get().rdtype == rdtype
assert self.db.DnsChange.get().response == '1.2.3.4, 4.3.2.1'
assert self.db.DnsChange.get().response == "1.2.3.4, 4.3.2.1"
assert self.db.DnsChange.get().status_id == status_id
assert result == status_id
def test_logDnsQuery_insertOnlyOnePerStatusIdIPUrl(self):
domain = "http://example.org"
resolver_ip = "127.0.0.1"
rdtype = 'foo'
answer_list = ['4.3.2.1', '1.2.3.4']
rdtype = "foo"
answer_list = ["4.3.2.1", "1.2.3.4"]
status_id = logStatus(self.db, "foo")
result = logDnsQuery(self.db, status_id, resolver_ip, domain, rdtype, answer_list)
answer_list.extend('127.0.0.1')
result = logDnsQuery(
self.db, status_id, resolver_ip, domain, rdtype, answer_list
)
answer_list.extend("127.0.0.1")
try:
logDnsQuery(self.db, status_id, resolver_ip, domain, rdtype, answer_list)
logDnsQuery(
self.db, status_id, resolver_ip, domain, rdtype, answer_list
)
except peewee.IntegrityError:
assert self.db.DnsChange.select().count() == 1
assert self.db.DnsChange.get().status_id == status_id
......@@ -54,18 +60,22 @@ class UrlCheckerDNSTestCase(unittest.TestCase):
def test_logDnsQuery_skipIdenticalPreviousValues(self):
domain = "http://example.org"
resolver_ip = "127.0.0.1"
rdtype = 'foo'
answer_list = ['4.3.2.1', '1.2.3.4']
rdtype = "foo"
answer_list = ["4.3.2.1", "1.2.3.4"]
status_id = logStatus(self.db, "foo")
result = logDnsQuery(self.db, status_id, resolver_ip, domain, rdtype, answer_list)
result = logDnsQuery(
self.db, status_id, resolver_ip, domain, rdtype, answer_list
)
status_id_2 = logStatus(self.db, "foo")
result_2 = logDnsQuery(self.db, status_id_2, resolver_ip, domain, rdtype, answer_list)
result_2 = logDnsQuery(
self.db, status_id_2, resolver_ip, domain, rdtype, answer_list
)
assert self.db.DnsChange.select().count() == 1
assert self.db.DnsChange.get().resolver_ip == resolver_ip
assert self.db.DnsChange.get().domain == domain
assert self.db.DnsChange.get().rdtype == rdtype
assert self.db.DnsChange.get().response == '1.2.3.4, 4.3.2.1'
assert self.db.DnsChange.get().response == "1.2.3.4, 4.3.2.1"
assert self.db.DnsChange.get().status_id == status_id
assert result == status_id
assert result == result_2
......@@ -73,46 +83,112 @@ class UrlCheckerDNSTestCase(unittest.TestCase):
def test_logDnsQuery_insertWhenDifferentAnswer(self):
domain = "http://example.org"
resolver_ip = "127.0.0.1"
rdtype = 'foo'
answer_list = ['4.3.2.1', '1.2.3.4']
rdtype = "foo"
answer_list = ["4.3.2.1", "1.2.3.4"]
status_id = logStatus(self.db, "foo")
result = logDnsQuery(self.db, status_id, resolver_ip, domain, rdtype, answer_list)
answer_list_2 = ['4.3.2.1', '1.2.3.4', '0.0.0.0']
result = logDnsQuery(
self.db, status_id, resolver_ip, domain, rdtype, answer_list
)
answer_list_2 = ["4.3.2.1", "1.2.3.4", "0.0.0.0"]
status_id_2 = logStatus(self.db, "foo")
result_2 = logDnsQuery(self.db, status_id_2, resolver_ip, domain, rdtype, answer_list_2)
result_2 = logDnsQuery(
self.db, status_id_2, resolver_ip, domain, rdtype, answer_list_2
)
assert result_2 != result
assert self.db.DnsChange.select().count() == 2
assert self.db.DnsChange.get(self.db.DnsChange.status == status_id).resolver_ip == resolver_ip
assert self.db.DnsChange.get(self.db.DnsChange.status == status_id).domain == domain
assert self.db.DnsChange.get(self.db.DnsChange.status == status_id).rdtype == rdtype
assert self.db.DnsChange.get(self.db.DnsChange.status == status_id).response == '1.2.3.4, 4.3.2.1'
assert self.db.DnsChange.get(self.db.DnsChange.status == status_id).status_id == status_id
assert self.db.DnsChange.get(self.db.DnsChange.status == status_id_2).resolver_ip == resolver_ip
assert self.db.DnsChange.get(self.db.DnsChange.status == status_id_2).domain == domain
assert self.db.DnsChange.get(self.db.DnsChange.status == status_id_2).rdtype == rdtype
assert self.db.DnsChange.get(self.db.DnsChange.status == status_id_2).response == '0.0.0.0, 1.2.3.4, 4.3.2.1'
assert self.db.DnsChange.get(self.db.DnsChange.status == status_id_2).status_id == status_id_2
assert (
self.db.DnsChange.get(
self.db.DnsChange.status == status_id
).resolver_ip
== resolver_ip
)
assert (
self.db.DnsChange.get(self.db.DnsChange.status == status_id).domain
== domain
)
assert (
self.db.DnsChange.get(self.db.DnsChange.status == status_id).rdtype
== rdtype
)
assert (
self.db.DnsChange.get(
self.db.DnsChange.status == status_id
).response
== "1.2.3.4, 4.3.2.1"
)
assert (
self.db.DnsChange.get(
self.db.DnsChange.status == status_id
).status_id
== status_id
)
assert (
self.db.DnsChange.get(
self.db.DnsChange.status == status_id_2
).resolver_ip
== resolver_ip
)
assert (
self.db.DnsChange.get(
self.db.DnsChange.status == status_id_2
).domain
== domain
)
assert (
self.db.DnsChange.get(
self.db.DnsChange.status == status_id_2
).rdtype
== rdtype
)
assert (
self.db.DnsChange.get(
self.db.DnsChange.status == status_id_2
).response
== "0.0.0.0, 1.2.3.4, 4.3.2.1"
)
assert (
self.db.DnsChange.get(
self.db.DnsChange.status == status_id_2
).status_id
== status_id_2
)
def test_logDnsQuery_insertDifferentUrl(self):
domain = "http://example.org"
domain_2 = domain + '.'
domain_2 = domain + "."
resolver_ip = "127.0.0.1"
resolver_ip_2 = resolver_ip + '1'
rdtype = 'foo'
rdtype_2 = rdtype + 'bar'
answer_list = ['4.3.2.1', '1.2.3.4']
resolver_ip_2 = resolver_ip + "1"
rdtype = "foo"
rdtype_2 = rdtype + "bar"
answer_list = ["4.3.2.1", "1.2.3.4"]
status_id = logStatus(self.db, "foo")
result = logDnsQuery(self.db, status_id, resolver_ip, domain, rdtype, answer_list)
logDnsQuery(self.db, status_id, resolver_ip_2, domain, rdtype, answer_list)
logDnsQuery(self.db, status_id, resolver_ip, domain_2, rdtype, answer_list)
logDnsQuery(self.db, status_id, resolver_ip, domain, rdtype_2, answer_list)
logDnsQuery(self.db, status_id, resolver_ip_2, domain_2, rdtype, answer_list)
logDnsQuery(self.db, status_id, resolver_ip_2, domain, rdtype_2, answer_list)
logDnsQuery(self.db, status_id, resolver_ip, domain_2, rdtype_2, answer_list)
logDnsQuery(self.db, status_id, resolver_ip_2, domain_2, rdtype_2, answer_list)
result = logDnsQuery(
self.db, status_id, resolver_ip, domain, rdtype, answer_list
)
logDnsQuery(
self.db, status_id, resolver_ip_2, domain, rdtype, answer_list
)
logDnsQuery(
self.db, status_id, resolver_ip, domain_2, rdtype, answer_list
)
logDnsQuery(
self.db, status_id, resolver_ip, domain, rdtype_2, answer_list
)
logDnsQuery(
self.db, status_id, resolver_ip_2, domain_2, rdtype, answer_list
)
logDnsQuery(
self.db, status_id, resolver_ip_2, domain, rdtype_2, answer_list
)
logDnsQuery(
self.db, status_id, resolver_ip, domain_2, rdtype_2, answer_list
)
logDnsQuery(
self.db, status_id, resolver_ip_2, domain_2, rdtype_2, answer_list
)
assert self.db.DnsChange.select().count() == 8
......
urlchecker_dns.py
View file @ b8bd2d34
......@@ -82,9 +82,7 @@ def getResolverDict(db, status_id, resolver_ip_list):
resolver_tuple_list = [x for x in resolver_dict.items()]
for ip, resolver in resolver_tuple_list:
resolver_state = "open"
answer_list = queryDNS(
db, status_id, ip, resolver, URL_TO_CHECK, "A"
)
answer_list = queryDNS(db, status_id, ip, resolver, URL_TO_CHECK, "A")
if len(answer_list) == 0:
# We expect a valid response
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment
GitLab Nexedi Edition | About GitLab | About Nexedi | 沪ICP备2021021310号-2 | 沪ICP备2021021310号-7