| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright 2013 The LUCI Authors. All rights reserved. | 2 # Copyright 2013 The LUCI Authors. All rights reserved. |
| 3 # Use of this source code is governed by the Apache v2.0 license that can be | 3 # Use of this source code is governed by the Apache v2.0 license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 import json | 6 import json |
| 7 import logging | 7 import logging |
| 8 import os | 8 import os |
| 9 import shutil | 9 import shutil |
| 10 import sys | 10 import sys |
| (...skipping 80 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 91 self.assertEqual(expected, dimensions) | 91 self.assertEqual(expected, dimensions) |
| 92 | 92 |
| 93 def test_get_dimensions_load_test(self): | 93 def test_get_dimensions_load_test(self): |
| 94 os.environ['SWARMING_LOAD_TEST'] = '1' | 94 os.environ['SWARMING_LOAD_TEST'] = '1' |
| 95 self.assertEqual(['id', 'load_test'], sorted(bot_main.get_dimensions(None))) | 95 self.assertEqual(['id', 'load_test'], sorted(bot_main.get_dimensions(None))) |
| 96 | 96 |
| 97 def test_generate_version(self): | 97 def test_generate_version(self): |
| 98 self.assertEqual('123', bot_main.generate_version()) | 98 self.assertEqual('123', bot_main.generate_version()) |
| 99 | 99 |
| 100 def test_get_state(self): | 100 def test_get_state(self): |
| 101 self.mock(time, 'time', lambda: 1460000000.0) | 101 self.mock(time, 'time', lambda: 1470000000.0) |
| 102 expected = os_utilities.get_state() | 102 expected = os_utilities.get_state() |
| 103 expected['sleep_streak'] = 12 | 103 expected['sleep_streak'] = 12 |
| 104 # During the execution of this test case, the free disk space could have | 104 # During the execution of this test case, the free disk space could have |
| 105 # changed. | 105 # changed. |
| 106 for disk in expected['disks'].itervalues(): | 106 for disk in expected['disks'].itervalues(): |
| 107 self.assertGreater(disk.pop('free_mb'), 1.) | 107 self.assertGreater(disk.pop('free_mb'), 1.) |
| 108 actual = bot_main.get_state(None, 12) | 108 actual = bot_main.get_state(None, 12) |
| 109 for disk in actual['disks'].itervalues(): | 109 for disk in actual['disks'].itervalues(): |
| 110 self.assertGreater(disk.pop('free_mb'), 1.) | 110 self.assertGreater(disk.pop('free_mb'), 1.) |
| 111 self.assertGreater(actual.pop('nb_files_in_temp'), 0) | 111 self.assertGreater(actual.pop('nb_files_in_temp'), 0) |
| (...skipping 431 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 543 self.assertEqual(0, bot_main.main([])) | 543 self.assertEqual(0, bot_main.main([])) |
| 544 | 544 |
| 545 | 545 |
| 546 if __name__ == '__main__': | 546 if __name__ == '__main__': |
| 547 fix_encoding.fix_encoding() | 547 fix_encoding.fix_encoding() |
| 548 if '-v' in sys.argv: | 548 if '-v' in sys.argv: |
| 549 TestBotMain.maxDiff = None | 549 TestBotMain.maxDiff = None |
| 550 logging.basicConfig( | 550 logging.basicConfig( |
| 551 level=logging.DEBUG if '-v' in sys.argv else logging.CRITICAL) | 551 level=logging.DEBUG if '-v' in sys.argv else logging.CRITICAL) |
| 552 unittest.main() | 552 unittest.main() |
| OLD | NEW |