if not provided, create temp HOME and notebook dirs

Author: bollwyvl

anaconda-project.yml

@@ -25,7 +25,7 @@ commands:
     env_spec: rfjl37
 
   atest:
-    unix: python -m atest.run
+    unix: rm -rf _artifacts/test_output && python -m atest.run
     env_spec: rfjl37
 
 

src/JupyterLibrary/keywords/server.py

@@ -1,5 +1,9 @@
+import os
+import shutil
 import subprocess
+import tempfile
 import time
+from os.path import join
 
 from robot.libraries.BuiltIn import BuiltIn
 from SeleniumLibrary.base import LibraryComponent, keyword
@@ -9,6 +13,7 @@
 
 class ServerKeywords(LibraryComponent):
     _nbserver_handles = []
+    _nbserver_tmpdirs = {}
 
     @keyword
     def start_new_jupyter_server(self, command="jupyter", *arguments, **configuration):
@@ -18,8 +23,23 @@ def start_new_jupyter_server(self, command="jupyter", *arguments, **configuratio
         if not arguments:
             arguments = self.build_jupyter_server_arguments()
 
+        tmpdir = tempfile.mkdtemp()
+
+        if "--notebook-dir" not in arguments:
+            notebook_dir = join(tmpdir, "notebooks")
+            os.mkdir(notebook_dir)
+            arguments += ["--notebook-dir", notebook_dir]
+
+        if "env:HOME" not in configuration:
+            home_dir = join(tmpdir, "home")
+            os.mkdir(home_dir)
+            configuration["env:HOME"] = home_dir
+
         handle = plib.start_process("jupyter", *arguments, **configuration)
+
         self._nbserver_handles += [handle]
+        self._nbserver_tmpdirs[handle] = tmpdir
+
         return handle
 
     @keyword
@@ -83,7 +103,11 @@ def terminate_all_jupyter_servers(self, kill=False):
             plib.terminate_process(handle, kill=kill)
             terminated += 1
 
+        for tmpdir in self._nbserver_tmpdirs.values():
+            shutil.rmtree(tmpdir)
+
         self._nbserver_handles = []
+        self._nbserver_tmpdirs = {}
 
         return terminated