anthony-melin
122 lines
3.2 KiB
Python
122 lines
3.2 KiB
Python
import unittest
|
|
from unittest.mock import patch
|
|
import json
|
|
import pathlib
|
|
|
|
from base_mob import BaseMob
|
|
from character_builder import CharacterBuilder
|
|
|
|
ARTHUR_JSON = """{
|
|
"name": "Arthur",
|
|
"max_pv": 100,
|
|
"current_pv": 100,
|
|
"strength": 20,
|
|
"protection": 10,
|
|
"speed": 5,
|
|
"mob_type": 0
|
|
}"""
|
|
|
|
ARTHUR_FILE = pathlib.Path('Arthur.json')
|
|
|
|
|
|
class TestCharacterBuilder(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
ARTHUR_FILE.unlink(missing_ok=True)
|
|
|
|
@patch("builtins.input")
|
|
def test_build_character(self, mock_input):
|
|
# Arrange
|
|
mock_input.side_effect = [
|
|
"Arthur", # name
|
|
"100", # max_pv
|
|
"20", # strength
|
|
"10", # protection
|
|
"5" # speed
|
|
]
|
|
|
|
# Act
|
|
character = CharacterBuilder.build_character()
|
|
|
|
# Assert
|
|
self.assertIsInstance(character, BaseMob)
|
|
self.assertEqual(character.name, "Arthur")
|
|
self.assertEqual(character.max_pv, 100)
|
|
self.assertEqual(character.strength, 20)
|
|
self.assertEqual(character.protection, 10)
|
|
self.assertEqual(character.speed, 5)
|
|
self.assertEqual(character.mob_type, 0)
|
|
|
|
def test_save_character(self):
|
|
# Arrange
|
|
character = BaseMob(
|
|
name="Arthur",
|
|
max_pv=100,
|
|
strength=20,
|
|
protection=10,
|
|
speed=5,
|
|
mob_type=0,
|
|
)
|
|
|
|
# Act
|
|
CharacterBuilder.save_character(character)
|
|
|
|
# Assert
|
|
self.assertTrue(ARTHUR_FILE.exists())
|
|
self.assertEqual(ARTHUR_JSON, ARTHUR_FILE.read_text(encoding='utf-8'))
|
|
|
|
def test_load_character_file_not_found(self):
|
|
# Act
|
|
result = CharacterBuilder.load_character("Lancelot")
|
|
|
|
# Assert
|
|
self.assertIsNone(result)
|
|
|
|
def test_load_character_success(self):
|
|
# Arrange
|
|
ARTHUR_FILE.write_text(ARTHUR_JSON, encoding='utf-8')
|
|
|
|
# Act
|
|
character = CharacterBuilder.load_character("Arthur")
|
|
|
|
# Assert
|
|
self.assertIsNotNone(character)
|
|
self.assertEqual(character.name, "Arthur")
|
|
self.assertEqual(character.max_pv, 100)
|
|
self.assertEqual(character.strength, 20)
|
|
self.assertEqual(character.protection, 10)
|
|
self.assertEqual(character.speed, 5)
|
|
self.assertEqual(character.mob_type, 0)
|
|
|
|
def test_load_character_missing_keys(self):
|
|
arthur_dict: dict = json.loads(ARTHUR_JSON)
|
|
|
|
for key in arthur_dict:
|
|
with self.subTest(f'test without key "{key}"'):
|
|
# Arrange
|
|
invalid_dict = arthur_dict.copy()
|
|
invalid_dict.pop(key)
|
|
ARTHUR_FILE.write_text(json.dumps(
|
|
invalid_dict), encoding='utf-8')
|
|
|
|
# Act
|
|
result = CharacterBuilder.load_character("Arthur")
|
|
|
|
# Assert
|
|
self.assertIsNone(result)
|
|
|
|
def test_load_character_invalid_json(self):
|
|
# Arrange
|
|
ARTHUR_FILE.write_text(ARTHUR_JSON[1:], encoding='utf-8')
|
|
|
|
# Act
|
|
result = CharacterBuilder.load_character("Arthur")
|
|
|
|
# Assert
|
|
self.assertIsNone(result)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|