File size: 4,323 Bytes
d1ceb73 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
"""Test yarn registry replacement"""
# Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License.
import logging
import subprocess
from os.path import join as pjoin
from unittest.mock import patch
from jupyterlab import commands
from .test_jupyterlab import AppHandlerTest
class TestAppHandlerRegistry(AppHandlerTest):
def test_node_not_available(self):
# patch should be applied on `jupyterlab.commands` and not on `jupyterlab_server.process`
# See https://docs.python.org/3/library/unittest.mock.html#where-to-patch
with patch("jupyterlab.commands.which") as which:
which.side_effect = ValueError("Command not found")
logger = logging.getLogger("jupyterlab")
config = commands._yarn_config(logger)
which.assert_called_once_with("node")
self.assertDictEqual(config, {"yarn config": {}, "npm config": {}})
def test_yarn_config(self):
with patch("subprocess.check_output") as check_output:
yarn_registry = "https://private.yarn/manager"
check_output.return_value = b"\n".join(
[
b'{"type":"info","data":"yarn config"}',
b'{"type":"inspect","data":{"registry":"'
+ bytes(yarn_registry, "utf-8")
+ b'"}}',
b'{"type":"info","data":"npm config"}',
b'{"type":"inspect","data":{"registry":"'
+ bytes(yarn_registry, "utf-8")
+ b'"}}',
]
)
logger = logging.getLogger("jupyterlab")
config = commands._yarn_config(logger)
self.assertDictEqual(
config,
{
"yarn config": {"registry": yarn_registry},
"npm config": {"registry": yarn_registry},
},
)
def test_yarn_config_failure(self):
with patch("subprocess.check_output") as check_output:
check_output.side_effect = subprocess.CalledProcessError(
1, ["yarn", "config", "list"], b"", stderr=b"yarn config failed."
)
logger = logging.getLogger("jupyterlab")
config = commands._yarn_config(logger)
self.assertDictEqual(config, {"yarn config": {}, "npm config": {}})
def test_get_registry(self):
with patch("subprocess.check_output") as check_output:
yarn_registry = "https://private.yarn/manager"
check_output.return_value = b"\n".join(
[
b'{"type":"info","data":"yarn config"}',
b'{"type":"inspect","data":{"registry":"'
+ bytes(yarn_registry, "utf-8")
+ b'"}}',
b'{"type":"info","data":"npm config"}',
b'{"type":"inspect","data":{"registry":"'
+ bytes(yarn_registry, "utf-8")
+ b'"}}',
]
)
handler = commands.AppOptions()
self.assertEqual(handler.registry, yarn_registry)
def test_populate_staging(self):
with patch("subprocess.check_output") as check_output:
yarn_registry = "https://private.yarn/manager"
check_output.return_value = b"\n".join(
[
b'{"type":"info","data":"yarn config"}',
b'{"type":"inspect","data":{"registry":"'
+ bytes(yarn_registry, "utf-8")
+ b'"}}',
b'{"type":"info","data":"npm config"}',
b'{"type":"inspect","data":{"registry":"'
+ bytes(yarn_registry, "utf-8")
+ b'"}}',
]
)
staging = pjoin(self.app_dir, "staging")
handler = commands._AppHandler(commands.AppOptions())
handler._populate_staging()
lock_path = pjoin(staging, "yarn.lock")
with open(lock_path) as f:
lock = f.read()
# yarn >=2.x does not record the registry in the lockfile
self.assertNotIn(commands.YARN_DEFAULT_REGISTRY, lock)
self.assertNotIn(yarn_registry, lock)
|