Token aware policy: Use lazy child iterator

- More tests