Source code for aiida_restapi.routers.tests.tests

from pathlib import Path

from fastapi import APIRouter, Request
from fastapi.responses import HTMLResponse
from fastapi.templating import Jinja2Templates

from aiida_restapi.config import API_CONFIG
from aiida_restapi.routers.nodes import model_registry

read_router = APIRouter(prefix='/tests')
write_router = APIRouter(prefix='/tests')

here = Path(__file__).parent
templates = Jinja2Templates(str(here / 'templates'))
css = here / 'styles.css'




@write_router.get('/nodes/mutate', response_class=HTMLResponse)
async def test_nodes_mutate(request: Request) -> HTMLResponse:
    """Serve page to test node mutation requests."""
    node_types = '\n'.join(
        f'<option value="{node_type}">{node_type}</option>'
        for node_type in sorted(model_registry.get_node_types())
        if not node_type.startswith('process')
        and node_type not in {'data.core.code.Code.', 'data.core.code.abstract.AbstractCode.'}
    )
    js = here / 'templates' / 'node_mutation.js'
    return templates.TemplateResponse(
        'node_mutation.html',
        {
            'request': request,
            'css': css.read_text(),
            'js': js.read_text(),
            'api_prefix': API_CONFIG['PREFIX'],
            'options': node_types,
        },
    )





@read_router.get('/querybuilder', response_class=HTMLResponse)
async def test_querybuilder(request: Request) -> HTMLResponse:
    """Serve page to test query builder endpoint."""
    js = here / 'templates' / 'querybuilder.js'
    return templates.TemplateResponse(
        'querybuilder.html',
        {
            'request': request,
            'css': css.read_text(),
            'js': js.read_text(),
            'api_prefix': API_CONFIG['PREFIX'],
        },
    )