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

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

Issue 1836293002: [Findit] Adding central config test class for unit tests (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@master
Patch Set: Addressing comments 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.ext import ndb 5 from google.appengine.ext import ndb
6 6 from google.appengine.api import users
7 from testing_utils import testing 7 from testing_utils import testing
8 8
9 from model.versioned_config import VersionedConfig 9 from model.versioned_config import VersionedConfig
10 10
11 11
12 class _Config(VersionedConfig): 12 class _Config(VersionedConfig):
13 a = ndb.IntegerProperty(indexed=False, default=0) 13 a = ndb.IntegerProperty(indexed=False, default=0)
14 14
15 15
16 class VersionedConfigTest(testing.AppengineTestCase): 16 class VersionedConfigTest(testing.AppengineTestCase):
17 def _CreateFirstVersion(self): 17 def _CreateFirstVersion(self):
18 config = _Config.Get() 18 config = _Config.Get()
19 config.Update(a=1) 19 config.Update(user=users.User(email='admin@chromium.org'),
20 is_admin=True, a=1)
20 21
21 def testGetWhenNoConfigCreatedYet(self): 22 def testGetWhenNoConfigCreatedYet(self):
22 config = _Config.Get() 23 config = _Config.Get()
23 self.assertIsNotNone(config) 24 self.assertIsNotNone(config)
24 self.assertEqual(0, config.a) 25 self.assertEqual(0, config.a)
25 26
26 def testNonAdminCanNotUpdate(self): 27 def testNonAdminCanNotUpdate(self):
27 config = _Config.Get() 28 config = _Config.Get()
28 with self.assertRaises(Exception): 29 with self.assertRaises(Exception):
29 config.Update() 30 config.Update(user=users.User(email='admin@chromium.org'),
31 is_admin=False, a=1)
30 32
31 def testUpdateWhenChanged(self): 33 def testUpdateWhenChanged(self):
32 self.mock_current_user(user_email='test@chromium.org', is_admin=True)
33 self._CreateFirstVersion() 34 self._CreateFirstVersion()
34 config = _Config.Get() 35 config = _Config.Get()
35 self.assertIsNotNone(config) 36 self.assertIsNotNone(config)
36 self.assertTrue(config.Update(a=2)) 37 self.assertTrue(config.Update(
38 user=users.User(email='admin@chromium.org'), is_admin=True,
39 a=2))
37 40
38 config = _Config.Get() 41 config = _Config.Get()
39 self.assertIsNotNone(config) 42 self.assertIsNotNone(config)
40 self.assertEqual(2, config.version) 43 self.assertEqual(2, config.version)
41 self.assertEqual(2, config.a) 44 self.assertEqual(2, config.a)
42 45
43 def testNotUpdateWhenNotChanged(self): 46 def testNotUpdateWhenNotChanged(self):
44 self.mock_current_user(user_email='test@chromium.org', is_admin=True)
45 self._CreateFirstVersion() 47 self._CreateFirstVersion()
46 config = _Config.Get() 48 config = _Config.Get()
47 self.assertIsNotNone(config) 49 self.assertIsNotNone(config)
48 self.assertFalse(config.Update(a=1)) 50 self.assertFalse(config.Update(
51 user=users.User(email='admin@chromium.org'), is_admin=True,
52 a=1))
49 53
50 config = _Config.Get() 54 config = _Config.Get()
51 self.assertIsNotNone(config) 55 self.assertIsNotNone(config)
52 self.assertEqual(1, config.version) 56 self.assertEqual(1, config.version)
53 self.assertEqual(1, config.a) 57 self.assertEqual(1, config.a)
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698