def test_tcp_query(): """Tests for successful DNS query using tcp.""" assert dnsck_query("8.8.8.8", "google.com", "AAAA", 1, True) == 0
def test_tcp_no_records(): """Tests response with no records returned.""" assert dnsck_query("8.8.8.8", "test.google.com", "A", 1, True) == 0
def test_tcp_unknown_rec_type(): """Unknown record types should raise an exception and exit.""" with pytest.raises(SystemExit): assert dnsck_query("8.8.8.8", "google.com", "abc", 1, True)
def test_tcp_bad_server(): """Tests response with bad server IP address.""" assert dnsck_query("8.8.8.88", "google.com", "A", 1, True) == 1
def test_udp_query(): """Tests for successful DNS query using UDP.""" assert dnsck_query("8.8.8.8", "google.com", "a", 1) == 0