Add tests for expected CQL for counts