| 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 |
| 11 | 11 |
| 12 | 12 |
| 13 class _Entity(VersionedModel): | 13 class _Entity(VersionedModel): |
| 14 value = ndb.IntegerProperty(indexed=False) | 14 value = ndb.IntegerProperty(indexed=False) |
| 15 | 15 |
| 16 | 16 |
| 17 class VersionedModelTest(testing.AppengineTestCase): | 17 class VersionedModelTest(testing.AppengineTestCase): |
| 18 | 18 |
| 19 def testGetRootModel(self): | 19 def testGetRootModel(self): |
| 20 root_model_class = _Entity._GetRootModel() | 20 root_model_class = _Entity._GetRootModel() |
| 21 self.assertEqual('_EntityRoot', root_model_class._get_kind()) | 21 self.assertEqual('_EntityRoot', root_model_class._get_kind()) |
| 22 self.assertTrue(issubclass(root_model_class, ndb.Model)) | 22 self.assertTrue(issubclass(root_model_class, ndb.Model)) |
| 23 self.assertEqual(3, root_model_class(current=3).current) | 23 self.assertEqual(3, root_model_class(current=3).current) |
| 24 | 24 |
| 25 def testDefaultVersionIsZero(self): | 25 def testDefaultVersionIsZero(self): |
| 26 self.assertEqual(0, _Entity.Create().version) | 26 self.assertEqual(0, _Entity.Create().version_number) |
| 27 self.assertEqual(0, _Entity.Create('m1').version) | 27 self.assertEqual(0, _Entity.Create('m1').version_number) |
| 28 | 28 |
| 29 def testRootId(self): | 29 def testRootId(self): |
| 30 self.assertEqual(1, _Entity.Create()._root_id) | 30 self.assertEqual(1, _Entity.Create()._root_id) |
| 31 self.assertEqual( | 31 self.assertEqual( |
| 32 1, _Entity(key=ndb.Key('_EntityRoot', 1, '_Entity', 2))._root_id) | 32 1, _Entity(key=ndb.Key('_EntityRoot', 1, '_Entity', 2))._root_id) |
| 33 self.assertEqual('m1', _Entity.Create('m1')._root_id) | 33 self.assertEqual('m1', _Entity.Create('m1')._root_id) |
| 34 | 34 |
| 35 def testCreate(self): | 35 def testCreate(self): |
| 36 self.assertEqual(_Entity(key=ndb.Key('_EntityRoot', 'm1', '_Entity', 0)), | 36 self.assertEqual(_Entity(key=ndb.Key('_EntityRoot', 'm1', '_Entity', 0)), |
| 37 _Entity.Create('m1')) | 37 _Entity.Create('m1')) |
| 38 | 38 |
| 39 def testGetMostRecentVersionWhenNoData(self): | 39 def testGetMostRecentVersionWhenNoData(self): |
| 40 self.assertIsNone(_Entity.GetVersion()) | 40 self.assertIsNone(_Entity.GetVersion()) |
| 41 self.assertIsNone(_Entity.GetVersion('m1')) | 41 self.assertIsNone(_Entity.GetVersion('m1')) |
| 42 | 42 |
| 43 def testGetMostRecentVersionWhenDataExists(self): | 43 def testGetMostRecentVersionWhenDataExists(self): |
| 44 entity = _Entity.Create() | 44 entity = _Entity.Create() |
| 45 entity.value = 1 | 45 entity.value = 1 |
| 46 entity.Save() | 46 entity.Save() |
| 47 entity.value = 2 | 47 entity.value = 2 |
| 48 entity.Save() | 48 entity.Save() |
| 49 | 49 |
| 50 entity = _Entity.GetVersion() | 50 entity = _Entity.GetVersion() |
| 51 self.assertEqual(2, entity.version) | 51 self.assertEqual(2, entity.version_number) |
| 52 self.assertEqual(2, entity.value) | 52 self.assertEqual(2, entity.value) |
| 53 | 53 |
| 54 def testGetSpecificVersion(self): | 54 def testGetSpecificVersion(self): |
| 55 entity = _Entity.Create() | 55 entity = _Entity.Create() |
| 56 entity.value = 1 | 56 entity.value = 1 |
| 57 entity.Save() | 57 entity.Save() |
| 58 entity.value = 2 | 58 entity.value = 2 |
| 59 entity.Save() | 59 entity.Save() |
| 60 | 60 |
| 61 entity_version_2 = _Entity.GetVersion(version=2) | 61 entity_version_2 = _Entity.GetVersion(version=2) |
| 62 self.assertEqual(2, entity_version_2.version) | 62 self.assertEqual(2, entity_version_2.version_number) |
| 63 self.assertEqual(2, entity_version_2.value) | 63 self.assertEqual(2, entity_version_2.value) |
| 64 self.assertIsNone(_Entity.GetVersion(version=0)) | 64 self.assertIsNone(_Entity.GetVersion(version=0)) |
| 65 self.assertIsNone(_Entity.GetVersion(version=3)) | 65 self.assertIsNone(_Entity.GetVersion(version=3)) |
| 66 | 66 |
| 67 def testGetLatestVersionNumber(self): | 67 def testGetLatestVersionNumber(self): |
| 68 entity = _Entity.Create() | 68 entity = _Entity.Create() |
| 69 entity.Save() | 69 entity.Save() |
| 70 self.assertEqual(1, _Entity.GetLatestVersionNumber()) | 70 self.assertEqual(1, _Entity.GetLatestVersionNumber()) |
| 71 | 71 |
| 72 def testGetVersionForEntityWithKeyId(self): | 72 def testGetVersionForEntityWithKeyId(self): |
| 73 entity = _Entity.Create('m1') | 73 entity = _Entity.Create('m1') |
| 74 entity.value = 1 | 74 entity.value = 1 |
| 75 entity.Save() | 75 entity.Save() |
| 76 entity.value = 2 | 76 entity.value = 2 |
| 77 entity.Save() | 77 entity.Save() |
| 78 | 78 |
| 79 self.assertEqual(2, _Entity.GetVersion('m1').version) | 79 self.assertEqual(2, _Entity.GetVersion('m1').version_number) |
| 80 self.assertEqual(2, _Entity.GetVersion('m1').value) | 80 self.assertEqual(2, _Entity.GetVersion('m1').value) |
| 81 self.assertIsNone(_Entity.GetVersion('m1', version=0)) | 81 self.assertIsNone(_Entity.GetVersion('m1', version=0)) |
| 82 self.assertIsNone(_Entity.GetVersion('m1', version=3)) | 82 self.assertIsNone(_Entity.GetVersion('m1', version=3)) |
| 83 | 83 |
| 84 def testGetLatestVersionNumberWhenNoRecordYet(self): | 84 def testGetLatestVersionNumberWhenNoRecordYet(self): |
| 85 self.assertEqual(-1, _Entity.GetLatestVersionNumber()) | 85 self.assertEqual(-1, _Entity.GetLatestVersionNumber()) |
| 86 self.assertEqual(-1, _Entity.GetLatestVersionNumber('m1')) | 86 self.assertEqual(-1, _Entity.GetLatestVersionNumber('m1')) |
| 87 | 87 |
| 88 def testSaveNewVersion(self): | 88 def testSaveNewVersion(self): |
| 89 entity = _Entity.Create() | 89 entity = _Entity.Create() |
| 90 entity.value = 1 | 90 entity.value = 1 |
| 91 key, saved = entity.Save() | 91 key, saved = entity.Save() |
| 92 | 92 |
| 93 expected_key = ndb.Key('_EntityRoot', 1, '_Entity', 1) | 93 expected_key = ndb.Key('_EntityRoot', 1, '_Entity', 1) |
| 94 self.assertEqual(expected_key, key) | 94 self.assertEqual(expected_key, key) |
| 95 self.assertTrue(saved) | 95 self.assertTrue(saved) |
| 96 | 96 |
| 97 entity = _Entity.GetVersion() | 97 entity = _Entity.GetVersion() |
| 98 self.assertEqual(1, entity.version) | 98 self.assertEqual(1, entity.version_number) |
| 99 self.assertEqual(1, entity.value) | 99 self.assertEqual(1, entity.value) |
| 100 | 100 |
| 101 def testSaveEntityWithRootId(self): | 101 def testSaveEntityWithRootId(self): |
| 102 entity = _Entity.Create('m1') | 102 entity = _Entity.Create('m1') |
| 103 entity.value = 1 | 103 entity.value = 1 |
| 104 entity.Save() | 104 entity.Save() |
| 105 entity.value = 3 | 105 entity.value = 3 |
| 106 entity.Save() | 106 entity.Save() |
| 107 | 107 |
| 108 self.assertEqual(entity.version, 2) | 108 self.assertEqual(entity.version_number, 2) |
| 109 self.assertEqual(entity.value, 3) | 109 self.assertEqual(entity.value, 3) |
| 110 | 110 |
| 111 def testSaveNewVersionOfEntityWithKeyId(self): | 111 def testSaveNewVersionOfEntityWithKeyId(self): |
| 112 entity = _Entity.Create('m1') | 112 entity = _Entity.Create('m1') |
| 113 entity.value = 1 | 113 entity.value = 1 |
| 114 key, saved = entity.Save() | 114 key, saved = entity.Save() |
| 115 expected_key = ndb.Key('_EntityRoot', 'm1', '_Entity', 1) | 115 expected_key = ndb.Key('_EntityRoot', 'm1', '_Entity', 1) |
| 116 self.assertEqual(expected_key, key) | 116 self.assertEqual(expected_key, key) |
| 117 self.assertTrue(saved) | 117 self.assertTrue(saved) |
| 118 | 118 |
| (...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 191 raise datastore_errors.BadRequestError() | 191 raise datastore_errors.BadRequestError() |
| 192 return original_ndb_transaction(func, **options) | 192 return original_ndb_transaction(func, **options) |
| 193 self.mock(ndb, 'transaction', MockNdbTransaction) | 193 self.mock(ndb, 'transaction', MockNdbTransaction) |
| 194 | 194 |
| 195 entity = _Entity.Create() | 195 entity = _Entity.Create() |
| 196 key, saved = entity.Save() | 196 key, saved = entity.Save() |
| 197 expected_key = ndb.Key('_EntityRoot', 1, '_Entity', 1) | 197 expected_key = ndb.Key('_EntityRoot', 1, '_Entity', 1) |
| 198 self.assertEqual(expected_key, key) | 198 self.assertEqual(expected_key, key) |
| 199 self.assertEqual([1], calls) | 199 self.assertEqual([1], calls) |
| 200 self.assertTrue(saved) | 200 self.assertTrue(saved) |
| OLD | NEW |