Reinitialize context and rerun custom_edit when rendering edit view due to failed update

Co-authored-by: Mason Ballengee <[email protected]>

Author: cjcolvar

Diff for: lib/avalon/workflow/workflow_controller_behavior.rb

@@ -70,6 +70,11 @@ def update
         flashes = { error: context[:error], notice: context[:notice]}
         if model_object.errors.present?
           flash.now[:error] = 'There are errors with your submission. Please correct them before continuing.'
+
+          # Refresh the context before rendering edit
+          context = perform_step_action :before_step
+          custom_edit #yield to custom_edit in the controller
+
           render :edit
         elsif model_object.workflow.published? && model_object.workflow.current?(@active_step)
           redirect_to(polymorphic_path(model_object), flash: flashes)