# HG changeset patch
# User Pierre-Yves David <pierre-yves.david@octobus.net>
# Date 1628011586 -7200
# Node ID 5ad37164a8fe8c11d270431e75cdad78494fb100
# Parent  580bca200874aecd9214ffaad1f3a75d8e1ff4c4
testing: make sure write_file is "atomic"

This make sure viewer cannot see the new file with partial content.

This was likely the cause of some flakiness in `test-nointerrupt.t`

Differential Revision: https://phab.mercurial-scm.org/D11250

diff -r 580bca200874 -r 5ad37164a8fe mercurial/testing/__init__.py
--- a/mercurial/testing/__init__.py	Wed Aug 04 19:45:13 2021 +0200
+++ b/mercurial/testing/__init__.py	Tue Aug 03 19:26:26 2021 +0200
@@ -33,5 +33,11 @@
 
 
 def write_file(path, content=b''):
-    with open(path, 'wb') as f:
+    if content:
+        write_path = b'%s.tmp' % path
+    else:
+        write_path = path
+    with open(write_path, 'wb') as f:
         f.write(content)
+    if path != write_path:
+        os.rename(write_path, path)