OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2014 The Swarming Authors. All rights reserved. | 2 # Copyright 2014 The Swarming 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 logging | 6 import logging |
7 import sys | 7 import sys |
8 import unittest | 8 import unittest |
9 | 9 |
10 from test_support import test_env | 10 import test_env |
11 test_env.setup_test_env() | 11 test_env.setup_test_env() |
12 | 12 |
13 from google.appengine.api import datastore_errors | 13 from google.appengine.api import datastore_errors |
14 from google.appengine.ext import ndb | 14 from google.appengine.ext import ndb |
15 | 15 |
16 from components.datastore_utils import txn | 16 from components.datastore_utils import txn |
17 from test_support import test_case | 17 from test_support import test_case |
18 | 18 |
19 | 19 |
20 class EntityX(ndb.Model): | 20 class EntityX(ndb.Model): |
(...skipping 80 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
101 self.assertEqual(2, future.get_result()) | 101 self.assertEqual(2, future.get_result()) |
102 self.assertEqual(1, EntityX.query().count()) | 102 self.assertEqual(1, EntityX.query().count()) |
103 | 103 |
104 | 104 |
105 if __name__ == '__main__': | 105 if __name__ == '__main__': |
106 if '-v' in sys.argv: | 106 if '-v' in sys.argv: |
107 unittest.TestCase.maxDiff = None | 107 unittest.TestCase.maxDiff = None |
108 logging.basicConfig( | 108 logging.basicConfig( |
109 level=logging.DEBUG if '-v' in sys.argv else logging.ERROR) | 109 level=logging.DEBUG if '-v' in sys.argv else logging.ERROR) |
110 unittest.main() | 110 unittest.main() |
OLD | NEW |