SetState & SetStateDynamicEntity use message property of EntityMoniker

Author: aelsmore

spkl/SparkleXrm.Tasks/PluginRegistraton.cs

@@ -456,23 +456,34 @@ private SdkMessageProcessingStepImage RegisterImage(CrmPluginRegistrationAttribu
             {
                 return null;
             }
-            var image = existingImages.Where(a =>
-                            a.SdkMessageProcessingStepId.Id == step.Id
-                            &&
-                            a.EntityAlias == imageName
-                            && a.ImageType == (sdkmessageprocessingstepimage_imagetype)imagetype).FirstOrDefault();
-            if (image == null)
-            {
-                image = new SdkMessageProcessingStepImage();
-            }
+            
+            var image = existingImages.FirstOrDefault(
+                            a => a.SdkMessageProcessingStepId.Id == step.Id
+                                 && a.EntityAlias == imageName
+                                 && a.ImageType == (sdkmessageprocessingstepimage_imagetype)imagetype) ??
+                        new SdkMessageProcessingStepImage();
 
             image.Name = imageName;
-           
+
             image.ImageType = (sdkmessageprocessingstepimage_imagetype)imagetype;
             image.SdkMessageProcessingStepId = new EntityReference(SdkMessageProcessingStep.EntityLogicalName, step.Id);
             image.Attributes1 = attributes;
             image.EntityAlias = imageName;
-            image.MessagePropertyName = stepAttribute.Message == "Create" ? "Id" : "Target";
+
+            switch (stepAttribute.Message)
+            {
+                case "Create":
+                    image.MessagePropertyName = "Id";
+                    break;
+                case "SetState":
+                case "SetStateDynamicEntity":
+                    image.MessagePropertyName = "EntityMoniker";
+                    break;
+                default:
+                    image.MessagePropertyName = "Target";
+                    break;
+            }
+
             if (image.Id == Guid.Empty)
             {
                 _trace.WriteLine("Registering Image '{0}'", image.Name);