feat: add Test cases for querying entities with count parameter set to true
check the result returned after querying entities when the count parameter is set to true.
check the result returned after querying entities when the count parameter is set to true.