add test for count query