Initial work fixing #201

Author: rushter

selectolax/lexbor.pxd

@@ -279,7 +279,7 @@ cdef class LexborHTMLParser:
 
     @staticmethod
     cdef LexborHTMLParser from_document(lxb_html_document_t * document, bytes raw_html)
-
+    cdef inline lxb_html_document_t* main_document(self)
 
 cdef extern from "lexbor/dom/dom.h" nogil:
     ctypedef enum lexbor_action_t:

selectolax/lexbor.pyx

@@ -55,6 +55,12 @@ cdef class LexborHTMLParser:
         self._parse_html(bytes_html, html_len)
         self.raw_html = bytes_html
 
+    cdef inline lxb_html_document_t* main_document(self):
+        if self._is_fragment:
+            return self._original_document
+        else:
+            return self.document
+
     cdef inline void _new_html_document(self):
         """Initialize a fresh Lexbor HTML document.
 

selectolax/lexbor/node.pxi

@@ -819,7 +819,7 @@ cdef class LexborNode:
         if isinstance(value, (str, bytes, unicode)):
             bytes_val = to_bytes(value)
             new_node = <lxb_dom_node_t *> lxb_dom_document_create_text_node(
-                &self.parser.document.dom_document,
+                &self.parser.main_document().dom_document,
                 <lxb_char_t *> bytes_val, len(bytes_val)
             )
             if new_node == NULL:

tests/test_lexbor.py

@@ -304,3 +304,10 @@ def test_fragment_parser_empty_doc():
 def test_fragment_parser(html, expected_html):
     parser = LexborHTMLParser(html, is_fragment=True)
     assert parser.html == expected_html
+
+
+def test_insert_node_fragment_parser():
+    html = "<div></div>"
+    p = LexborHTMLParser(html, is_fragment=True)
+    p.root.insert_child("text")
+    assert p.html == "<div>text</div>"