def test_run(self):
print('test run')
console = Console(None)
mock_state = MockState()
console.add_state('mock', mock_state)
console.set_current_state('mock')
console.run()
def test_add_state(self):
print('test add_state')
console = Console(None)
mock_state = mock.Mock()
console.add_state('mock', mock_state)
def test_set_current_state(self):
print('test set current state')
console = Console(None)
mock_state = mock.Mock()
console.add_state('mock', mock_state)
console.set_current_state('mock')
