Index: appengine/findit/model/test/versioned_config_test.py |
diff --git a/appengine/findit/model/test/versioned_config_test.py b/appengine/findit/model/test/versioned_config_test.py |
index a1d2b268a91a642f0f917552c6b3644b81731c65..a19ce8b4f60ba762935c51fe405111e507cb1024 100644 |
--- a/appengine/findit/model/test/versioned_config_test.py |
+++ b/appengine/findit/model/test/versioned_config_test.py |
@@ -3,7 +3,7 @@ |
# found in the LICENSE file. |
from google.appengine.ext import ndb |
- |
+from google.appengine.api import users |
from testing_utils import testing |
from model.versioned_config import VersionedConfig |
@@ -14,9 +14,10 @@ class _Config(VersionedConfig): |
class VersionedConfigTest(testing.AppengineTestCase): |
+ |
def _CreateFirstVersion(self): |
config = _Config.Get() |
- config.Update(a=1) |
+ config.Update(users.User(email='admin@chromium.org'), True, a=1) |
def testGetWhenNoConfigCreatedYet(self): |
config = _Config.Get() |
@@ -26,14 +27,14 @@ class VersionedConfigTest(testing.AppengineTestCase): |
def testNonAdminCanNotUpdate(self): |
config = _Config.Get() |
with self.assertRaises(Exception): |
- config.Update() |
+ config.Update(users.User(email='admin@chromium.org'), False, a=1) |
def testUpdateWhenChanged(self): |
- self.mock_current_user(user_email='test@chromium.org', is_admin=True) |
self._CreateFirstVersion() |
config = _Config.Get() |
self.assertIsNotNone(config) |
- self.assertTrue(config.Update(a=2)) |
+ self.assertTrue(config.Update(users.User(email='admin@chromium.org'), True, |
+ a=2)) |
config = _Config.Get() |
self.assertIsNotNone(config) |
@@ -41,11 +42,11 @@ class VersionedConfigTest(testing.AppengineTestCase): |
self.assertEqual(2, config.a) |
def testNotUpdateWhenNotChanged(self): |
- self.mock_current_user(user_email='test@chromium.org', is_admin=True) |
self._CreateFirstVersion() |
config = _Config.Get() |
self.assertIsNotNone(config) |
- self.assertFalse(config.Update(a=1)) |
+ self.assertFalse(config.Update(users.User(email='admin@chromium.org'), True, |
+ a=1)) |
config = _Config.Get() |
self.assertIsNotNone(config) |