add get_counts to mock interface