def test_get_self_config(self): expected = service_config_pb2.AclCfg(project_access_group='group') self.mock(storage, 'get_config_hash_async', mock.Mock()) self.mock(storage, 'get_config_by_hash_async', mock.Mock()) storage.get_config_hash_async.return_value = future( ('deadbeef', 'beefdead')) storage.get_config_by_hash_async.return_value = future( 'project_access_group: "group"') actual = storage.get_self_config_async( 'acl.cfg', service_config_pb2.AclCfg).get_result() self.assertEqual(expected, actual) storage.get_config_hash_async.assert_called_once_with( 'services/sample-app', 'acl.cfg') storage.get_config_by_hash_async.assert_called_once_with('beefdead') # memcached: storage.get_config_hash_async.reset_mock() storage.get_config_by_hash_async.reset_mock() actual = storage.get_latest_as_message_async( 'services/sample-app', 'acl.cfg', service_config_pb2.AclCfg).get_result() self.assertEqual(expected, actual) self.assertFalse(storage.get_config_hash_async.called) self.assertFalse(storage.get_config_by_hash_async.called)
def test_get_self_config(self): expected = service_config_pb2.AclCfg(project_access_group="group") self.mock(storage, "get_config_hash_async", mock.Mock()) self.mock(storage, "get_config_by_hash_async", mock.Mock()) storage.get_config_hash_async.return_value = future(("deadbeef", "beefdead")) storage.get_config_by_hash_async.return_value = future('project_access_group: "group"') actual = storage.get_self_config_async("acl.cfg", service_config_pb2.AclCfg).get_result() self.assertEqual(expected, actual) storage.get_config_hash_async.assert_called_once_with("services/sample-app", "acl.cfg") storage.get_config_by_hash_async.assert_called_once_with("beefdead") # memcached: storage.get_config_hash_async.reset_mock() storage.get_config_by_hash_async.reset_mock() actual = storage.get_latest_as_message_async( "services/sample-app", "acl.cfg", service_config_pb2.AclCfg ).get_result() self.assertEqual(expected, actual) self.assertFalse(storage.get_config_hash_async.called) self.assertFalse(storage.get_config_by_hash_async.called)
def _get_project_config(project_id, path, message_factory): assert project_id return storage.get_latest_as_message_async('projects/%s' % project_id, path, message_factory).get_result()
def _get_project_config(project_id, path, message_factory): assert project_id return storage.get_latest_as_message_async( 'projects/%s' % project_id, path, message_factory).get_result()