Add tests

app/helpers/success_criteria_helper.rb

@@ -10,4 +10,20 @@ module SuccessCriteriaHelper
       'bi bi-question text-warning'
     end
   end
+
+  def success_criterion_edit_button(success_criterion, edit_mode)
+    path = if success_criterion.persisted?
+             if edit_mode
+               success_criterion
+             else
+               [:edit,
+                success_criterion]
+             end
+           else
+             success_criterion.element
+           end
+    link_to tag.i(class: 'bi bi-pencil'),
+            path,
+            class: "btn btn-#{edit_mode ? 'link text-warning' : 'link text-secondary'}"
+  end
 end