def get_bot_config(self, _request):
"""Retrieves the current version of bot_config.py."""
obj = bot_code.get_bot_config()
return swarming_rpcs.FileContent(content=obj.content.decode('utf-8'),
who=obj.who,
when=obj.when,
version=obj.version)
def test_get_bot_version(self):
actual, additionals = bot_code.get_bot_version('http://localhost')
self.assertTrue(re.match(r'^[0-9a-f]{64}$', actual), actual)
expected = {
'config/bot_config.py': bot_code.get_bot_config().content,
}
self.assertEqual(expected, additionals)
def test_get_bot_config(self):
def get_self_config_mock(path, revision=None, store_last_good=False):
self.assertEqual('scripts/bot_config.py', path)
self.assertEqual(None, revision)
self.assertEqual(True, store_last_good)
return None, 'foo bar'
self.mock(config, 'get_self_config', get_self_config_mock)
f = bot_code.get_bot_config()
self.assertEqual('foo bar', f.content)
def get(self):
bot_config = bot_code.get_bot_config()
params = {
"content": bot_config.content.decode("utf-8"),
"path": self.request.path,
"when": bot_config.when,
"who": bot_config.who,
"xsrf_token": self.generate_xsrf_token(),
}
self.response.write(template.render("swarming/restricted_upload_bot_config.html", params))
def get_bot_config(self, request):
"""Retrieves the current or a previous version of bot_config.py."""
obj = bot_code.get_bot_config(request.version)
if not obj:
return swarming_rpcs.FileContent()
return swarming_rpcs.FileContent(
content=obj.content.decode('utf-8'),
who=obj.who.to_bytes() if obj.who else None,
when=obj.when,
version=obj.version)
def get_bot_config(self, request):
"""Retrieves the current or a previous version of bot_config.py."""
obj = bot_code.get_bot_config(request.version)
if not obj:
return swarming_rpcs.FileContent()
return swarming_rpcs.FileContent(
content=obj.content.decode('utf-8'),
who=obj.who.to_bytes() if obj.who else None,
when=obj.when,
version=obj.version)
def get(self):
bot_config = bot_code.get_bot_config()
params = {
'content': bot_config.content.decode('utf-8'),
'path': self.request.path,
'when': bot_config.when,
'who': bot_config.who,
'xsrf_token': self.generate_xsrf_token(),
}
self.response.write(
template.render('swarming/restricted_upload_bot_config.html', params))
def get(self):
bot_config = bot_code.get_bot_config()
params = {
'content': bot_config.content.decode('utf-8'),
'path': self.request.path,
'when': bot_config.when,
'who': bot_config.who,
'xsrf_token': self.generate_xsrf_token(),
}
self.response.write(
template.render('swarming/restricted_upload_bot_config.html', params))
