add callCount assertion to tests