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)
