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

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

Issue 1866883002: [Findit] A huge refactoring and some bug fixing. (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@master
Patch Set: Fix nit. Created 4 years, 8 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
(...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after
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
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)
OLDNEW
« no previous file with comments | « appengine/findit/model/result_status.py ('k') | appengine/findit/model/test/wf_analysis_test.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698