Add tests for the DocumentQuery class. Make sure it returns the correct values.