/
test_registry.py
99 lines (78 loc) · 3.91 KB
/
test_registry.py
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
"""Test yarn registry replacement"""
# Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License.
import logging
import os
from os.path import join as pjoin
import subprocess
from tempfile import TemporaryDirectory
from unittest import TestCase
from unittest.mock import patch
from jupyter_core import paths
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'], 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()
self.assertNotIn(commands.YARN_DEFAULT_REGISTRY, lock)
self.assertIn(yarn_registry, lock)