# HG changeset patch
# User Georges Racinet <georges.racinet@octobus.net>
# Date 1605112416 -3600
#      Wed Nov 11 17:33:36 2020 +0100
# Node ID e7f429bf44bd21300b6ec6caad05e104ebc9a090
# Parent  aa085f8e14de303d0af3a0246947f55a96274a35
testhelpers: extracted a working directory preparation method

it will be used in forthcoming new helpers

diff --git a/heptapod/testhelpers.py b/heptapod/testhelpers.py
--- a/heptapod/testhelpers.py
+++ b/heptapod/testhelpers.py
@@ -153,6 +153,13 @@
             random.random(),
             node.hex(self.repo.dirstate.p1()))
 
+    def prepare_wdir(self, parent=None):
+        if parent is not None:
+            if isinstance(parent, bytes):
+                self.update_bin(parent)
+            else:
+                self.update(parent.hex())
+
     def write_commit(self, rpath,
                      content=None, message=None,
                      return_ctx=True,
@@ -195,11 +202,7 @@
         """
         repo = self.repo
         path = os.path.join(repo.root, as_bytes(rpath))
-        if parent is not None:
-            if isinstance(parent, bytes):
-                self.update_bin(parent)
-            else:
-                self.update(parent.hex())
+        self.prepare_wdir(parent=parent)
 
         if content is None:
             content = self.random_content()