def test_attempt_to_retrieve_all_flags(self):
"""
Test case to retrieve all the flag values from a hypothetical flags value for a dns message.
"""
get_response_value: bool = DNSMessage._get_response_value_from_flags(
46218)
get_opcode_value: int = DNSMessage._get_opcode_value_from_flags(46218)
get_authoritative_value: bool = DNSMessage._get_authoritative_value_from_flags(
46218)
get_is_truncated_value: bool = DNSMessage._get_is_truncated_value_from_flags(
46218)
get_recursion_desired_value: bool = DNSMessage._get_recursion_desired_value_from_flags(
46218)
get_recursion_available_value: bool = DNSMessage._get_recursion_available_value_from_flags(
46218)
get_rcode_value: int = DNSMessage._get_rcode_value_from_flags(46218)
self.assertEqual(get_response_value, True)
self.assertEqual(get_opcode_value, 6)
self.assertEqual(get_authoritative_value, True)
self.assertEqual(get_is_truncated_value, False)
self.assertEqual(get_recursion_desired_value, False)
self.assertEqual(get_recursion_available_value, True)
self.assertEqual(get_rcode_value, 10)
def test_get_authoritative_value_from_flags_false(self):
"""
Test case to retrieve the authoritative field when set to False in the flags.
"""
# flags = 0 has authoritative = False
get_authoritative_value_false: bool = DNSMessage._get_authoritative_value_from_flags(
0)
self.assertEqual(get_authoritative_value_false, False)
def test_get_authoritative_value_from_flags_true(self):
"""
Test case to retrieve the authoritative field when set to True in the flags.
"""
# flags = 1024 has authoritative = True
get_authoritative_value_true: bool = DNSMessage._get_authoritative_value_from_flags(
1024)
self.assertEqual(get_authoritative_value_true, True)
