sportarchive / CloudProcessingEngine-Decider

tests/state_test.py

"""Unit tests for pydecider.state_machine
"""

import os
import sys
import unittest

sys.path.append(
    os.path.realpath(
        os.path.join(
            os.path.dirname(__file__),
            '..'
        )
    )
)

import collections
import mock
import yaml

The import yaml could not be resolved.

The import yaml seems to be unused.

import pydecider
import pydecider.state


TestParent = collections.namedtuple('TestParent', ['name', 'output'])


class StepStateTest(unittest.TestCase):

This class should have a docstring.

    def setUp(self):
        self.mock_step = mock.Mock()
        self.step_state = pydecider.state.StepState(step=self.mock_step,
                                                    status='running',
                                                    context='__test__')

    def test_step_prepare_no_parents(self):
        """No parents, `step.prepare` called with an empty dict."""
        self.step_state.update('ready', '__test_update__')
        self.mock_step.prepare.assert_called_with({})

    def test_step_prepare_parents(self):
        """Make sure `step.prepare` is called with a dict of all the parents
        attributes.
        """
        self.step_state.parents = [
            TestParent('foo', {'a': 1}),
            TestParent('bar', {'b': 1}),
            TestParent('baz', {'c': 1}),
        ]
        self.step_state.update('ready', '__test_update__')
        self.mock_step.prepare.assert_called_with(
            {
                'foo': {'a': 1},
                'bar': {'b': 1},
                'baz': {'c': 1},
            }
        )

    def test_step_prepare_parents_with_input(self):

The name test_step_prepare_parents_with_input does not conform to the method naming conventions ([a-z_][a-z0-9_]{2,30}$).

        """Make sure `step.prepare` is called with a dict of all the parents
        attributes and that the special input step in properly named.
        """
        self.step_state.parents = [
            TestParent('foo', {'a': 1}),
            TestParent(pydecider.state.INIT_STEP, {'b': 1}),
            TestParent('baz', {'c': 1}),
        ]
        self.step_state.update('ready', '__test_update__')
        self.mock_step.prepare.assert_called_with(
            {
                'foo': {'a': 1},
                '__input__': {'b': 1},
                'baz': {'c': 1},
            }
        )

    def test_step_prepare_parents_with_none(self):

The name test_step_prepare_parents_with_none does not conform to the method naming conventions ([a-z_][a-z0-9_]{2,30}$).

        """Make sure `step.prepare` is called with a dict of all the parents
        attributes and parents with None as output are properly handled.
        """
        self.step_state.parents = [
            TestParent('foo', {'a': 1}),
            TestParent('bar', None),
            TestParent('baz', {'c': 1}),
        ]
        self.step_state.update('ready', '__test_update__')
        self.mock_step.prepare.assert_called_with(
            {
                'foo': {'a': 1},
                'bar': None,
                'baz': {'c': 1},
            }
        )


if __name__ == '__main__':
    import logging
    logging.basicConfig(level=logging.DEBUG)
    unittest.main()