|
|
|
@ -13,11 +13,10 @@ |
|
|
|
|
# limitations under the License. |
|
|
|
|
"""The Python implementation of the GRPC interoperability test server.""" |
|
|
|
|
|
|
|
|
|
import argparse |
|
|
|
|
from concurrent import futures |
|
|
|
|
import logging |
|
|
|
|
|
|
|
|
|
from absl import app |
|
|
|
|
from absl.flags import argparse_flags |
|
|
|
|
import grpc |
|
|
|
|
|
|
|
|
|
from src.proto.grpc.testing import test_pb2_grpc |
|
|
|
@ -29,8 +28,8 @@ logging.basicConfig() |
|
|
|
|
_LOGGER = logging.getLogger(__name__) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def parse_interop_server_arguments(argv): |
|
|
|
|
parser = argparse_flags.ArgumentParser() |
|
|
|
|
def parse_interop_server_arguments(): |
|
|
|
|
parser = argparse.ArgumentParser() |
|
|
|
|
parser.add_argument( |
|
|
|
|
"--port", type=int, required=True, help="the port on which to serve" |
|
|
|
|
) |
|
|
|
@ -46,7 +45,7 @@ def parse_interop_server_arguments(argv): |
|
|
|
|
type=resources.parse_bool, |
|
|
|
|
help="require an ALTS connection", |
|
|
|
|
) |
|
|
|
|
return parser.parse_args(argv[1:]) |
|
|
|
|
return parser.parse_args() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_server_credentials(use_tls): |
|
|
|
@ -58,7 +57,9 @@ def get_server_credentials(use_tls): |
|
|
|
|
return grpc.alts_server_credentials() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def serve(args): |
|
|
|
|
def serve(): |
|
|
|
|
args = parse_interop_server_arguments() |
|
|
|
|
|
|
|
|
|
server = test_common.test_server() |
|
|
|
|
test_pb2_grpc.add_TestServiceServicer_to_server( |
|
|
|
|
service.TestService(), server |
|
|
|
@ -76,6 +77,4 @@ def serve(args): |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
|
|
app.run( |
|
|
|
|
serve, flags_parser=parse_interop_server_arguments |
|
|
|
|
) |
|
|
|
|
serve() |
|
|
|
|