1 files changed, 5 insertions, 0 deletions

diff --git a/test/test_helper.rb b/test/test_helper.rb
index c494c68..7ab9c8c 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -67,4 +67,9 @@ class ActiveSupport::TestCase
     node.reload
     node
   end
+
+  def find_or_create_draft(node, user)
+    node.lock_for_editing!(user)
+    node.draft || node.create_new_draft(user)
+  end
 end