"""Tests for rf_switch_client_box."""

import common
import mock
import unittest

from autotest_lib.server.cros.network import rf_switch_client_box


class RfSwitchClientBoxTest(unittest.TestCase):
    """Tests for the RfSwitchClientBox."""


    def setUp(self):
        """Initial set up for the tests."""
        rf_switch_client_box.frontend = mock.MagicMock()
        self.client_box_host = rf_switch_client_box.frontend.Host('', '')
        self.client_box_host.hostname = 'chromeos9-clientbox1'
        self.client_box_host.labels = [
                'rf_switch_1', 'client_box_1', 'rf_switch_client']


    def testGetDevices(self):
        """Test to get all devices from a Client Box."""
        rf_switch_client_box.frontend = mock.MagicMock()
        dut_host = rf_switch_client_box.frontend.Host('', '')
        dut_host.hostname = 'chromeos9-device1'
        dut_host.labels = ['rf_switch_1', 'client_box_1', 'rf_switch_dut']
        # Add a device to the Client Box and verify.
        afe_instance = rf_switch_client_box.frontend.AFE()
        afe_instance.get_hosts.return_value = [self.client_box_host, dut_host]
        client_box = rf_switch_client_box.ClientBox(self.client_box_host)
        devices = client_box.get_devices()
        self.assertEqual(len(devices), 1)
        device = devices[0]
        self.assertEqual(device, 'chromeos9-device1')


    def testNoDevicesInClientbox(self):
        """Test for no devices in the Client Box."""
        rf_switch_client_box.frontend = mock.MagicMock()
        afe_instance =rf_switch_client_box.frontend.AFE()
        afe_instance.get_hosts.return_value = [self.client_box_host]
        client_box = rf_switch_client_box.ClientBox(self.client_box_host)
        devices = client_box.get_devices()
        self.assertEqual(len(devices), 0)


    def testGetOtherDevices(self):
        """Test to get stumpy from ClientBox if installed."""
        rf_switch_client_box.frontend = mock.MagicMock()
        stumpy_host = rf_switch_client_box.frontend.Host('', '')
        stumpy_host.hostname = 'chromeos9-stumpy1'
        stumpy_host.labels = ['rf_switch_1', 'client_box_1', 'stumpy']
        afe_instance = rf_switch_client_box.frontend.AFE()
        afe_instance.get_hosts.side_effect = [[stumpy_host]]
        client_box = rf_switch_client_box.ClientBox(self.client_box_host)
        self.assertEqual(
                client_box.get_devices_using_labels(stumpy_host.labels),
                ['chromeos9-stumpy1'])


    def testGetOtherDeviceWithWrongLabels(self):
        """Test to get Devices when using wrong list of labels."""
        rf_switch_client_box.frontend = mock.MagicMock()
        stumpy_host = rf_switch_client_box.frontend.Host('', '')
        stumpy_host.hostname = 'chromeos9-stumpy1'
        stumpy_host.labels = ['rf_switch_1', 'client_box_1', 'stumpy']
        list_of_labels = ['rf_switch_1', 'client_box_1', 'packet_capture']
        afe_instance = rf_switch_client_box.frontend.AFE()
        afe_instance.get_hosts.side_effect = [[self.client_box_host],
                                              [stumpy_host]]
        client_box = rf_switch_client_box.ClientBox(self.client_box_host)
        self.assertEqual(
                len(client_box.get_devices_using_labels(list_of_labels)), 0)


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