def mock_xcom_push(*args, **kwargs):
    key = kwargs['key']
    value = kwargs['value']
    messages[key] = value
def mock_xcom_push(*args, **kwargs):
    key = kwargs['key']
    value = kwargs['value']
    messages[key] = value