def test_update_service_config(self):
# Missing.
self.assertIsNone(config._get_service_config('abc.cfg'))
self.assertIsNone(config._get_service_config_rev('abc.cfg'))
# Updated.
rev = config.Revision('rev', 'url')
self.assertTrue(config._update_service_config('abc.cfg', rev, 'body'))
self.assertEqual('body', config._get_service_config('abc.cfg'))
self.assertEqual(rev, config._get_service_config_rev('abc.cfg'))
# Same body, returns False, though updates rev.
rev2 = config.Revision('rev2', 'url')
self.assertFalse(config._update_service_config('abc.cfg', rev2, 'body'))
self.assertEqual(rev2, config._get_service_config_rev('abc.cfg'))
def test_update_service_config(self):
# Missing.
self.assertIsNone(config._get_service_config('abc.cfg'))
self.assertIsNone(config._get_service_config_rev('abc.cfg'))
# Updated.
rev = config.Revision('rev', 'url')
self.assertTrue(config._update_service_config('abc.cfg', rev, 'body'))
self.assertEqual('body', config._get_service_config('abc.cfg'))
self.assertEqual(rev, config._get_service_config_rev('abc.cfg'))
# Same body, returns False, though updates rev.
rev2 = config.Revision('rev2', 'url')
self.assertFalse(config._update_service_config('abc.cfg', rev2,
'body'))
self.assertEqual(rev2, config._get_service_config_rev('abc.cfg'))
def test_get_delegation_config(self):
# Missing -> returns empty proto.
proto = config.get_delegation_config()
self.assertFalse(proto.rules)
# Add some.
body = """rules {
user_id: "service:abc"
target_service: "*"
max_validity_duration: 3600
}"""
config._update_service_config('delegation.cfg',
config.Revision('rev', 'url'), body)
utils.clear_cache(config.get_delegation_config)
proto = config.get_delegation_config()
self.assertEqual(1, len(proto.rules))
def test_get_delegation_config(self):
# Missing -> returns empty proto.
proto = config.get_delegation_config()
self.assertFalse(proto.rules)
# Add some.
body = """rules {
user_id: "service:abc"
target_service: "*"
max_validity_duration: 3600
}"""
config._update_service_config(
'delegation.cfg', config.Revision('rev', 'url'), body)
utils.clear_cache(config.get_delegation_config)
proto = config.get_delegation_config()
self.assertEqual(1, len(proto.rules))
