the-littlest-jupyterhub/integration-tests/plugins/simplest/tljh_simplest.py

"""
Simplest plugin that exercises all the hooks
"""
from textwrap import dedent

from tljh import systemd
from tljh.hooks import hookimpl


@hookimpl
def tljh_extra_user_conda_packages():
    return [
        'hypothesis',
    ]


@hookimpl
def tljh_extra_user_pip_packages():
    return [
        'django',
    ]

@hookimpl
def tljh_extra_hub_pip_packages():
    return [
        'there',
    ]

@hookimpl
def tljh_extra_apt_packages():
    return [
        'sl',
    ]


@hookimpl
def tljh_config_post_install(config):
    # Put an arbitrary marker we can test for
    config['simplest_plugin'] = {
        'present': True
    }

@hookimpl
def tljh_custom_jupyterhub_config(c):
    c.JupyterHub.authenticator_class = 'tmpauthenticator.TmpAuthenticator'


@hookimpl
def tljh_post_install():
    post_install_service = dedent("""
    [Unit]
    Description=Post Install Test Service

    [Service]
    ExecStart=ls

    [Install]
    WantedBy=multi-user.target
    """)
    service = "post-install-test.service"
    systemd.install_unit(service, post_install_service)
    systemd.enable_service(service)
    systemd.reload_daemon()