valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-06 16:45:27 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Comment the tcp message client test better

Browse Source
This commit is contained in:
Daniel A. Wozniak 2019-06-20 18:31:13 +00:00
parent dcf576f4c9
commit 9606db79fc
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/unit/transport/test_tcp.py
View File

@ -335,19 +335,29 @@ class SaltMessageClientCleanupTest(TestCase, AdaptedConfigurationTestCaseMixin):
'''
opts = self.get_temp_config('master')
client = SaltMessageClient(opts, 'localhost', self.port)
# Mock the io_loop's stop method so we know when it has been called.
orig_loop = tornado.ioloop.IOLoop.current()
orig_loop.real_stop = orig_loop.stop
orig_loop.stop_called = False
def stop(*args, **kwargs):
orig_loop.stop_called = True
orig_loop.real_stop()
orig_loop.stop = stop
try:
orig_loop.stop = stop
assert client.io_loop == orig_loop
client.io_loop.run_sync(client.connect)
# Ensure we are testing the _read_until_future and io_loop teardown
assert client._stream is not None
assert client._read_until_future is not None
assert orig_loop.stop_called is True
# The run_sync call will set stop_called, reset it
orig_loop.stop_called = False
client.close()
# Stop should be called again, client's io_loop should be None
assert orig_loop.stop_called is True
assert client.io_loop is None
finally: