test_jupyter_websocket.py 文件源码

python
阅读 25 收藏 0 点赞 0 评论 0

项目:enterprise_gateway 作者: jupyter-incubator 项目源码 文件源码
def test_kernel_env(self):
        """Kernel should start with environment vars defined in the request."""
        self.app.personality.env_whitelist = ['TEST_VAR']
        kernel_body = json.dumps({
            'name': 'python',
            'env': {
                'KERNEL_FOO': 'kernel-foo-value',
                'NOT_KERNEL': 'ignored',
                'KERNEL_GATEWAY': 'overridden',
                'TEST_VAR': 'allowed'
            }
        })
        ws = yield self.spawn_kernel(kernel_body)
        req = self.execute_request('import os; print(os.getenv("KERNEL_FOO"), os.getenv("NOT_KERNEL"), os.getenv("KERNEL_GATEWAY"), os.getenv("TEST_VAR"))')
        ws.write_message(json_encode(req))
        content = yield self.await_stream(ws)
        self.assertEqual(content['name'], 'stdout')
        self.assertIn('kernel-foo-value', content['text'])
        self.assertNotIn('ignored', content['text'])
        self.assertNotIn('overridden', content['text'])
        self.assertIn('allowed', content['text'])

        ws.close()
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号