Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(514)

Side by Side Diff: appengine/findit/model/test/versioned_model_test.py

Issue 2376263002: [Findit] Renaming version property in versioned_model.py to avoid conflicts with base_analysis.py (Closed)
Patch Set: Created 4 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
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)
OLDNEW
« no previous file with comments | « appengine/findit/model/test/versioned_config_test.py ('k') | appengine/findit/model/versioned_model.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698