Python Flow.use_grpc_gateway Beispiele

Beispiel #1

Datei: test_flow_change_gateway.py Projekt: stjordanis/jina

def test_change_gateway(restful, changeto_gateway, mocker):
    f = Flow(restful=restful).add().add().add(needs='pod1').needs_all()

    with f:
        mock = mocker.Mock()
        f.post('', random_docs(10), on_done=mock)
        mock.assert_called()

        mock = mocker.Mock()
        if changeto_gateway == 'RESTGateway':
            f.use_rest_gateway()
        if changeto_gateway == 'GRPCGateway':
            f.use_grpc_gateway()

        f.post('', random_docs(10), on_done=mock)
        mock.assert_called()

Beispiel #2

def get_flow():
    f = Flow().load_config('flows/query-multimodal.yml')
    f.use_grpc_gateway()
    return f