Updated DocumentService unit test