Eliminate the need for a serial parameter to inbound_xfr()

1 files changed, 15 insertions, 0 deletions

diff --git a/tests/test_xfr.py b/tests/test_xfr.py
index c1a011c..7429fa3 100644
--- a/tests/test_xfr.py
+++ b/tests/test_xfr.py
@@ -593,6 +593,21 @@ def test_make_query_bad_serial():
     with pytest.raises(ValueError):
         dns.xfr.make_query(z, serial=4294967296)
 
+def test_extract_serial_from_query():
+    z = dns.versioned.Zone('example.')
+    (q, s) = dns.xfr.make_query(z)
+    xs = dns.xfr.extract_serial_from_query(q)
+    assert s is None
+    assert s == xs
+    (q, s) = dns.xfr.make_query(z, serial=10)
+    print(q)
+    xs = dns.xfr.extract_serial_from_query(q)
+    assert s == 10
+    assert s == xs
+    q = dns.message.make_query('example', 'a')
+    with pytest.raises(ValueError):
+        dns.xfr.extract_serial_from_query(q)
+
 
 class XFRNanoNameserver(Server):