Added tests for the document service