From b412d5413db89b256f7c3439d74c54e31dba4bb3 2014-05-07 00:35:28
From: Thomas Kluyver <takowl@gmail.com>
Date: 2014-05-07 00:35:28
Subject: [PATCH] Add test for installing kernel spec

---

diff --git a/IPython/kernel/tests/test_kernelspec.py b/IPython/kernel/tests/test_kernelspec.py
index 2479d10..a48e530 100644
--- a/IPython/kernel/tests/test_kernelspec.py
+++ b/IPython/kernel/tests/test_kernelspec.py
@@ -3,6 +3,7 @@ import os
 from os.path import join as pjoin
 import unittest
 
+from IPython.testing.decorators import onlyif
 from IPython.utils.tempdir import TemporaryDirectory
 from IPython.kernel import kernelspec
 
@@ -13,7 +14,8 @@ sample_kernel_json = {'argv':['cat', '{connection_file}'],
 
 class KernelSpecTests(unittest.TestCase):
     def setUp(self):
-        self.tempdir = td = TemporaryDirectory()
+        td = TemporaryDirectory()
+        self.addCleanup(td.cleanup)
         self.sample_kernel_dir = pjoin(td.name, 'kernels', 'Sample')
         os.makedirs(self.sample_kernel_dir)
         json_file = pjoin(self.sample_kernel_dir, 'kernel.json')
@@ -21,9 +23,12 @@ class KernelSpecTests(unittest.TestCase):
             json.dump(sample_kernel_json, f)
 
         self.ksm = kernelspec.KernelSpecManager(ipython_dir=td.name)
-
-    def tearDown(self):
-        self.tempdir.cleanup()
+        
+        td2 = TemporaryDirectory()
+        self.addCleanup(td2.cleanup)
+        self.installable_kernel = td2.name
+        with open(pjoin(self.installable_kernel, 'kernel.json'), 'w') as f:
+            json.dump(sample_kernel_json, f)
 
     def test_find_kernel_specs(self):
         kernels = self.ksm.find_kernel_specs()
@@ -36,4 +41,25 @@ class KernelSpecTests(unittest.TestCase):
         self.assertEqual(ks.display_name, sample_kernel_json['display_name'])
         self.assertEqual(ks.language, sample_kernel_json['language'])
         self.assertEqual(ks.codemirror_mode, sample_kernel_json['language'])
-        self.assertEqual(ks.env, {})
\ No newline at end of file
+        self.assertEqual(ks.env, {})
+    
+    def test_install_kernel_spec(self):
+        self.ksm.install_kernel_spec(self.installable_kernel,
+                                     kernel_name='tstinstalled')
+        self.assertIn('tstinstalled', self.ksm.find_kernel_specs())
+        
+        with self.assertRaises(OSError):
+            self.ksm.install_kernel_spec(self.installable_kernel,
+                                         kernel_name='tstinstalled')
+        
+        # Smoketest that this succeeds
+        self.ksm.install_kernel_spec(self.installable_kernel,
+                                     kernel_name='tstinstalled',
+                                     replace=True)
+    
+    @onlyif(os.name != 'nt' and os.getuid() != 0, "needs Unix system and non-root uid")
+    def test_cant_install_kernel_spec(self):
+        with self.assertRaises(OSError):
+            self.ksm.install_kernel_spec(self.installable_kernel,
+                                         kernel_name='tstinstalled',
+                                         system=True)