It would be nice here to assert that `mock_query` was actually called with the `scan_index_forward` and `limit` arguments.