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()
test_jupyter_websocket.py 文件源码
python
阅读 25
收藏 0
点赞 0
评论 0
评论列表
文章目录