From 053bae33053ee584188f860584bc4b2303a1d355 Mon Sep 17 00:00:00 2001
From: Kesara Rathnayake <kesara@fq.nz>
Date: Thu, 26 Sep 2024 20:52:42 +1200
Subject: [PATCH] test: Test DatatrackerToBibConverter

---
 test.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/test.py b/test.py
index c039687d..fa7c9bdf 100644
--- a/test.py
+++ b/test.py
@@ -12,6 +12,7 @@
 from xml2rfc.boilerplate_rfc_7841 import boilerplate_rfc_status_of_memo
 from xml2rfc.walkpdf import xmldoc
 from xml2rfc.writers.base import default_options, BaseV3Writer, RfcWriterError
+from xml2rfc.writers import DatatrackerToBibConverter
 from xml2rfc.writers.text import MAX_WIDTH
 
 try:
@@ -847,5 +848,39 @@ def test_validate_draft_name(self):
             self.writer.validate_draft_name()
 
 
+class DatatrackerToBibConverterTest(unittest.TestCase):
+    """DatatrackerToBibConverter tests"""
+
+    def setUp(self):
+        xml2rfc.log.quiet = True
+        path = "tests/input/elements.xml"
+        self.parser = xml2rfc.XmlRfcParser(path,
+                                           quiet=True,
+                                           options=default_options,
+                                           **options_for_xmlrfcparser)
+        self.xmlrfc = self.parser.parse()
+        self.writer = xml2rfc.DatatrackerToBibConverter(self.xmlrfc, quiet=True)
+
+    def test_convert(self):
+        reference_a = "reference.I-D.ietf-sipcore-multiple-reasons.xml"
+        reference_b = "reference.I-D.draft-ietf-sipcore-multiple-reasons-01.xml"
+        rfc = lxml.etree.fromstring(f"""
+<rfc xmlns:xi="http://www.w3.org/2001/XInclude">
+    <xi:include href="https://datatracker.ietf.org/doc/bibxml3/{reference_a}"/>
+    <xi:include href="https://datatracker.ietf.org/doc/bibxml3/{reference_b}"/>
+</rfc>""")
+        self.writer.root = rfc
+        self.writer.convert_xincludes()
+
+        ns = {"xi": b"http://www.w3.org/2001/XInclude"}
+        xincludes = self.writer.root.xpath("//xi:include", namespaces=ns)
+
+        # reference without revision
+        self.assertEqual(xincludes[0].get("href"), f"https://bib.ietf.org/public/rfc/bibxml-ids/{reference_a}")
+
+        # reference with revision
+        self.assertEqual(xincludes[1].get("href"), f"https://bib.ietf.org/public/rfc/bibxml-ids/{reference_b}")
+
+
 if __name__ == '__main__':
     unittest.main()