SMART-Lab / smartdispatch

    def test_set_running_command_as_failed(self):

        # SetUp

        command = self.command_manager.get_command_to_run()

        error_code = 1

        # The function to test

        self.command_manager.set_running_command_as_finished(command, error_code)

        # Test validation

        with open(self.command_manager._commands_filename, "r") as commands_file:

            assert_equal(commands_file.read(), self.command2 + self.command3)

        with open(self.command_manager._running_commands_filename, "r") as running_commands_file:

            assert_equal(running_commands_file.read(), "")

        with open(self.command_manager._failed_commands_filename, "r") as failed_commands_file:

            assert_equal(failed_commands_file.read(), self.command1)

        assert_true(not os.path.isfile(self.command_manager._finished_commands_filename))

    def test_reset_running_commands(self):

        # Test validation

        assert_equal(len(failed_commands), 0)

    def test_get_command_to_run(self):

        # The function to test

        command = self.command_manager.get_command_to_run()

        # Test validation

        assert_equal(command, self.command1.strip())

        with open(self.command_manager._commands_filename, "r") as commands_file:

            assert_equal(commands_file.read(), self.command2 + self.command3)

        with open(self.command_manager._running_commands_filename, "r") as running_commands_file:

            assert_equal(running_commands_file.read(), self.command1)

        assert_true(not os.path.isfile(self.command_manager._finished_commands_filename))

    def test_get_nb_commands_to_run(self):

        assert_equal(self.command_manager.get_nb_commands_to_run(), self.nb_commands)

    def tearDown(self):

        shutil.rmtree(self._base_dir)

    def test_set_commands_to_run(self):

        # SetUp

        commands = ["4", "5", "6"]

        # The function to test

        self.command_manager.set_commands_to_run(commands)

        # Test validation

        with open(self.command_manager._commands_filename, "r") as commands_file:

            assert_equal(commands_file.read(), self.command1 + self.command2 + self.command3 + "4\n5\n6\n")

        assert_true(not os.path.isfile(self.command_manager._running_commands_filename))

        assert_true(not os.path.isfile(self.command_manager._finished_commands_filename))

    def test_get_failed_commands(self):

        # Setup