| OLD | NEW |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 # pylint: disable=F0401 | 5 # pylint: disable=F0401 |
| 6 | 6 |
| 7 import unittest | 7 import unittest |
| 8 | 8 |
| 9 from infra.tools.bot_setup.version import version | 9 from infra.tools.bot_setup.version import version |
| 10 | 10 |
| 11 | 11 |
| 12 ENABLED_SLAVES = [ | 12 ENABLED_SLAVES = [ |
| 13 'slave2-c7', | 13 'slave2-c7', |
| 14 'slave12-c7', | 14 'slave12-c7', |
| 15 'slave100-c4', | 15 'slave100-c4', |
| 16 'slave250-c4', | 16 'slave250-c4', |
| 17 'slave260-c4', | 17 'slave260-c4', |
| 18 'swarm1-c4', | 18 'swarm1-c4', |
| 19 'swarm11-c4', | 19 'swarm11-c4', |
| 20 'winslave1-c4', |
| 21 'swarm-win1-c4', |
| 20 ] | 22 ] |
| 21 | 23 |
| 22 DISABLED_SLAVES = [ | 24 DISABLED_SLAVES = [ |
| 23 'test_disabled_slave', | 25 'test_disabled_slave', |
| 24 ] | 26 ] |
| 25 | 27 |
| 26 | 28 |
| 27 class TestVersions(unittest.TestCase): | 29 class TestVersions(unittest.TestCase): |
| 28 @staticmethod | 30 @staticmethod |
| 29 def test_expected_active_versions(): | 31 def test_expected_active_versions(): |
| 30 return [ | 32 return [ |
| 31 'Slave %s is on version %s' | 33 'Slave %s is on version %s' |
| 32 % (slave_name, version.get_version(slave_name, None)) | 34 % (slave_name, version.get_version(slave_name, None)) |
| 33 for slave_name in sorted(ENABLED_SLAVES) | 35 for slave_name in sorted(ENABLED_SLAVES) |
| 34 ] | 36 ] |
| 35 | 37 |
| 36 def test_expected_disabled_versions(self): | 38 def test_expected_disabled_versions(self): |
| 37 for slave_name in sorted(DISABLED_SLAVES): | 39 for slave_name in sorted(DISABLED_SLAVES): |
| 38 self.assertRaises( | 40 self.assertRaises( |
| 39 version.BuilderDisabled, version.get_version, slave_name, None) | 41 version.BuilderDisabled, version.get_version, slave_name, None) |
| OLD | NEW |