| OLD | NEW |
| 1 # Copyright 2015 The Chromium Authors. All rights reserved. | 1 # Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 from google.appengine.api import datastore_errors | 5 from google.appengine.api import datastore_errors |
| 6 from google.appengine.ext import ndb | 6 from google.appengine.ext import ndb |
| 7 | 7 |
| 8 from testing_utils import testing | 8 from testing_utils import testing |
| 9 | 9 |
| 10 from model.versioned_model import VersionedModel | 10 from model.versioned_model import VersionedModel |
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 52 self.assertIsNone(_Entity.GetVersion(0)) | 52 self.assertIsNone(_Entity.GetVersion(0)) |
| 53 self.assertIsNone(_Entity.GetVersion(3)) | 53 self.assertIsNone(_Entity.GetVersion(3)) |
| 54 | 54 |
| 55 def testGetLatestVersionNumber(self): | 55 def testGetLatestVersionNumber(self): |
| 56 root_key = ndb.Key('_EntityRoot', 1) | 56 root_key = ndb.Key('_EntityRoot', 1) |
| 57 _Entity._GetRootModel()(key=root_key, current=1).put() | 57 _Entity._GetRootModel()(key=root_key, current=1).put() |
| 58 _Entity(key=ndb.Key('_Entity', 1, parent=root_key), value=2).put() | 58 _Entity(key=ndb.Key('_Entity', 1, parent=root_key), value=2).put() |
| 59 | 59 |
| 60 self.assertEqual(1, _Entity.GetLatestVersionNumber()) | 60 self.assertEqual(1, _Entity.GetLatestVersionNumber()) |
| 61 | 61 |
| 62 def testGetLatestVersionNumberWhenNoRecordYet(self): |
| 63 self.assertEqual(-1, _Entity.GetLatestVersionNumber()) |
| 64 |
| 62 def testSaveNewVersion(self): | 65 def testSaveNewVersion(self): |
| 63 entity = _Entity() | 66 entity = _Entity() |
| 64 entity.value = 1 | 67 entity.value = 1 |
| 65 key = entity.Save() | 68 key = entity.Save() |
| 66 | 69 |
| 67 expected_key = ndb.Key('_EntityRoot', 1, '_Entity', 1) | 70 expected_key = ndb.Key('_EntityRoot', 1, '_Entity', 1) |
| 68 self.assertEqual(expected_key, key) | 71 self.assertEqual(expected_key, key) |
| 69 | 72 |
| 70 entity = _Entity.GetVersion() | 73 entity = _Entity.GetVersion() |
| 71 self.assertEqual(1, entity.version) | 74 self.assertEqual(1, entity.version) |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 111 calls.append(1) | 114 calls.append(1) |
| 112 raise datastore_errors.BadRequestError() | 115 raise datastore_errors.BadRequestError() |
| 113 return original_ndb_transaction(func, **options) | 116 return original_ndb_transaction(func, **options) |
| 114 self.mock(ndb, 'transaction', MockNdbTransaction) | 117 self.mock(ndb, 'transaction', MockNdbTransaction) |
| 115 | 118 |
| 116 entity = _Entity() | 119 entity = _Entity() |
| 117 key = entity.Save() | 120 key = entity.Save() |
| 118 expected_key = ndb.Key('_EntityRoot', 1, '_Entity', 1) | 121 expected_key = ndb.Key('_EntityRoot', 1, '_Entity', 1) |
| 119 self.assertEqual(expected_key, key) | 122 self.assertEqual(expected_key, key) |
| 120 self.assertEqual([1], calls) | 123 self.assertEqual([1], calls) |
| OLD | NEW |