smartdispatch.tests.CommandFilesTests

Complexity

Total Complexity

22

Size/Duplication

Total Lines	125
Duplicated Lines	0 %

10 Methods

Rating	Name	Duplication	Size	Complexity
A	CommandFilesTests.test_set_running_command_as_failed()	19	19	4
A	CommandFilesTests.test_get_nb_commands_to_run()	0	2	1
A	CommandFilesTests.test_get_command_to_run()	0	14	3
A	CommandFilesTests.test_set_running_command_as_finished()	18	18	4
A	CommandFilesTests.test_reset_running_commands()	0	15	3
A	CommandFilesTests.setUp()	0	13	2
A	CommandFilesTests.test_get_failed_commands_empty()	0	6	1
A	CommandFilesTests.test_get_failed_commands()	0	11	1
A	CommandFilesTests.test_set_commands_to_run()	0	14	2
A	CommandFilesTests.tearDown()	0	2	1

import os
import unittest
import tempfile as tmp
import shutil

from smartdispatch.command_manager import CommandManager
from nose.tools import assert_equal, assert_true


class CommandFilesTests(unittest.TestCase):

    def setUp(self):
        self._base_dir = tmp.mkdtemp()
        self.nb_commands = 3
        self.command1 = "1\n"
        self.command2 = "2\n"
        self.command3 = "3\n"

        command_filename = os.path.join(self._base_dir, "commands.txt")

        with open(command_filename, "w+") as commands_file:
            commands_file.write(self.command1 + self.command2 + self.command3)

        self.command_manager = CommandManager(command_filename)

    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
        command = self.command_manager.get_command_to_run()
        self.command_manager.set_running_command_as_finished(command, 1)

        # The function to test
        failed_commands = self.command_manager.get_failed_commands()

        # Test validation
        assert_equal(len(failed_commands), 1)
        assert_equal(failed_commands[0], self.command1)

    def test_get_failed_commands_empty(self):
        # The function to test
        failed_commands = self.command_manager.get_failed_commands()

        # 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 test_set_running_command_as_finished(self):

This code seems to be duplicated in your project.

        # SetUp
        command = self.command_manager.get_command_to_run()

        # The function to test
        self.command_manager.set_running_command_as_finished(command)

        # 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._finished_commands_filename, "r") as finished_commands_file:
            assert_equal(finished_commands_file.read(), self.command1)

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

    def test_set_running_command_as_failed(self):

This code seems to be duplicated in your project.

        # 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):
        # SetUp
        self.command_manager.get_command_to_run()

        # The function to test
        self.command_manager.reset_running_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)

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

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