OLD | NEW |
1 #!/usr/bin/python | 1 #!/usr/bin/python |
2 # Copyright 2015 The LUCI Authors. All rights reserved. | 2 # Copyright 2015 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 under the Apache License, Version 2.0 |
4 # found in the LICENSE file. | 4 # that can be found in the LICENSE file. |
5 | 5 |
6 """Unit tests for lease_management.py.""" | 6 """Unit tests for lease_management.py.""" |
7 | 7 |
8 import datetime | 8 import datetime |
9 import json | 9 import json |
10 import unittest | 10 import unittest |
11 | 11 |
12 import test_env | 12 import test_env |
13 test_env.setup_test_env() | 13 test_env.setup_test_env() |
14 | 14 |
(...skipping 533 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
548 updated_machine_type = machine_type.key.get() | 548 updated_machine_type = machine_type.key.get() |
549 self.assertEqual(len(updated_machine_type.leases), 2) | 549 self.assertEqual(len(updated_machine_type.leases), 2) |
550 request_ids = sorted( | 550 request_ids = sorted( |
551 request.client_request_id for request in updated_machine_type.leases) | 551 request.client_request_id for request in updated_machine_type.leases) |
552 self.assertEqual(request_ids[0], 'fake-id-1') | 552 self.assertEqual(request_ids[0], 'fake-id-1') |
553 self.assertEqual(request_ids[1], 'fake-id-3') | 553 self.assertEqual(request_ids[1], 'fake-id-3') |
554 | 554 |
555 | 555 |
556 if __name__ == '__main__': | 556 if __name__ == '__main__': |
557 unittest.main() | 557 unittest.main() |
OLD | NEW |