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(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))