From 6c71c5bfefa5552a2241a7daabf308b52d6b9bfe Mon Sep 17 00:00:00 2001
From: David Wilson <dw@botanicus.net>
Date: Wed, 31 Oct 2018 18:30:03 +0000
Subject: [PATCH] issue #369: disable reset_connection on Ansible<2.5.6

https://github.com/ansible/ansible/issues/27520
---
 ansible_mitogen/connection.py                 | 27 +++++++++++++++----
 .../ansible/integration/connection/reset.yml  |  6 +++++
 2 files changed, 28 insertions(+), 5 deletions(-)

diff --git a/ansible_mitogen/connection.py b/ansible_mitogen/connection.py
index b566b541..e017608e 100644
--- a/ansible_mitogen/connection.py
+++ b/ansible_mitogen/connection.py
@@ -31,6 +31,7 @@ from __future__ import unicode_literals
 
 import logging
 import os
+import pprint
 import random
 import stat
 import time
@@ -699,11 +700,16 @@ class Connection(ansible.plugins.connection.ConnectionBase):
         representing the target. If no connection exists yet, ContextService
         will establish it before returning it or throwing an error.
         """
-        dct = self.parent.call_service(
-            service_name='ansible_mitogen.services.ContextService',
-            method_name='get',
-            stack=mitogen.utils.cast(list(stack)),
-        )
+        try:
+            dct = self.parent.call_service(
+                service_name='ansible_mitogen.services.ContextService',
+                method_name='get',
+                stack=mitogen.utils.cast(list(stack)),
+            )
+        except mitogen.core.CallError:
+            LOG.warning('Connection failed; stack configuration was:\n%s',
+                        pprint.pformat(stack))
+            raise
 
         if dct['msg']:
             if dct['method_name'] in self.become_methods:
@@ -809,6 +815,10 @@ class Connection(ansible.plugins.connection.ConnectionBase):
             self.broker = None
             self.router = None
 
+    reset_compat_msg = (
+        'Mitogen only supports "reset_connection" on Ansible 2.5.6 or later'
+    )
+
     def reset(self):
         """
         Explicitly terminate the connection to the remote host. This discards
@@ -816,6 +826,13 @@ class Connection(ansible.plugins.connection.ConnectionBase):
         the 'disconnected' state, and informs ContextService the connection is
         bad somehow, and should be shut down and discarded.
         """
+        if self._play_context.remote_addr is None:
+            # <2.5.6 incorrectly populate PlayContext for reset_connection
+            # https://github.com/ansible/ansible/issues/27520
+            raise ansible.errors.AnsibleConnectionFailure(
+                self.reset_compat_msg
+            )
+
         self._connect()
         self._mitogen_reset(mode='reset')
 
diff --git a/tests/ansible/integration/connection/reset.yml b/tests/ansible/integration/connection/reset.yml
index a0ba520b..768cd2d5 100644
--- a/tests/ansible/integration/connection/reset.yml
+++ b/tests/ansible/integration/connection/reset.yml
@@ -9,6 +9,12 @@
     - meta: end_play
       when: not is_mitogen
 
+    - debug: msg="reset.yml skipped on Ansible<2.5.6"
+      when: ansible_version.full < '2.5.6'
+
+    - meta: end_play
+      when: ansible_version.full < '2.5.6'
+
     - custom_python_detect_environment:
       register: out