fix #138 and #151 + 2 new unit tests (#154)

* fix #138 and #151: pass correct list_headers arguments + 2 new unit tests

* added whitespaces for flake8

Co-authored-by: Vinit Kumar <[email protected]>

Author: DirkRichter

json2xml/dicttoxml.py

@@ -253,7 +253,7 @@ def dict2xml_str(
     else:
         # we can not use convert_dict, because rawitem could be non-dict
         subtree = convert(
-            rawitem, ids, attr_type, item_func, cdata, item_wrap, item_name, list_headers=True
+            rawitem, ids, attr_type, item_func, cdata, item_wrap, item_name, list_headers=list_headers
         )
     if parentIsList and list_headers:
         return f"<{parent}>{subtree}</{parent}>"
@@ -365,7 +365,7 @@ def convert_dict(
                 dict2xml_str(
                     attr_type, attr, val, item_func, cdata, key, item_wrap,
                     False,
-                    list_headers=True
+                    list_headers=list_headers
                 )
             )
 

tests/test_dict2xml.py

@@ -211,3 +211,15 @@ def test_dict2xml_with_item_func(self):
         assert b'<?xml version="1.0" encoding="UTF-8" ?>' \
             b'<root><a>1</a><a>2</a><a>3</a><non_flat_list><a>4</a><a>5</a><a>6</a></non_flat_list>' \
             b'</root>' == result
+
+    def test_dict2xml_with_item_func_issue_151(self):
+        data = [{'x': [1]}]
+        result = dicttoxml.dicttoxml(data, root=False, attr_type=False, item_func=lambda y: y+'item')
+        print(result)
+        assert b'<item><x><xitem>1</xitem></x></item>' == result
+
+    def test_dict2xml_issue_151(self):
+        data = [{'x': [1]}]
+        result = dicttoxml.dicttoxml(data, root=False, attr_type=False)
+        print(result)
+        assert b'<item><x><item>1</item></x></item>' == result