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_recursion_desired_value_from_flags_false(self):
"""
Test case to retrieve the recursion_desired field when set to False in the flags.
"""
# flags = 0 has recursion_desired = False
get_recursion_desired_value_false: bool = DNSMessage._get_recursion_desired_value_from_flags(
0)
self.assertEqual(get_recursion_desired_value_false, False)
def test_get_recursion_desired_value_from_flags_true(self):
"""
Test case to retrieve the recursion_desired field when set to True in the flags.
"""
# flags = 256 has recursion_desired = True
get_recursion_desired_value_true: bool = DNSMessage._get_recursion_desired_value_from_flags(
256)
self.assertEqual(get_recursion_desired_value_true, True)
