the-littlest-jupyterhub/tests/test_utils.py

import pytest
from tljh import utils
import subprocess
import logging


def test_run_subprocess_exception(mocker):
    logger = logging.getLogger("tljh")
    mocker.patch.object(logger, "error")
    with pytest.raises(subprocess.CalledProcessError):
        utils.run_subprocess(["/bin/bash", "-c", "echo error; exit 1"])
    logger.error.assert_called_with("error\n")


def test_run_subprocess(mocker):
    logger = logging.getLogger("tljh")
    mocker.patch.object(logger, "debug")
    utils.run_subprocess(["/bin/bash", "-c", "echo success"])
    logger.debug.assert_called_with("success\n")