def _get_server_info(api_endpoint=None): # TODO(cais): Add more plugins to the list when more plugin/data types # are supported plugins = ["scalars"] if api_endpoint: return server_info_lib.create_server_info(DEFAULT_ORIGIN, api_endpoint, plugins) return server_info_lib.fetch_server_info(DEFAULT_ORIGIN, plugins)
def test_response_with_plugins(self): frontend = "http://localhost:8080" backend = "localhost:10000" result = server_info.create_server_info(frontend, backend, ["plugin1", "plugin2"]) self.assertEqual(result.plugin_control.allowed_plugins, ["plugin1", "plugin2"])
def _get_server_info(flags): origin = flags.origin or _DEFAULT_ORIGIN if flags.api_endpoint and not flags.origin: return server_info_lib.create_server_info(origin, flags.api_endpoint) server_info = server_info_lib.fetch_server_info(origin) # Override with any API server explicitly specified on the command # line, but only if the server accepted our initial handshake. if flags.api_endpoint and server_info.api_server.endpoint: server_info.api_server.endpoint = flags.api_endpoint return server_info
def test(self): frontend = "http://localhost:8080" backend = "localhost:10000" result = server_info.create_server_info(frontend, backend) expected_compatibility = server_info_pb2.Compatibility() expected_compatibility.verdict = server_info_pb2.VERDICT_OK expected_compatibility.details = "" self.assertEqual(result.compatibility, expected_compatibility) expected_api_server = server_info_pb2.ApiServer() expected_api_server.endpoint = backend self.assertEqual(result.api_server, expected_api_server) url_format = result.url_format actual_url = url_format.template.replace(url_format.id_placeholder, "123") expected_url = "http://localhost:8080/experiment/123/" self.assertEqual(actual_url, expected_url)