server/cros/wifi_test_utils_unittest.py

#!/usr/bin/python
#
# Copyright (c) 2013 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/wifi_test_utils.py."""

import unittest

from autotest_lib.server.cros import wifi_test_utils


class WiFiTestUtilsSimpleTest(unittest.TestCase):
    """Tests various methods in wifi_test_utils.py that require no mox."""


    HOST = 'chromeos1-row1-rack1-host1'


    def testGetMachineDomain_withHostnameNoDot(self):
        """Tests parsing of hostname with no embedded dot."""
        machine, domain = wifi_test_utils._get_machine_domain(self.HOST)
        self.assertEquals(self.HOST, machine)
        self.assertEquals('', domain)


    def testGetMachineDomain_withHostnameAndDot(self):
        """Tests parsing of hostname with embedded dot."""
        hostname = '.'.join([self.HOST, 'cros'])
        machine, domain = wifi_test_utils._get_machine_domain(hostname)
        self.assertEquals(self.HOST, machine)
        self.assertEquals('.cros', domain)


    def testGetServerAddrInLab_withHostNameNoDot(self):
        """Tests return of server hostname with no embedded dot."""
        expected = ''.join([self.HOST, '-router'])
        actual = wifi_test_utils.get_server_addr_in_lab(self.HOST)
        self.assertEquals(expected, actual)


    def testGetServerAddrInLab_withHostNameAndDot(self):
        """Tests return of server hostname with embedded dot."""
        hostname = '.'.join([self.HOST, 'cros'])
        expected = ''.join([self.HOST, '-router', '.cros'])
        actual = wifi_test_utils.get_server_addr_in_lab(hostname)
        self.assertEquals(expected, actual)


    def testGetRouterAddrInLab_withHostNameNoDot(self):
        """Tests return of router hostname with no embedded dot."""
        expected = ''.join([self.HOST, '-router'])
        actual = wifi_test_utils.get_router_addr_in_lab(self.HOST)
        self.assertEquals(expected, actual)


    def testGetRouterAddrInLab_withHostNameAndDot(self):
        """Tests return of router hostname with embedded dot."""
        hostname = '.'.join([self.HOST, 'cros'])
        expected = ''.join([self.HOST, '-router', '.cros'])
        actual = wifi_test_utils.get_router_addr_in_lab(hostname)
        self.assertEquals(expected, actual)


    def testGetAttenuatorAddrInLab_withHostNameNoDot(self):
        """Tests return of attenuator hostname with no embedded dot."""
        expected = ''.join([self.HOST, '-attenuator'])
        actual = wifi_test_utils.get_attenuator_addr_in_lab(self.HOST)
        self.assertEquals(expected, actual)


    def testGetAttenuatorAddrInLab_withHostNameAndDot(self):
        """Tests return of attenuator hostname with embedded dot."""
        hostname = '.'.join([self.HOST, 'cros'])
        expected = ''.join([self.HOST, '-attenuator', '.cros'])
        actual = wifi_test_utils.get_attenuator_addr_in_lab(hostname)
        self.assertEquals(expected, actual)