commit 786e6a6c38de0d0141b30095cbea5566cb8ff719
author Andrew Jeffery <andrew@aj.id.au> Fri May 04 23:03:50 2018 +0930
committer Andrew Jeffery <andrew@aj.id.au> Mon May 14 10:08:18 2018 +0930
tree 1197e16d5906f567d131fd5bd05c643698769c98
parent ffe66a91512a304d5bd31586cbadbf46fbd8ea0d

utils: Add characterisation tests for PathTree

Change-Id: I1ad4832779e85a54a75b3551a8beeba429679e1c
Signed-off-by: Andrew Jeffery <andrew@aj.id.au>

obmc/utils/pathtree.py

diff --git a/obmc/utils/pathtree.py b/obmc/utils/pathtree.py
index 3ca9a7e..da56cfb 100644
--- a/obmc/utils/pathtree.py
+++ b/obmc/utils/pathtree.py
@@ -53,9 +53,9 @@
         try:
             while True:
                 x = next(self.it)
-                depth_exceeded = len(self.path) + 1 > self.depth
-                if self.depth and depth_exceeded:
-                    continue
+                if self.depth:
+                    if len(self.path) + 1 > self.depth:
+                        continue
                 self.itlist.append(self.it)
                 self.path.append(x[0])
                 # TODO: openbmc/openbmc#2994 remove python 2 support
@@ -122,7 +122,7 @@
         return d[elements[-1]]
 
     def __iter__(self):
-        return self
+        return PathTreeItemIterator(self, '/', None)
 
     def __missing__(self, key):
         for x in self.iterkeys():