Add unit test for document reader