Fixed tests

Capitains · Jan 27, 2017 · 6bea1bb · 6bea1bb
1 parent 0bb48d1
commit 6bea1bb
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/MyCapytain/resources/prototypes/cts/inventory.py b/MyCapytain/resources/prototypes/cts/inventory.py
@@ -232,6 +232,8 @@ def __export__(self, output=None, domain="", namespaces=True, lines="\n"):
         """
         if output == Mimetypes.XML.CTS:
             attrs = {"urn": self.id, "xml:lang": self.lang}
+            if self.parent is not None and self.parent.id:
+                attrs["workUrn"] = self.parent.id
             if namespaces is True:
                 attrs.update(self.__namespaces_header__())
 

diff --git a/tests/resources/collections/test_cts.py b/tests/resources/collections/test_cts.py
@@ -36,6 +36,14 @@ def sortbytext(elem):
         else:
             return ''
 
+    @staticmethod
+    def sortbyattr(elem):
+        attrkeys = sorted(elem.keys())
+        if len(attrkeys):
+            return elem.get(attrkeys[0])
+        else:
+            return ""
+
     @staticmethod
     def sortAttrs(item, sorteditem):
         attrkeys = sorted(item.keys())
@@ -57,6 +65,7 @@ def sortElements(items, newroot):
         # We do this by performing three sorts in the reverse order
         items = sorted(items, key=XML_Compare.sortbyid)
         items = sorted(items, key=XML_Compare.sortbytext)
+        items = sorted(items, key=XML_Compare.sortbyattr)
         items = sorted(items, key=attrgetter('tag'))
 
         # Once sorted, we sort each of the items