SERVER-15495 add tests for query_stage_count