|
1 | 1 | import unittest |
2 | 2 | from unittest.mock import MagicMock, patch |
3 | 3 |
|
| 4 | +from core.ProjectAliceExceptions import StateAlreadyRegistered |
4 | 5 | from core.base.StateManager import StateManager |
5 | 6 | from core.base.model.State import State |
6 | 7 | from core.base.model.StateType import StateType |
@@ -37,13 +38,25 @@ def test_register(self, mock_superManager): |
37 | 38 | self.assertIsInstance(stateManager.register('unit.tests'), State) |
38 | 39 |
|
39 | 40 |
|
| 41 | + @patch('core.base.SuperManager.SuperManager') |
| 42 | + def test__build_dict(self, mock_superManager): |
| 43 | + mock_instance = MagicMock() |
| 44 | + mock_superManager.getInstance.return_value = mock_instance |
| 45 | + mock_instance.commonsManager.getFunctionCaller.return_value = 'unittest' |
| 46 | + stateManager = StateManager() |
| 47 | + |
| 48 | + self.assertIsNone(stateManager._buildDict('unit.test.is.awesome', State('awesome'))) |
| 49 | + self.assertRaises(StateAlreadyRegistered, stateManager._buildDict, statePath='unit.test.is.awesome', state=State('awesome')) |
| 50 | + |
| 51 | + |
40 | 52 | @patch('core.base.SuperManager.SuperManager') |
41 | 53 | def test_get_state(self, mock_superManager): |
42 | 54 | mock_instance = MagicMock() |
43 | 55 | mock_superManager.getInstance.return_value = mock_instance |
44 | 56 | mock_instance.commonsManager.getFunctionCaller.return_value = 'unittest' |
45 | 57 | stateManager = StateManager() |
46 | 58 |
|
| 59 | + self.assertIsNone(stateManager.getState('')) |
47 | 60 | self.assertIsNone(stateManager.getState('unittest')) |
48 | 61 |
|
49 | 62 | stateManager.register('unit.test.is.awesome') |
@@ -74,6 +87,9 @@ def test_set_state(self, mock_superManager): |
74 | 87 | state = stateManager.getState('unit.test.is.awesome') |
75 | 88 | self.assertEqual(state, mockState) |
76 | 89 |
|
| 90 | + state = stateManager.setState('unit.test.is.crappy', StateType.WAITING) |
| 91 | + self.assertEqual(state, False) |
| 92 | + |
77 | 93 | stateManager.setState('unit.test.is.awesome', StateType.WAITING) |
78 | 94 | state = stateManager.getState('unit.test.is.awesome') |
79 | 95 | self.assertEqual(state.currentState, StateType.WAITING) |
|
0 commit comments