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)
