def METHOD_NAME(func):
  def wait_until_done(*args, **kwargs):
    actions = list(client_actions.REGISTRY.values())
    client_mock = action_mocks.ActionMock(*actions)
    flow_test_lib.FinishAllFlows(client_mock=client_mock)
    func(*args, **kwargs)
  return wait_until_done