def setUp(self):
self.test_dir = tempfile.mkdtemp("LocalSchedulerTest")
write_shell_script(self.test_dir, "touch.sh", ["touch $1"])
write_shell_script(self.test_dir, "fail.sh", ["exit 1"])
write_shell_script(self.test_dir, "sleep.sh", ["sleep $1"])
self.image_fetcher = LocalDirectoryImageFetcher()
self.scheduler = LocalScheduler(self.image_fetcher)
self.test_container = Container(image=self.test_dir)
def setUp(self):
self.test_dir = tempfile.mkdtemp("StandaloneSessionTest")
write_shell_script(self.test_dir, "touch.sh", ["touch $1"])
write_shell_script(self.test_dir, "fail.sh", ["exit 1"])
write_shell_script(self.test_dir, "sleep.sh", ["sleep $1"])
self.image_fetcher = LocalDirectoryImageFetcher()
self.scheduler = LocalScheduler(self.image_fetcher)
# resource ignored for local scheduler; adding as an example
self.test_container = Container(image=self.test_dir).require(
Resource.SMALL)
def test_fetch_does_not_exist_should_throw(self):
non_existent_dir = join(self.test_dir, "non_existent_dir")
fetcher = LocalDirectoryImageFetcher()
with self.assertRaises(ValueError):
fetcher.fetch(non_existent_dir)
def test_fetch_relative_path_should_throw(self):
fetcher = LocalDirectoryImageFetcher()
with self.assertRaises(ValueError):
fetcher.fetch(self.test_dir_name)
def test_fetch_abs_path(self):
fetcher = LocalDirectoryImageFetcher()
self.assertEqual(self.test_dir, fetcher.fetch(self.test_dir))
