issue #76, #370: add disconnect cleanup test

issue260
David Wilson 6 years ago
parent 6bf4d79a54
commit c148c869e6

@ -268,6 +268,23 @@ class ContextService(mitogen.service.Service):
finally:
self._lock.release()
@mitogen.service.expose(mitogen.service.AllowParents())
def dump(self):
"""
For testing, return a list of dicts describing every currently
connected context.
"""
return [
{
'context_name': context.name,
'via': getattr(self._via_by_context.get(context),
'name', None),
'refs': self._refs_by_context.get(context),
}
for context, key in sorted(self._key_by_context.items(),
key=lambda c_k: c_k[0].context_id)
]
@mitogen.service.expose(mitogen.service.AllowParents())
def shutdown_all(self):
"""

@ -1,2 +1,3 @@
- import_playbook: disconnect_cleanup.yml
- import_playbook: lru_one_target.yml
- import_playbook: reconnection.yml

@ -0,0 +1,46 @@
# issue #76, #370: ensure context state is forgotten on disconnect, including
# state of dependent contexts (e.g. sudo, connection delegation, ..).
- name: integration/context_service/disconnect_cleanup.yml
hosts: test-targets
any_errors_fatal: true
tasks:
- meta: end_play
when: not is_mitogen
# Start with a clean slate.
- mitogen_shutdown_all:
# Connect a few users.
- shell: "true"
become: true
become_user: "mitogen__user{{item}}"
with_items: [1, 2, 3]
# Verify current state.
- mitogen_action_script:
script: |
self._connection._connect()
result['dump'] = self._connection.parent.call_service(
service_name='ansible_mitogen.services.ContextService',
method_name='dump'
)
register: out
- assert:
that: out.dump|length == 4 # ssh account + 3 sudo accounts
- meta: reset_connection
# Verify current state.
- mitogen_action_script:
script: |
self._connection._connect()
result['dump'] = self._connection.parent.call_service(
service_name='ansible_mitogen.services.ContextService',
method_name='dump'
)
register: out
- assert:
that: out.dump|length == 1 # just the ssh account
Loading…
Cancel
Save