# HG changeset patch
# User Martin von Zweigbergk <martinvonz@google.com>
# Date 2016-02-08 05:14:01
# Node ID 8ab91d9290ce8facf4d90a08330b13e708ed7ae1
# Parent  53f42c8d5f71c3f5caf11a00f827a47cea756645

treemanifest: implement iterentries()

To make tests pass with _treeinmem manually set to True, we need to
implement the recently added iterentries() on the treemanifest class
too.

diff --git a/mercurial/manifest.py b/mercurial/manifest.py
--- a/mercurial/manifest.py
+++ b/mercurial/manifest.py
@@ -520,6 +520,15 @@ class treemanifest(object):
         self._node = node
         self._dirty = False
 
+    def iterentries(self):
+        self._load()
+        for p, n in sorted(self._dirs.items() + self._files.items()):
+            if p in self._files:
+                yield self._subpath(p), n, self._flags.get(p, '')
+            else:
+                for x in n.iterentries():
+                    yield x
+
     def iteritems(self):
         self._load()
         for p, n in sorted(self._dirs.items() + self._files.items()):