mirror of
https://github.com/containers/podman.git
synced 2025-07-18 01:57:24 +08:00
8aed3857d3
Catching the tests up with 60427ab3 (add podman remote client,
2018-06-22, #986) to avoid non-fatal smoketest failures like [1]:
======================================================================
FAIL: test_tunnel (test.test_tunnel.TestTunnel)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/usr/lib64/python3.6/unittest/mock.py", line 1179, in patched
return func(*args, **keywargs)
File "/go/src/github.com/projectatomic/libpod/contrib/python/test/test_tunnel.py", line 79, in test_tunnel
mock_Popen.assert_called_once_with(cmd, close_fds=True)
File "/usr/lib64/python3.6/unittest/mock.py", line 825, in assert_called_once_with
return self.assert_called_with(*args, **kwargs)
File "/usr/lib64/python3.6/unittest/mock.py", line 814, in assert_called_with
raise AssertionError(_error_message()) from cause
AssertionError: Expected call: Popen(['ssh', '-nNT', '-L', '/tmp/user/socket:/run/podman/socket', '-i', '~/.ssh/id_rsa', 'ssh://user@hostname'], close_fds=True)
Actual call: Popen(['ssh', '-nNTq', '-L', '/tmp/user/socket:/run/podman/socket', '-i', '~/.ssh/id_rsa', 'ssh://user@hostname'], close_fds=True)
[1]: 0d792d5c92.1.1529764423989739036/output.log
Signed-off-by: W. Trevor King <wking@tremily.us>
Closes: #1035
Approved by: mheon
80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
from __future__ import absolute_import
|
|
|
|
import time
|
|
import unittest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import podman
|
|
from podman.libs.tunnel import Context, Portal, Tunnel
|
|
|
|
|
|
class TestTunnel(unittest.TestCase):
|
|
def setUp(self):
|
|
self.tunnel_01 = MagicMock(spec=Tunnel)
|
|
self.tunnel_02 = MagicMock(spec=Tunnel)
|
|
|
|
def test_portal_ops(self):
|
|
portal = Portal(sweap=500)
|
|
portal['unix:/01'] = self.tunnel_01
|
|
portal['unix:/02'] = self.tunnel_02
|
|
|
|
self.assertEqual(portal.get('unix:/01'), self.tunnel_01)
|
|
self.assertEqual(portal.get('unix:/02'), self.tunnel_02)
|
|
|
|
del portal['unix:/02']
|
|
with self.assertRaises(KeyError):
|
|
portal['unix:/02']
|
|
self.assertEqual(len(portal), 1)
|
|
|
|
def test_portal_reaping(self):
|
|
portal = Portal(sweap=0.5)
|
|
portal['unix:/01'] = self.tunnel_01
|
|
portal['unix:/02'] = self.tunnel_02
|
|
|
|
self.assertEqual(len(portal), 2)
|
|
for entry in portal:
|
|
self.assertIn(entry, (self.tunnel_01, self.tunnel_02))
|
|
|
|
time.sleep(1)
|
|
portal.reap()
|
|
self.assertEqual(len(portal), 0)
|
|
|
|
def test_portal_no_reaping(self):
|
|
portal = Portal(sweap=500)
|
|
portal['unix:/01'] = self.tunnel_01
|
|
portal['unix:/02'] = self.tunnel_02
|
|
|
|
portal.reap()
|
|
self.assertEqual(len(portal), 2)
|
|
for entry in portal:
|
|
self.assertIn(entry, (self.tunnel_01, self.tunnel_02))
|
|
|
|
@patch('subprocess.Popen')
|
|
@patch('os.path.exists', return_value=True)
|
|
@patch('weakref.finalize')
|
|
def test_tunnel(self, mock_finalize, mock_exists, mock_Popen):
|
|
context = Context(
|
|
'unix:/01',
|
|
'io.projectatomic.podman',
|
|
'/tmp/user/socket',
|
|
'/run/podman/socket',
|
|
'user',
|
|
'hostname',
|
|
'~/.ssh/id_rsa',
|
|
)
|
|
tunnel = Tunnel(context).bore('unix:/01')
|
|
|
|
cmd = [
|
|
'ssh',
|
|
'-nNTq',
|
|
'-L',
|
|
'{}:{}'.format(context.local_socket, context.remote_socket),
|
|
'-i',
|
|
context.identity_file,
|
|
'ssh://{}@{}'.format(context.username, context.hostname),
|
|
]
|
|
|
|
mock_finalize.assert_called_once_with(tunnel, tunnel.close, 'unix:/01')
|
|
mock_exists.assert_called_once_with(context.local_socket)
|
|
mock_Popen.assert_called_once_with(cmd, close_fds=True)
|