public void Can_get_image_no_image_registered(int stage)
{
var context = Substitute.For <IPluginExecutionContext>();
context.PreEntityImages.Returns(new EntityImageCollection());
context.PostEntityImages.Returns(new EntityImageCollection());
context.Stage.Returns(stage);
var handler = ImageEntityGetter.GetHandler(context.Stage);
Assert.Null(handler.Get(context));
}
public void Can_get_image_stage_postoperation()
{
var entity = new Entity("entity")
{
Id = Guid.NewGuid()
};
var context = Substitute.For <IPluginExecutionContext>();
context.Stage.Returns((int)SdkMessageProcessingStepStage.Postoperation);
context.PostEntityImages.Returns(new EntityImageCollection
{
[ImageEntityGetter.ImageKey] = entity
});
var handler = ImageEntityGetter.GetHandler(context.Stage);
Assert.Equal(entity, handler.Get(context));
}
