add count support to mock