def test_fails_if_info_dir_name_is_taken_by_a_regular_file(self): os.rmdir(self.info_dir) with open(self.info_dir, "w") as outfile: pass with self.assertRaises(OSError) as cm: manager._get_info_dir() self.assertEqual(cm.exception.errno, errno.EEXIST, cm.exception)
def setUp(self): super(ManagerEndToEndTest, self).setUp() # Spy on subprocesses spawned so that we can kill them. self.popens = [] class PopenSpy(subprocess.Popen): def __init__(p, *args, **kwargs): super(PopenSpy, p).__init__(*args, **kwargs) self.popens.append(p) popen_patcher = mock.patch.object(subprocess, "Popen", PopenSpy) popen_patcher.start() # Make sure that temporary files (including .tensorboard-info) are # created under our purview. self.tmproot = os.path.join(self.get_temp_dir(), "tmproot") os.mkdir(self.tmproot) self._patch_environ({"TMPDIR": self.tmproot}) tempfile.tempdir = None # force `gettempdir` to reinitialize from env self.assertEqual(tempfile.gettempdir(), self.tmproot) self.info_dir = manager._get_info_dir() # ensure that directory exists # Add our Bazel-provided `tensorboard` to the system path. (The # //tensorboard:tensorboard target is made available in the same # directory as //tensorboard:manager_e2e_test.) tensorboard_binary_dir = os.path.dirname(os.environ["TEST_BINARY"]) self._patch_environ({ "PATH": os.pathsep.join((tensorboard_binary_dir, os.environ["PATH"])), }) self._ensure_tensorboard_on_path(tensorboard_binary_dir)
def setUp(self): super(TensorBoardInfoIoTest, self).setUp() patcher = mock.patch.dict(os.environ, {"TMPDIR": self.get_temp_dir()}) patcher.start() self.addCleanup(patcher.stop) tempfile.tempdir = None # force `gettempdir` to reinitialize from env self.info_dir = manager._get_info_dir() # ensure that directory exists
def tensorboard_cleanup(): info_dir = manager._get_info_dir() shutil.rmtree(info_dir)