Add helper methods for ZK mock for easy stubbing