Add tests for the document level focus APIs