<Bug>[PyutSDInstance]: <Need to update Pyut Model>

[]

[#26]

Author: Humberto Sanchez II

tests/untanglepyut/TestUnTangleSequenceDiagram.py

@@ -10,6 +10,7 @@
 from pkg_resources import resource_filename
 
 from pyutmodel.PyutLinkType import PyutLinkType
+from pyutmodel.PyutSDInstance import PyutSDInstance
 from pyutmodel.PyutSDMessage import PyutSDMessage
 
 from untanglepyut.UnTangler import Document
@@ -65,6 +66,19 @@ def testSequenceMessageTypeIsCorrect(self):
             pyutSDMessage: PyutSDMessage = sdMessage.getPyutObject()
             self.assertEqual(PyutLinkType.SD_MESSAGE, pyutSDMessage.linkType, 'Link type not correctly set')
 
+    def testSequenceMessagePyutSDMessageIsCorrect(self):
+        document: Document = self._retrieveSequenceDiagramDocument()
+
+        sdMessages: OglSDMessages = document.oglSDMessages
+        for sdMessage in sdMessages.values():
+            pyutSDMessage: PyutSDMessage = sdMessage.getPyutObject()
+
+            pyutSrcInstance: PyutSDInstance = pyutSDMessage.getSource()
+            pyutDstInstance: PyutSDInstance = pyutSDMessage.getDest()
+
+            self.assertIsNotNone(pyutSrcInstance, 'Missing source sd instance')
+            self.assertIsNotNone(pyutDstInstance, 'Missing source sd instance')
+
     def _retrieveSequenceDiagramDocument(self) -> Document:
 
         fqFileName: str       = resource_filename(TestBase.RESOURCES_PACKAGE_NAME, 'SequenceDiagram.xml')

untanglepyut/UntangleSequenceDiagram.py

@@ -94,6 +94,8 @@ def _untangleSDMessages(self, pyutDocument: Element) -> OglSDMessages:
             srcInstance: OglSDInstance = self._oglSDInstances[bogus.sourceId]
             dstInstance: OglSDInstance = self._oglSDInstances[bogus.destinationId]
 
+            pyutSDMessage.setSource(srcInstance.getPyutObject)          # Ugh, time was set by sdMessageToPyutSDMessage
+            pyutSDMessage.setDestination(dstInstance.getPyutObject)     # This "split" functionality must be fixed
             oglSDMessage: OglSDMessage = OglSDMessage(srcShape=srcInstance, pyutSDMessage=pyutSDMessage, dstShape=dstInstance)
 
             srcInstance.addLink(link=oglSDMessage)