Start adding tests for async API of ZK mock