SMART-Lab / smartdispatch

        assert_equal(exit_status_100, 2)

        assert_true(os.path.isdir(self.logs_dir))

    @patch('subprocess.check_output')

    def test_launch_job_check(self, mock_check_output):

        #For this test, we won't call the script directly, since we want to mock subprocess.check_output

        argv = ['-t', '0:0:1', '-G', '1', '-C', '1', '-q', 'random', 'launch', 'echo', 'testing123']

        # Test if the test pass (i.e the script run normaly)

        mock_check_output.side_effect = None

        mock_check_output.return_value = ""

        try:

            smartdispatch_script.main(argv=argv)

        except SystemExit as e:

            self.fail("The launcher had no problem, but the script failed nonetheless.")

        # Test if the check fail

        mock_check_output.side_effect = subprocess.CalledProcessError(1, 1, "A wild error appeared!")

        try:

            with self.assertRaises(SystemExit) as context:

                smartdispatch_script.main(argv=argv)

                self.assertTrue(context.exception.code, 2)

        except subprocess.CalledProcessError:

            # Rerasing the exception

            orig_exc_type, orig_exc_value, orig_exc_traceback = sys.exc_info()

            new_exc = Exception("smartdispatch_script.main() raised subprocess.CalledProcessError unexpectedly")

            new_exc.reraised = True

            six.reraise(type(new_exc), new_exc, orig_exc_traceback)

    def test_main_resume(self):

        # Setup

    @patch('subprocess.check_output')

    def test_launch_job_check(self, mock_check_output):

        argv = ['-t', '0:0:1', '-G', '1', '-C', '1', '-q', 'random', 'launch', 'echo', 'testing123']

        # Test if the test pass (i.e the script run normaly)

        mock_check_output.side_effect = None

        mock_check_output.return_value = ""

        try:

            smartdispatch_script.main(argv=argv)

        except SystemExit as e:

          self.fail("The launcher had no problem, but the script failed nonetheless.")

        mock_check_output.side_effect = subprocess.CalledProcessError(1, "echo blabla", "A wild error appeared!")

        # Test if the test fail.

        try:

            with self.assertRaises(SystemExit) as context:

                smartdispatch_script.main(argv=argv)

                self.assertTrue(context.exception.code, 2)

        except subprocess.CalledProcessError as e:

            # Rerasing the exception

            orig_exc_type, orig_exc_value, orig_exc_traceback = sys.exc_info()

            new_exc = Exception("smartdispatch_script.main() raised subprocess.CalledProcessError unexpectedly")

            new_exc.reraised = True

            six.reraise(type(new_exc), new_exc, orig_exc_traceback)