Return the message in question_only mode. [#556]

diff --git a/dns/message.py b/dns/message.py
index bc4cced19cbd24992e4751c80585f0d836845df5..59e2692bd7d4a08ac3f42cdb36200387637dab7e 100644 (file)
--- a/dns/message.py
+++ b/dns/message.py
@@ -954,7 +954,7 @@ class _WireReader:
             self.message._get_one_rr_per_rrset(self.one_rr_per_rrset)
         self._get_question(MessageSection.QUESTION, qcount)
         if self.question_only:
-            return
+            return self.message
         self._get_section(MessageSection.ANSWER, ancount)
         self._get_section(MessageSection.AUTHORITY, aucount)
         self._get_section(MessageSection.ADDITIONAL, adcount)