1*9c5db199SXin Li#!/usr/bin/python3 2*9c5db199SXin Li# 3*9c5db199SXin Li# Copyright 2014 The Chromium OS Authors. All rights reserved. 4*9c5db199SXin Li# Use of this source code is governed by a BSD-style license that can be 5*9c5db199SXin Li# found in the LICENSE file. 6*9c5db199SXin Li 7*9c5db199SXin Liimport unittest 8*9c5db199SXin Li 9*9c5db199SXin Liimport common 10*9c5db199SXin Lifrom autotest_lib.server.cros import dnsname_mangler 11*9c5db199SXin Li 12*9c5db199SXin LiHOST = 'chromeos1-row1-rack1-host1' 13*9c5db199SXin LiROUTER = 'chromeos1-row1-rack1-host1-router' 14*9c5db199SXin LiATTENUATOR = 'chromeos1-row1-rack1-host1-attenuator' 15*9c5db199SXin Li 16*9c5db199SXin LiHOST_FROM_OUTSIDE_LAB = HOST + '.cros' 17*9c5db199SXin LiROUTER_FROM_OUTSIDE_LAB = ROUTER + '.cros' 18*9c5db199SXin LiATTENUATOR_FROM_OUTSIDE_LAB = ATTENUATOR + '.cros' 19*9c5db199SXin Li 20*9c5db199SXin Li 21*9c5db199SXin Liclass DnsnameMangerUnittest(unittest.TestCase): 22*9c5db199SXin Li """Check that we're correctly mangling DNS names.""" 23*9c5db199SXin Li 24*9c5db199SXin Li 25*9c5db199SXin Li def testRouterNamesCorrect(self): 26*9c5db199SXin Li """Router names should look like <dut_dns_name>-router[.cros]""" 27*9c5db199SXin Li self.assertEquals(ROUTER, dnsname_mangler.get_router_addr(HOST)) 28*9c5db199SXin Li self.assertEquals( 29*9c5db199SXin Li ROUTER_FROM_OUTSIDE_LAB, 30*9c5db199SXin Li dnsname_mangler.get_router_addr(HOST_FROM_OUTSIDE_LAB)) 31*9c5db199SXin Li 32*9c5db199SXin Li 33*9c5db199SXin Li def testAttenuatorNamesCorrect(self): 34*9c5db199SXin Li """Router names should look like <dut_dns_name>-attenuator[.cros]""" 35*9c5db199SXin Li self.assertEquals(ATTENUATOR, dnsname_mangler.get_attenuator_addr(HOST)) 36*9c5db199SXin Li self.assertEquals( 37*9c5db199SXin Li ATTENUATOR_FROM_OUTSIDE_LAB, 38*9c5db199SXin Li dnsname_mangler.get_attenuator_addr(HOST_FROM_OUTSIDE_LAB)) 39*9c5db199SXin Li 40*9c5db199SXin Li 41*9c5db199SXin Liif __name__ == '__main__': 42*9c5db199SXin Li unittest.main() 43