X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/3a0c83394d5783715fab2be29fa1a9cfc3574e28..011b98a4661b754a0789ae78e145437b5b86e5fe:/tests/test_ref.py?ds=sidebyside

diff --git a/tests/test_ref.py b/tests/test_ref.py
index 2383866..d75228e 100644
--- a/tests/test_ref.py
+++ b/tests/test_ref.py
@@ -11,12 +11,14 @@ from lxml import etree
 class RefTests(TestCase):
     def test_snippet(self):
         doc = WLDocument(filename=get_fixture('text', 'miedzy-nami-nic-nie-bylo.xml'))
-        doc._compat_assign_section_ids()
+
+        hb = builders['html']()
+        hb.assign_ids(doc.tree)
 
         refs = []
         for ref in doc.references():
             snippet = ref.get_snippet()
-            b = builders['html']()
+            b = builders['html-snippet']()
 
             for s in snippet:
                 s.html_build(b)