class TestCharacterControllerGetCharacter(unittest.TestCase):
def setUp(self):
self.character_controller = CharacterController()
def test_get_character(self):
mock_char = Mock()
MockCharacter = Mock()
MockCharacter.get_objects = Mock(return_value=mock_char)
with patch.object(zoneserver, 'Object', MockCharacter):
result = self.character_controller.get_character("character")
self.assertNotEqual(result, False)
self.assertEqual(mock_char, result)
def test_get_character_non_existent_no_states(self):
from elixir_models import Object
MockCharacter = Mock()
MockCharacter.get_objects = Mock(side_effect=Object.DoesNotExist(''))
MockCharacter.DoesNotExist = Object.DoesNotExist
with patch.object(zoneserver, 'Object', MockCharacter):
result = self.character_controller.get_character("character")
self.assertFalse(result)
