zulip/tools/test-help-documentation

#!/usr/bin/env python3
import argparse
import contextlib
import os
import subprocess
import sys
from typing import Iterator

ZULIP_PATH = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

# check for the venv
from lib import sanity_check

sanity_check.check_venv(__file__)

parser = argparse.ArgumentParser()
parser.add_argument('--force', action="store_true", help='Run tests despite possible problems.')
parser.add_argument(
    '--skip-external-links',
    dest="skip_external_link_check",
    action="store_true",
    help='Skip checking of external links.',
)
options = parser.parse_args()

os.chdir(ZULIP_PATH)
sys.path.insert(0, ZULIP_PATH)
from tools.lib.test_server import test_server_running

os.makedirs('var/help-documentation', exist_ok=True)

LOG_FILE = 'var/help-documentation/server.log'
external_host = "localhost:9981"

extra_args = ['-a', 'validate_html=set']

if options.skip_external_link_check:
    extra_args += ['-a', 'skip_external=set']


@contextlib.contextmanager
def vnu_servlet() -> Iterator[None]:
    with subprocess.Popen(
        [
            'java',
            '-cp',
            os.path.join(
                os.path.dirname(__file__),
                '../node_modules/vnu-jar/build/dist/vnu.jar',
            ),
            '-Dnu.validator.servlet.bind-address=127.0.0.1',
            'nu.validator.servlet.Main',
            '9988',
        ]
    ) as proc:
        yield
        proc.terminate()


with vnu_servlet(), test_server_running(options.force, external_host, log_file=LOG_FILE, dots=True):
    ret_help_doc = subprocess.call(
        ['scrapy', 'crawl_with_status', *extra_args, 'help_documentation_crawler'],
        cwd='tools/documentation_crawler',
    )
    ret_api_doc = subprocess.call(
        ['scrapy', 'crawl_with_status', *extra_args, 'api_documentation_crawler'],
        cwd='tools/documentation_crawler',
    )
    ret_portico_doc = subprocess.call(
        ['scrapy', 'crawl_with_status', *extra_args, 'portico_documentation_crawler'],
        cwd='tools/documentation_crawler',
    )

if ret_help_doc != 0 or ret_api_doc != 0 or ret_portico_doc != 0:
    print("\033[0;91m")
    print("Failed")
    print("\033[0m")
else:
    print("\033[0;92m")
    print("Passed!")
    print("\033[0m")


sys.exit(ret_help_doc or ret_api_doc or ret_portico_doc)