Add tests for calling stream and page with a query