Keys is extensively tested in other tests already. (but in those tests there is no total_rows check). I'd suggest to remove keys tests from here and add total_rows check there. What do you think?