OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2014 The LUCI Authors. All rights reserved. | 2 # Copyright 2014 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 import hashlib | 6 import hashlib |
7 import sys | 7 import sys |
8 import unittest | 8 import unittest |
9 | 9 |
10 from test_support import test_env | 10 from test_support import test_env |
11 test_env.setup_test_env() | 11 test_env.setup_test_env() |
12 | 12 |
13 from google.appengine.ext import ndb | 13 from google.appengine.ext import ndb |
14 | 14 |
(...skipping 14 matching lines...) Expand all Loading... |
29 def test_hashed_shard_key(self): | 29 def test_hashed_shard_key(self): |
30 actual = sharding.hashed_shard_key('1234', 2, 'Root') | 30 actual = sharding.hashed_shard_key('1234', 2, 'Root') |
31 expected = "Key('Root', '%s')" % hashlib.md5('1234').hexdigest()[:2] | 31 expected = "Key('Root', '%s')" % hashlib.md5('1234').hexdigest()[:2] |
32 self.assertEqual(expected, str(actual)) | 32 self.assertEqual(expected, str(actual)) |
33 | 33 |
34 | 34 |
35 if __name__ == '__main__': | 35 if __name__ == '__main__': |
36 if '-v' in sys.argv: | 36 if '-v' in sys.argv: |
37 unittest.TestCase.maxDiff = None | 37 unittest.TestCase.maxDiff = None |
38 unittest.main() | 38 unittest.main() |
OLD | NEW |