summaryrefslogtreecommitdiffstats
path: root/tests/pytests/test_edns.py
blob: b19af3773fbf6397337b75e8cce061881faeaf66 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# SPDX-License-Identifier: GPL-3.0-or-later
"""EDNS tests"""

import dns
import pytest

import utils


@pytest.mark.parametrize('dname, code, text', [
    ('deny.test.', dns.edns.EDECode.BLOCKED, 'CR36'),
    ('refuse.test.', dns.edns.EDECode.PROHIBITED, 'EIM4'),
    ('forge.test.', dns.edns.EDECode.FORGED_ANSWER, '5DO5'),
])
def test_edns_ede(kresd_sock, dname, code, text):
    """Check that kresd responds with EDNS EDE codes in selected cases."""
    buff, msgid = utils.get_msgbuff(dname)
    kresd_sock.sendall(buff)
    answer = utils.receive_parse_answer(kresd_sock)
    assert answer.id == msgid
    assert answer.options[0].code == code
    assert answer.options[0].text == text