Move common parts of ZK mock to companion object