# HG changeset patch
# User Pierre-Yves David <pierre-yves.david@octobus.net>
# Date 1629835670 -7200
# Node ID 74a8d5c6fdc6db20cf7da9cc8ec8eac0688cee49
# Parent  bbae583333e905b262d2a4e627ddeaa449ca92ee
dirstate: also wrap the new method in `dirstatenonnormalcheck`

The goal of this is to make sure we set the data right, so we need to make sure
it run after the new method, that we actually call, in addition to the old one,
that we no longer call.

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

diff -r bbae583333e9 -r 74a8d5c6fdc6 contrib/dirstatenonnormalcheck.py
--- a/contrib/dirstatenonnormalcheck.py	Fri Aug 20 16:12:33 2021 +0200
+++ b/contrib/dirstatenonnormalcheck.py	Tue Aug 24 22:07:50 2021 +0200
@@ -67,3 +67,12 @@
             extensions.wrapfunction(dirstatecl, 'remove', _checkdirstate)
             extensions.wrapfunction(dirstatecl, 'merge', _checkdirstate)
             extensions.wrapfunction(dirstatecl, 'drop', _checkdirstate)
+            extensions.wrapfunction(dirstatecl, 'set_tracked', _checkdirstate)
+            extensions.wrapfunction(dirstatecl, 'set_untracked', _checkdirstate)
+            extensions.wrapfunction(
+                dirstatecl, 'set_possibly_dirty', _checkdirstate
+            )
+            extensions.wrapfunction(
+                dirstatecl, 'update_file_p1', _checkdirstate
+            )
+            extensions.wrapfunction(dirstatecl, 'update_file', _checkdirstate)