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

Side by Side Diff: appengine/config_service/storage_test.py

Issue 1224913002: luci-config: fine-grained acls (Closed) Base URL: git@github.com:luci/luci-py.git@master
Patch Set: fine-grained acls for service configs Created 5 years, 5 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 #!/usr/bin/env python 1 #!/usr/bin/env python
2 # Copyright 2015 The Swarming Authors. All rights reserved. 2 # Copyright 2015 The Swarming Authors. All rights reserved.
3 # Use of this source code is governed by the Apache v2.0 license that can be 3 # Use of this source code is governed by the Apache v2.0 license that can be
4 # found in the LICENSE file. 4 # found in the LICENSE file.
5 5
6 from test_env import future 6 from test_env import future
7 import test_env 7 import test_env
8 test_env.setup_test_env() 8 test_env.setup_test_env()
9 9
10 from test_support import test_case 10 from test_support import test_case
(...skipping 130 matching lines...) Expand 10 before | Expand all | Expand 10 after
141 default_msg = service_config_pb2.ImportCfg( 141 default_msg = service_config_pb2.ImportCfg(
142 gitiles=service_config_pb2.ImportCfg.Gitiles(fetch_log_deadline=42)) 142 gitiles=service_config_pb2.ImportCfg.Gitiles(fetch_log_deadline=42))
143 self.mock(storage, 'get_latest_async', mock.Mock()) 143 self.mock(storage, 'get_latest_async', mock.Mock())
144 storage.get_latest_async.return_value = future( 144 storage.get_latest_async.return_value = future(
145 'gitiles { fetch_archive_deadline: 10 }') 145 'gitiles { fetch_archive_deadline: 10 }')
146 msg = storage.get_self_config_async( 146 msg = storage.get_self_config_async(
147 'import.cfg', lambda: default_msg).get_result() 147 'import.cfg', lambda: default_msg).get_result()
148 self.assertEqual(msg.gitiles.fetch_log_deadline, 42) 148 self.assertEqual(msg.gitiles.fetch_log_deadline, 42)
149 149
150 def test_get_self_config(self): 150 def test_get_self_config(self):
151 expected = service_config_pb2.AclCfg(service_access_group='group') 151 expected = service_config_pb2.AclCfg(project_access_group='group')
152 152
153 self.mock(storage, 'get_config_hash_async', mock.Mock()) 153 self.mock(storage, 'get_config_hash_async', mock.Mock())
154 self.mock(storage, 'get_config_by_hash_async', mock.Mock()) 154 self.mock(storage, 'get_config_by_hash_async', mock.Mock())
155 storage.get_config_hash_async.return_value = future( 155 storage.get_config_hash_async.return_value = future(
156 ('deadbeef', 'beefdead')) 156 ('deadbeef', 'beefdead'))
157 storage.get_config_by_hash_async.return_value = future( 157 storage.get_config_by_hash_async.return_value = future(
158 'service_access_group: "group"') 158 'project_access_group: "group"')
159 159
160 actual = storage.get_self_config_async( 160 actual = storage.get_self_config_async(
161 'acl.cfg', service_config_pb2.AclCfg).get_result() 161 'acl.cfg', service_config_pb2.AclCfg).get_result()
162 self.assertEqual(expected, actual) 162 self.assertEqual(expected, actual)
163 163
164 storage.get_config_hash_async.assert_called_once_with( 164 storage.get_config_hash_async.assert_called_once_with(
165 'services/sample-app', 'acl.cfg') 165 'services/sample-app', 'acl.cfg')
166 storage.get_config_by_hash_async.assert_called_once_with('beefdead') 166 storage.get_config_by_hash_async.assert_called_once_with('beefdead')
167 167
168 # memcached: 168 # memcached:
169 storage.get_config_hash_async.reset_mock() 169 storage.get_config_hash_async.reset_mock()
170 storage.get_config_by_hash_async.reset_mock() 170 storage.get_config_by_hash_async.reset_mock()
171 actual = storage.get_latest_as_message_async( 171 actual = storage.get_latest_as_message_async(
172 'services/sample-app', 'acl.cfg', 172 'services/sample-app', 'acl.cfg',
173 service_config_pb2.AclCfg).get_result() 173 service_config_pb2.AclCfg).get_result()
174 self.assertEqual(expected, actual) 174 self.assertEqual(expected, actual)
175 self.assertFalse(storage.get_config_hash_async.called) 175 self.assertFalse(storage.get_config_hash_async.called)
176 self.assertFalse(storage.get_config_by_hash_async.called) 176 self.assertFalse(storage.get_config_by_hash_async.called)
177 177
178 178
179 if __name__ == '__main__': 179 if __name__ == '__main__':
180 test_env.main() 180 test_env.main()
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698