def fastapi_client(): from daemon import _get_app app = _get_app() tc = TestClient(app) yield tc del tc
def fastapi_client(): from daemon import __root_workspace__ pathlib.Path(__root_workspace__).mkdir(parents=True, exist_ok=True) from daemon import _get_app app = _get_app() tc = TestClient(app) yield tc del tc
def get_partial_client(mode, monkeypatch): from daemon import __root_workspace__ pathlib.Path(__root_workspace__).mkdir(parents=True, exist_ok=True) from daemon import _get_app from daemon.models.enums import PartialDaemonModes from daemon import jinad_args from daemon import stores from importlib import reload jinad_args.mode = PartialDaemonModes(mode) reload(stores) app = _get_app(mode=mode) tc = TestClient(app) yield tc del tc
from jina.peapods.runtimes.gateway.http.app import get_fastapi_app JINA_LOGO_URL = 'https://api.jina.ai/logo/logo-product/jina-core/horizontal-layout/colored/Product%20logo_Core_vertical_colorful%402x-margin.png' DAEMON_SCHEMA_FILENAME = 'daemon.json' GATEWAY_SCHEMA_FILENAME = 'gateway.json' args = set_gateway_parser().parse_args([]) logger = JinaLogger('') gateway_app = get_fastapi_app(args, logger) gateway_schema = gateway_app.openapi() gateway_schema['info']['x-logo'] = {'url': JINA_LOGO_URL} gateway_schema['servers'] = [] gateway_schema['servers'].append({ 'url': f'http://localhost:{args.port_expose}', 'description': 'Local Jina gateway' }) with open(GATEWAY_SCHEMA_FILENAME, 'w') as f: json.dump(gateway_schema, f) args = get_main_parser().parse_args([]) daemon_app = _get_app() daemon_schema = daemon_app.openapi() daemon_schema['info']['x-logo'] = {'url': JINA_LOGO_URL} daemon_schema['servers'] = [] daemon_schema['servers'].append({ 'url': f'http://localhost:{args.port}', 'description': 'Local JinaD server' }) with open(DAEMON_SCHEMA_FILENAME, 'w') as f: json.dump(daemon_schema, f)
def fastapi_client(): from daemon import _get_app app = _get_app() client = TestClient(app) return client
def test_get_app_flow(monkeypatch, common_endpoints, flow_endpoints): monkeypatch.setattr(jinad_config, 'CONTEXT', 'flow') routes = [(route.name, route.path) for route in _get_app().routes] assert sorted(routes) == sorted(common_endpoints + flow_endpoints)