1#!/usr/bin/python2 2# 3# Copyright 2015 The Chromium OS Authors. All rights reserved. 4# Use of this source code is governed by a BSD-style license that can be 5# found in the LICENSE file. 6 7import unittest 8 9import common 10from autotest_lib.server.hosts import servo_host 11 12 13class ServoAfeBoardMapTest(unittest.TestCase): 14 """Tests for how we map AFE boards to servo boards.""" 15 16 def test_afe_board_mapping(self): 17 """Tests mappings.""" 18 afe_map = servo_host._map_afe_board_to_servo_board 19 self.assertEqual(afe_map('kip'), 'kip') 20 self.assertEqual(afe_map('gizmo'), 'panther') 21 self.assertEqual(afe_map('link_freon'), 'link') 22 self.assertEqual(afe_map('nyan_blaze-freon'), 'nyan_blaze') 23 self.assertEqual(afe_map('stumpy_moblab'), 'stumpy') 24 self.assertEqual(afe_map('veyron_minnie-cheets'), 'veyron_minnie') 25 26 27if __name__ == '__main__': 28 unittest.main() 29