add Condor launcher tests

IPython/parallel/tests/test_launcher.py

@@ -1,179 +1,185
 """Tests for launchers
 
 Doesn't actually start any subprocesses, but goes through the motions of constructing
 objects, which should test basic config.
 
 Authors:
 
 * Min RK
 """
 
 #-------------------------------------------------------------------------------
 #  Copyright (C) 2013 The IPython Development Team
 #
 #  Distributed under the terms of the BSD License.  The full license is in
 #  the file COPYING, distributed as part of this software.
 #-------------------------------------------------------------------------------
 
 #-------------------------------------------------------------------------------
 # Imports
 #-------------------------------------------------------------------------------
 
 import logging
 import os
 import shutil
 import sys
 import tempfile
 
 from unittest import TestCase
 
 from nose import SkipTest
 
 from IPython.config import Config
 
 from IPython.parallel.apps import launcher
 
 from IPython.testing import decorators as dec
 
 
 #-------------------------------------------------------------------------------
 # TestCase Mixins
 #-------------------------------------------------------------------------------
 
 class LauncherTest:
     """Mixin for generic launcher tests"""
     def setUp(self):
         self.profile_dir = tempfile.mkdtemp(prefix="profile_")
     
     def tearDown(self):
         shutil.rmtree(self.profile_dir)
     
     @property
     def config(self):
         return Config()
     
     def build_launcher(self, **kwargs):
         kw = dict(
             work_dir=self.profile_dir,
             profile_dir=self.profile_dir,
             config=self.config,
             cluster_id='',
             log=logging.getLogger(),
         )
         kw.update(kwargs)
         return self.launcher_class(**kw)
 
     def test_profile_dir_arg(self):
         launcher = self.build_launcher()
         self.assertTrue("--profile-dir" in launcher.cluster_args)
         self.assertTrue(self.profile_dir in launcher.cluster_args)
 
     def test_cluster_id_arg(self):
         launcher = self.build_launcher()
         self.assertTrue("--cluster-id" in launcher.cluster_args)
         idx = launcher.cluster_args.index("--cluster-id")
         self.assertEqual(launcher.cluster_args[idx+1], '')
         launcher.cluster_id = 'foo'
         self.assertEqual(launcher.cluster_args[idx+1], 'foo')
     
     def test_args(self):
         launcher = self.build_launcher()
         for arg in launcher.args:
             self.assertTrue(isinstance(arg, basestring), str(arg))
 
 class BatchTest:
     """Tests for batch-system launchers (LSF, SGE, PBS)"""
     def test_batch_template(self):
         launcher = self.build_launcher()
         batch_file = os.path.join(self.profile_dir, launcher.batch_file_name)
         self.assertEqual(launcher.batch_file, batch_file)
         launcher.write_batch_script(1)
         self.assertTrue(os.path.isfile(batch_file))
 
 class SSHTest:
     """Tests for SSH launchers"""
     def test_cluster_id_arg(self):
         raise SkipTest("SSH Launchers don't support cluster ID")
     
     def test_remote_profile_dir(self):
         cfg = Config()
         launcher_cfg = getattr(cfg, self.launcher_class.__name__)
         launcher_cfg.remote_profile_dir = "foo"
         launcher = self.build_launcher(config=cfg)
         self.assertEqual(launcher.remote_profile_dir, "foo")
 
     def test_remote_profile_dir_default(self):
         launcher = self.build_launcher()
         self.assertEqual(launcher.remote_profile_dir, self.profile_dir)
 
 #-------------------------------------------------------------------------------
 # Controller Launcher Tests
 #-------------------------------------------------------------------------------
 
 class ControllerLauncherTest(LauncherTest):
     """Tests for Controller Launchers"""
     pass
 
 class TestLocalControllerLauncher(ControllerLauncherTest, TestCase):
     launcher_class = launcher.LocalControllerLauncher
 
 class TestMPIControllerLauncher(ControllerLauncherTest, TestCase):
     launcher_class = launcher.MPIControllerLauncher
 
 class TestPBSControllerLauncher(BatchTest, ControllerLauncherTest, TestCase):
     launcher_class = launcher.PBSControllerLauncher
 
 class TestSGEControllerLauncher(BatchTest, ControllerLauncherTest, TestCase):
     launcher_class = launcher.SGEControllerLauncher
 
 class TestLSFControllerLauncher(BatchTest, ControllerLauncherTest, TestCase):
     launcher_class = launcher.LSFControllerLauncher
 
+class TestCondorControllerLauncher(BatchTest, ControllerLauncherTest, TestCase):
+    launcher_class = launcher.CondorControllerLauncher
+
 class TestSSHControllerLauncher(SSHTest, ControllerLauncherTest, TestCase):
     launcher_class = launcher.SSHControllerLauncher
 
 #-------------------------------------------------------------------------------
 # Engine Set Launcher Tests
 #-------------------------------------------------------------------------------
 
 class EngineSetLauncherTest(LauncherTest):
     """Tests for EngineSet launchers"""
     pass
 
 class TestLocalEngineSetLauncher(EngineSetLauncherTest, TestCase):
     launcher_class = launcher.LocalEngineSetLauncher
 
 class TestMPIEngineSetLauncher(EngineSetLauncherTest, TestCase):
     launcher_class = launcher.MPIEngineSetLauncher
 
 class TestPBSEngineSetLauncher(BatchTest, EngineSetLauncherTest, TestCase):
     launcher_class = launcher.PBSEngineSetLauncher
 
 class TestSGEEngineSetLauncher(BatchTest, EngineSetLauncherTest, TestCase):
     launcher_class = launcher.SGEEngineSetLauncher
 
 class TestLSFEngineSetLauncher(BatchTest, EngineSetLauncherTest, TestCase):
     launcher_class = launcher.LSFEngineSetLauncher
 
+class TestCondorEngineSetLauncher(BatchTest, EngineSetLauncherTest, TestCase):
+    launcher_class = launcher.CondorEngineSetLauncher
+
 class TestSSHEngineSetLauncher(EngineSetLauncherTest, TestCase):
     launcher_class = launcher.SSHEngineSetLauncher
     
     def test_cluster_id_arg(self):
         raise SkipTest("SSH Launchers don't support cluster ID")
 
 class TestSSHProxyEngineSetLauncher(SSHTest, LauncherTest, TestCase):
     launcher_class = launcher.SSHProxyEngineSetLauncher
 
 class TestSSHEngineLauncher(SSHTest, LauncherTest, TestCase):
     launcher_class = launcher.SSHEngineLauncher
 
 #-------------------------------------------------------------------------------
 # Windows Launcher Tests
 #-------------------------------------------------------------------------------
 
 if sys.platform.startswith("win"):
     class TestWinHPCControllerLauncher(ControllerLauncherTest, TestCase):
         launcher_class = launcher.WindowsHPCControllerLauncher
 
     class TestWinHPCEngineSetLauncher(EngineSetLauncherTest, TestCase):
         launcher_class = launcher.WindowsHPCEngineSetLauncher