| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # coding: utf-8 | 2 # coding: utf-8 |
| 3 # Copyright 2015 The LUCI Authors. All rights reserved. | 3 # Copyright 2015 The LUCI Authors. All rights reserved. |
| 4 # Use of this source code is governed under the Apache License, Version 2.0 | 4 # Use of this source code is governed under the Apache License, Version 2.0 |
| 5 # that can be found in the LICENSE file. | 5 # that can be found in the LICENSE file. |
| 6 | 6 |
| 7 import base64 | 7 import base64 |
| 8 import datetime | 8 import datetime |
| 9 import logging | 9 import logging |
| 10 import os | 10 import os |
| (...skipping 275 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 286 u'os': u'Amiga', | 286 u'os': u'Amiga', |
| 287 u'pool': u'default', | 287 u'pool': u'default', |
| 288 }, | 288 }, |
| 289 u'env': {}, | 289 u'env': {}, |
| 290 u'extra_args': [], | 290 u'extra_args': [], |
| 291 u'grace_period': 30, | 291 u'grace_period': 30, |
| 292 u'hard_timeout': 3600, | 292 u'hard_timeout': 3600, |
| 293 u'host': u'http://localhost:8080', | 293 u'host': u'http://localhost:8080', |
| 294 u'isolated': None, | 294 u'isolated': None, |
| 295 u'io_timeout': 1200, | 295 u'io_timeout': 1200, |
| 296 u'outputs': [u'foo', u'path/to/foobar'], |
| 296 u'service_account': u'none', | 297 u'service_account': u'none', |
| 297 u'task_id': task_id, | 298 u'task_id': task_id, |
| 298 }, | 299 }, |
| 299 } | 300 } |
| 300 self.assertEqual(expected, response) | 301 self.assertEqual(expected, response) |
| 301 response = self.client_get_results(task_id) | 302 response = self.client_get_results(task_id) |
| 302 expected = { | 303 expected = { |
| 303 u'bot_dimensions': [ | 304 u'bot_dimensions': [ |
| 304 {u'key': u'id', u'value': [u'bot1']}, | 305 {u'key': u'id', u'value': [u'bot1']}, |
| 305 {u'key': u'os', u'value': [u'Amiga']}, | 306 {u'key': u'os', u'value': [u'Amiga']}, |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 351 u'os': u'Amiga', | 352 u'os': u'Amiga', |
| 352 u'pool': u'default', | 353 u'pool': u'default', |
| 353 }, | 354 }, |
| 354 u'env': {}, | 355 u'env': {}, |
| 355 u'extra_args': [], | 356 u'extra_args': [], |
| 356 u'grace_period': 30, | 357 u'grace_period': 30, |
| 357 u'hard_timeout': 3600, | 358 u'hard_timeout': 3600, |
| 358 u'host': u'http://localhost:8080', | 359 u'host': u'http://localhost:8080', |
| 359 u'isolated': None, | 360 u'isolated': None, |
| 360 u'io_timeout': 1200, | 361 u'io_timeout': 1200, |
| 362 u'outputs': [u'foo', u'path/to/foobar'], |
| 361 u'service_account': u'bot', | 363 u'service_account': u'bot', |
| 362 u'task_id': task_id, | 364 u'task_id': task_id, |
| 363 }, | 365 }, |
| 364 } | 366 } |
| 365 self.assertEqual(expected, response) | 367 self.assertEqual(expected, response) |
| 366 | 368 |
| 367 def test_poll_confliciting_dimensions(self): | 369 def test_poll_confliciting_dimensions(self): |
| 368 params = self.do_handshake() | 370 params = self.do_handshake() |
| 369 self.assertEqual(params['dimensions']['pool'], ['default']) | 371 self.assertEqual(params['dimensions']['pool'], ['default']) |
| 370 | 372 |
| (...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 440 u'extra_args': [], | 442 u'extra_args': [], |
| 441 u'hard_timeout': 3600, | 443 u'hard_timeout': 3600, |
| 442 u'grace_period': 30, | 444 u'grace_period': 30, |
| 443 u'host': u'http://localhost:8080', | 445 u'host': u'http://localhost:8080', |
| 444 u'isolated': { | 446 u'isolated': { |
| 445 u'input': u'0123456789012345678901234567890123456789', | 447 u'input': u'0123456789012345678901234567890123456789', |
| 446 u'server': u'http://localhost:1', | 448 u'server': u'http://localhost:1', |
| 447 u'namespace': u'default-gzip', | 449 u'namespace': u'default-gzip', |
| 448 }, | 450 }, |
| 449 u'io_timeout': 1200, | 451 u'io_timeout': 1200, |
| 452 u'outputs': [u'foo', u'path/to/foobar'], |
| 450 u'service_account': u'none', | 453 u'service_account': u'none', |
| 451 u'task_id': task_id, | 454 u'task_id': task_id, |
| 452 }, | 455 }, |
| 453 } | 456 } |
| 454 self.assertEqual(expected, response) | 457 self.assertEqual(expected, response) |
| 455 | 458 |
| 456 # Complete the isolated task. | 459 # Complete the isolated task. |
| 457 params = { | 460 params = { |
| 458 'cost_usd': 0.1, | 461 'cost_usd': 0.1, |
| 459 'duration': 3., | 462 'duration': 3., |
| (...skipping 132 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 592 u'extra_args': [], | 595 u'extra_args': [], |
| 593 u'hard_timeout': 3600, | 596 u'hard_timeout': 3600, |
| 594 u'grace_period': 30, | 597 u'grace_period': 30, |
| 595 u'host': u'http://localhost:8080', | 598 u'host': u'http://localhost:8080', |
| 596 u'isolated': { | 599 u'isolated': { |
| 597 u'input': u'0123456789012345678901234567890123456789', | 600 u'input': u'0123456789012345678901234567890123456789', |
| 598 u'server': u'http://localhost:1', | 601 u'server': u'http://localhost:1', |
| 599 u'namespace': u'default-gzip', | 602 u'namespace': u'default-gzip', |
| 600 }, | 603 }, |
| 601 u'io_timeout': 1200, | 604 u'io_timeout': 1200, |
| 605 u'outputs': [u'foo', u'path/to/foobar'], |
| 602 u'service_account': u'none', | 606 u'service_account': u'none', |
| 603 u'task_id': task_id, | 607 u'task_id': task_id, |
| 604 }, | 608 }, |
| 605 } | 609 } |
| 606 self.assertEqual(expected, response) | 610 self.assertEqual(expected, response) |
| 607 | 611 |
| 608 def test_bot_ereporter2_error(self): | 612 def test_bot_ereporter2_error(self): |
| 609 # ereporter2's //client/utils/on_error.py traps unhandled exceptions | 613 # ereporter2's //client/utils/on_error.py traps unhandled exceptions |
| 610 # automatically. | 614 # automatically. |
| 611 self.mock(random, 'getrandbits', lambda _: 0x88) | 615 self.mock(random, 'getrandbits', lambda _: 0x88) |
| (...skipping 430 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1042 self.app.get('/bot_code?tok=%s' % tok, status=200) | 1046 self.app.get('/bot_code?tok=%s' % tok, status=200) |
| 1043 | 1047 |
| 1044 | 1048 |
| 1045 if __name__ == '__main__': | 1049 if __name__ == '__main__': |
| 1046 if '-v' in sys.argv: | 1050 if '-v' in sys.argv: |
| 1047 unittest.TestCase.maxDiff = None | 1051 unittest.TestCase.maxDiff = None |
| 1048 logging.basicConfig( | 1052 logging.basicConfig( |
| 1049 level=logging.DEBUG if '-v' in sys.argv else logging.CRITICAL, | 1053 level=logging.DEBUG if '-v' in sys.argv else logging.CRITICAL, |
| 1050 format='%(levelname)-7s %(filename)s:%(lineno)3d %(message)s') | 1054 format='%(levelname)-7s %(filename)s:%(lineno)3d %(message)s') |
| 1051 unittest.main() | 1055 unittest.main() |
| OLD | NEW |