# Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License.
import time

import pytest


def test_cli_args(container, http_client):
    """Container should respect notebook server command line args
    (e.g., disabling token security)"""
    container.run(
        command=['start-notebook.sh', '--NotebookApp.token=""']
    )
    resp = http_client.get('http://localhost:8888')
    resp.raise_for_status()
    assert 'login_submit' not in resp.text


@pytest.mark.filterwarnings('ignore:Unverified HTTPS request')
def test_unsigned_ssl(container, http_client):
    """Container should generate a self-signed SSL certificate
    and notebook server should use it to enable HTTPS.
    """
    container.run(
        environment=['GEN_CERT=yes']
    )
    # NOTE: The requests.Session backing the http_client fixture  does not retry
    # properly while the server is booting up. An SSL handshake error seems to
    # abort the retry logic. Forcing a long sleep for the moment until I have
    # time to dig more.
    time.sleep(5)
    resp = http_client.get('https://localhost:8888', verify=False)
    resp.raise_for_status()
    assert 'login_submit' in resp.text


def test_uid_change(container):
    """Container should change the UID of the default user."""
    c = container.run(
        tty=True,
        user='root',
        environment=['NB_UID=1010'],
        command=['start.sh', 'id && touch /opt/conda/test-file']
    )
    # usermod is slow so give it some time
    c.wait(timeout=120)
    assert 'uid=1010(jovyan)' in c.logs(stdout=True).decode('utf-8')


def test_gid_change(container):
    """Container should change the GID of the default user."""
    c = container.run(
        tty=True,
        user='root',
        environment=['NB_GID=110'],
        command=['start.sh', 'id']
    )
    c.wait(timeout=10)
    assert 'gid=110(users)' in c.logs(stdout=True).decode('utf-8')


def test_sudo(container):
    """Container should grant passwordless sudo to the default user."""
    c = container.run(
        tty=True,
        user='root',
        environment=['GRANT_SUDO=yes'],
        command=['start.sh', 'sudo', 'id']
    )
    rv = c.wait(timeout=10)
    assert rv == 0
    assert 'uid=0(root)' in c.logs(stdout=True).decode('utf-8')

@pytest.mark.dev
def test_group_add(container, tmpdir):
    """Container should run with the specified uid, gid, and secondary
    group.
    """
    c = container.run(
        user='1010:1010',
        group_add=['users'],
        command=['start.sh', 'id']
    )
    rv = c.wait(timeout=5)
    assert rv == 0
    assert 'uid=1010 gid=1010 groups=1010,100(users)' in c.logs(stdout=True).decode('utf-8')

@pytest.mark.dev
def test_host_mount(container, tmpdir):
    """Container should start the notebook server properly when
    the user home directory is host mounted.
    """
    home = tmpdir.mkdir('home')
    home.chmod(0o777)

    script = home.join('test.sh')
    script.write('''\
    #!/bin/bash
    echo "test content" > test.txt
    cat test.txt
    ''')
    script.chmod(0o755)

    c = container.run(
        volumes={
            home.strpath: {'bind': '/home/jovyan', 'mode': 'rw'},
        },
        command=['start.sh', '/home/jovyan/test.sh']
    )
    rv = c.wait(timeout=5)
    stdout = c.logs(stdout=True).decode('utf-8')
    assert rv == 0
    assert 'test content' in stdout