]>
git.ipfire.org Git - thirdparty/pdns.git/blob - regression-tests.dnsdist/test_TCPOnly.py
4 from dnsdisttests
import DNSDistTest
6 class TestTCPOnly(DNSDistTest
):
9 newServer{address="127.0.0.1:%s", tcpOnly=true}
14 TCP Only: UDP query is sent via TCP
16 name
= 'udp.tcp-only.test.powerdns.com.'
17 query
= dns
.message
.make_query(name
, 'A', 'IN')
18 expectedResponse
= dns
.message
.make_response(query
)
19 rrset
= dns
.rrset
.from_text(name
,
24 expectedResponse
.answer
.append(rrset
)
26 (receivedQuery
, receivedResponse
) = self
.sendUDPQuery(query
, expectedResponse
)
27 receivedQuery
.id = query
.id
28 self
.assertEqual(query
, receivedQuery
)
29 self
.assertEqual(receivedResponse
, expectedResponse
)
31 if 'UDP Responder' in self
._responsesCounter
:
32 self
.assertEqual(self
._responsesCounter
['UDP Responder'], 0)
33 if 'TCP Responder' in self
._responsesCounter
:
34 self
.assertEqual(self
._responsesCounter
['TCP Responder'], 1)
38 TCP Only: TCP query is sent via TCP
40 name
= 'tcp.tcp-only.test.powerdns.com.'
41 query
= dns
.message
.make_query(name
, 'A', 'IN')
42 expectedResponse
= dns
.message
.make_response(query
)
43 rrset
= dns
.rrset
.from_text(name
,
48 expectedResponse
.answer
.append(rrset
)
50 (receivedQuery
, receivedResponse
) = self
.sendTCPQuery(query
, expectedResponse
)
51 receivedQuery
.id = query
.id
52 self
.assertEqual(query
, receivedQuery
)
53 self
.assertEqual(receivedResponse
, expectedResponse
)
54 if 'UDP Responder' in self
._responsesCounter
:
55 self
.assertEqual(self
._responsesCounter
['UDP Responder'], 0)
56 if 'TCP Responder' in self
._responsesCounter
:
57 self
.assertEqual(self
._responsesCounter
['TCP Responder'], 1)