def setUp(self):
provision_info = json_format.Parse('{"retrievalToken": "test_token"}',
ProvisionInfo())
response = GetProvisionInfoResponse(info=provision_info)
def get_unused_port():
sock = socket.socket()
sock.bind(('', 0))
port = sock.getsockname()[1]
sock.close()
return port
class ProvisionService(ProvisionServiceServicer):
def GetProvisionInfo(self, request, context):
return response
def start_test_provision_server():
server = grpc.server(futures.ThreadPoolExecutor(max_workers=1))
add_ProvisionServiceServicer_to_server(ProvisionService(), server)
port = get_unused_port()
server.add_insecure_port('[::]:' + str(port))
server.start()
return server, port
self.provision_server, self.provision_port = start_test_provision_server(
)
self.env = dict(os.environ)
self.env["python"] = sys.executable
self.env["FLINK_BOOT_TESTING"] = "1"
self.env["BOOT_LOG_DIR"] = os.path.join(self.env["FLINK_HOME"], "log")
self.tmp_dir = tempfile.mkdtemp(str(time.time()), dir=self.tempdir)
# assume that this file is in flink-python source code directory.
flink_python_source_root = os.path.dirname(
os.path.dirname(
os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
runner_script = "pyflink-udf-runner.bat" if on_windows() else \
"pyflink-udf-runner.sh"
self.runner_path = os.path.join(flink_python_source_root, "bin",
runner_script)
def setUp(self):
manifest_response = json_format.Parse(manifest, GetManifestResponse())
artifact_chunks = dict()
for file_name in file_data:
artifact_chunks[file_name] = json_format.Parse(
file_data[file_name], ArtifactChunk())
provision_info = json_format.Parse(test_provision_info_json,
ProvisionInfo())
response = GetProvisionInfoResponse(info=provision_info)
def get_unused_port():
sock = socket.socket()
sock.bind(('', 0))
port = sock.getsockname()[1]
sock.close()
return port
class ArtifactService(ArtifactRetrievalServiceServicer):
def GetManifest(self, request, context):
return manifest_response
def GetArtifact(self, request, context):
yield artifact_chunks[request.name]
def start_test_artifact_server():
server = grpc.server(futures.ThreadPoolExecutor(max_workers=1))
add_ArtifactRetrievalServiceServicer_to_server(
ArtifactService(), server)
port = get_unused_port()
server.add_insecure_port('[::]:' + str(port))
server.start()
return server, port
class ProvisionService(ProvisionServiceServicer):
def GetProvisionInfo(self, request, context):
return response
def start_test_provision_server():
server = grpc.server(futures.ThreadPoolExecutor(max_workers=1))
add_ProvisionServiceServicer_to_server(ProvisionService(), server)
port = get_unused_port()
server.add_insecure_port('[::]:' + str(port))
server.start()
return server, port
self.artifact_server, self.artifact_port = start_test_artifact_server()
self.provision_server, self.provision_port = start_test_provision_server(
)
self.env = dict(os.environ)
self.env["python"] = sys.executable
self.env["FLINK_BOOT_TESTING"] = "1"
self.env["FLINK_LOG_DIR"] = os.path.join(self.env["FLINK_HOME"], "log")
self.tmp_dir = None
# assume that this file is in flink-python source code directory.
flink_python_source_root = os.path.dirname(
os.path.dirname(
os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
self.runner_path = os.path.join(flink_python_source_root, "bin",
"pyflink-udf-runner.sh")