# Copyright (c) 2017 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unit tests for server/cros/network/rf_switch/rf_switch_ssh_tunnel.py."""

import unittest
import socket

from autotest_lib.client.common_lib.test_utils import mock
from autotest_lib.server.cros.network.rf_switch import rf_switch
from autotest_lib.server.cros.network.rf_switch import rf_switch_ssh_tunnel
from autotest_lib.server.cros.network.rf_switch import scpi_ssh_tunnel


class RfSwitchSshTunnelTestCases(unittest.TestCase):
    """Unit tests for RfSwitchSshTunnel."""

    _HOST = '1.2.3.4'
    _PROXY_HOST = '4.3.2.1'
    _PROXY_USER = 'user'
    _PROXY_PASSWORD = 'VerySecret'

    def setUp(self):
        self.god = mock.mock_god(debug=False)
        self.god.stub_class(socket, 'socket')

    def tearDown(self):
        self.god.unstub_all()

    def testCorrectParentInitializerCalled(self):
        """RFSwitchSshTunnel calls correct parent."""
        self.god.stub_class_method(scpi_ssh_tunnel.ScpiSshTunnel, '__init__')
        self.god.stub_class_method(rf_switch.RfSwitch, '__init__')
        scpi_ssh_tunnel.ScpiSshTunnel.__init__.expect_any_call()
        rf_switch_ssh_tunnel.RfSwitchSshTunnel(self._HOST,
                                               self._PROXY_HOST,
                                               self._PROXY_USER,
                                               self._PROXY_PASSWORD)
        self.god.check_playback()
        self.assertEqual(0, rf_switch.RfSwitch.__init__.num_calls,
                         'RfSwitch Initializer was called.')

if __name__ == '__main__':
    unittest.main()