1 # Copyright (c) 2017 The Chromium OS Authors. All rights reserved. 2 # Use of this source code is governed by a BSD-style license that can be 3 # found in the LICENSE file. 4 5 """Unit tests for server/cros/network/rf_switch/rf_switch_ssh_tunnel.py.""" 6 7 import unittest 8 import socket 9 10 from autotest_lib.client.common_lib.test_utils import mock 11 from autotest_lib.server.cros.network.rf_switch import rf_switch 12 from autotest_lib.server.cros.network.rf_switch import rf_switch_ssh_tunnel 13 from autotest_lib.server.cros.network.rf_switch import scpi_ssh_tunnel 14 15 16 class RfSwitchSshTunnelTestCases(unittest.TestCase): 17 """Unit tests for RfSwitchSshTunnel.""" 18 19 _HOST = '1.2.3.4' 20 _PROXY_HOST = '4.3.2.1' 21 _PROXY_USER = 'user' 22 _PROXY_PASSWORD = 'VerySecret' 23 24 def setUp(self): 25 self.god = mock.mock_god(debug=False) 26 self.god.stub_class(socket, 'socket') 27 28 def tearDown(self): 29 self.god.unstub_all() 30 31 def testCorrectParentInitializerCalled(self): 32 """RFSwitchSshTunnel calls correct parent.""" 33 self.god.stub_class_method(scpi_ssh_tunnel.ScpiSshTunnel, '__init__') 34 self.god.stub_class_method(rf_switch.RfSwitch, '__init__') 35 scpi_ssh_tunnel.ScpiSshTunnel.__init__.expect_any_call() 36 rf_switch_ssh_tunnel.RfSwitchSshTunnel(self._HOST, 37 self._PROXY_HOST, 38 self._PROXY_USER, 39 self._PROXY_PASSWORD) 40 self.god.check_playback() 41 self.assertEqual(0, rf_switch.RfSwitch.__init__.num_calls, 42 'RfSwitch Initializer was called.') 43 44 if __name__ == '__main__': 45 unittest.main() 46