Fix test index() reader to work in same way as real one

* Going to remove and just use real index reader
Add todo