177 lines
6.7 KiB
Python
177 lines
6.7 KiB
Python
import unittest
|
|
from io import StringIO
|
|
from pathlib import Path
|
|
from textwrap import dedent
|
|
from unittest.mock import patch, MagicMock
|
|
|
|
import contrib.playerInspect as player_inspect
|
|
|
|
|
|
class TestPlayerInspect(unittest.TestCase):
|
|
def setUp(self):
|
|
self.player_data = {
|
|
'AbsorptionAmount': 0.0,
|
|
'Air': 300,
|
|
'Attributes': [
|
|
{'Base': 20.0, 'Name': 'generic.maxHealth'},
|
|
{'Base': 0.0, 'Name': 'generic.knockbackResistance'},
|
|
{'Base': 0.10000000149011612, 'Name': 'generic.movementSpeed'},
|
|
{'Base': 0.0, 'Name': 'generic.armor'},
|
|
{'Base': 0.0, 'Name': 'generic.armorToughness'},
|
|
{'Base': 1.0, 'Name': 'generic.attackDamage'},
|
|
{'Base': 4.0, 'Name': 'generic.attackSpeed'},
|
|
{'Base': 0.0, 'Name': 'generic.luck'}
|
|
],
|
|
'DataVersion': 1631,
|
|
'DeathTime': 0,
|
|
'Dimension': 0,
|
|
'EnderItems': [],
|
|
'FallDistance': 0.0,
|
|
'FallFlying': 0,
|
|
'Fire': -20,
|
|
'Health': 20.0,
|
|
'HurtByTimestamp': 0,
|
|
'HurtTime': 0,
|
|
'Inventory': [{'Count': 1, 'Slot': -106, 'id': 'minecraft:sign'}],
|
|
'Invulnerable': 0,
|
|
'Motion': [0.0, -0.0784000015258789, 0.0],
|
|
'OnGround': 1,
|
|
'PortalCooldown': 0,
|
|
'Pos': [-96.11859857363737, 70.0, -44.17768261916891],
|
|
'Rotation': [-72.00011444091797, 38.250030517578125],
|
|
'Score': 0,
|
|
'SelectedItemSlot': 0,
|
|
'SleepTimer': 0,
|
|
'Sleeping': 0,
|
|
"SpawnX": 10,
|
|
"SpawnY": 52,
|
|
"SpawnZ": 10,
|
|
'UUIDLeast': -7312926203658200544,
|
|
'UUIDMost': 6651100054519957107,
|
|
'XpLevel': 0,
|
|
'XpP': 0.0,
|
|
'XpSeed': 0,
|
|
'XpTotal': 0,
|
|
'abilities': {
|
|
'flySpeed': 0.05000000074505806,
|
|
'flying': 0,
|
|
'instabuild': 1,
|
|
'invulnerable': 1,
|
|
'mayBuild': 1,
|
|
'mayfly': 1,
|
|
'walkSpeed': 0.10000000149011612
|
|
},
|
|
'foodExhaustionLevel': 0.0,
|
|
'foodLevel': 20,
|
|
'foodSaturationLevel': 5.0,
|
|
'foodTickTimer': 0,
|
|
'playerGameType': 1,
|
|
'recipeBook': {
|
|
'isFilteringCraftable': 0,
|
|
'isFurnaceFilteringCraftable': 0,
|
|
'isFurnaceGuiOpen': 0,
|
|
'isGuiOpen': 0,
|
|
'recipes': [],
|
|
'toBeDisplayed': []
|
|
},
|
|
'seenCredits': 0
|
|
}
|
|
|
|
@patch('sys.stdout', new_callable=StringIO)
|
|
def test_print_player(self, mock_stdout):
|
|
expected = "\n".join([
|
|
"Position:\t-96, 70, -44\t(dim: 0)",
|
|
"Spawn:\t\t10, 52, 10",
|
|
"Health:\t20\tLevel:\t\t0\t\tGameType:\t1",
|
|
"Food:\t20\tTotal XP:\t0",
|
|
"Inventory: 1 items",
|
|
" 1 minecraft:sign\n"])
|
|
|
|
player_inspect.print_player(self.player_data)
|
|
|
|
self.assertEqual(mock_stdout.getvalue(), expected)
|
|
|
|
@patch('sys.stdout', new_callable=StringIO)
|
|
def test_print_player_no_spawn(self, mock_stdout):
|
|
expected = "\n".join([
|
|
"Position:\t-96, 70, -44\t(dim: 0)",
|
|
"Health:\t20\tLevel:\t\t0\t\tGameType:\t1",
|
|
"Food:\t20\tTotal XP:\t0",
|
|
"Inventory: 1 items",
|
|
" 1 minecraft:sign\n"])
|
|
|
|
player_data = {
|
|
k: v for k, v in self.player_data.items()
|
|
if k not in("SpawnX", "SpawnY", "SpawnZ")
|
|
}
|
|
player_inspect.print_player(player_data)
|
|
|
|
self.assertEqual(mock_stdout.getvalue(), expected)
|
|
|
|
@patch('sys.stdout', new_callable=StringIO)
|
|
def test_print_player_sub_entry(self, mock_stdout):
|
|
expected = "\n".join([
|
|
"\tPosition:\t-96, 70, -44\t(dim: 0)",
|
|
"\tSpawn:\t\t10, 52, 10",
|
|
"\tHealth:\t20\tLevel:\t\t0\t\tGameType:\t1",
|
|
"\tFood:\t20\tTotal XP:\t0",
|
|
"\tInventory: 1 items\n"])
|
|
|
|
player_inspect.print_player(self.player_data, sub_entry=True)
|
|
|
|
self.assertEqual(mock_stdout.getvalue(), expected)
|
|
|
|
@patch('sys.stdout', new_callable=StringIO)
|
|
def test_print_player_sub_entry_no_spawn(self, mock_stdout):
|
|
expected = "\n".join([
|
|
"\tPosition:\t-96, 70, -44\t(dim: 0)",
|
|
"\tHealth:\t20\tLevel:\t\t0\t\tGameType:\t1",
|
|
"\tFood:\t20\tTotal XP:\t0",
|
|
"\tInventory: 1 items\n"])
|
|
|
|
player_data = {
|
|
k: v for k, v in self.player_data.items()
|
|
if k not in("SpawnX", "SpawnY", "SpawnZ")
|
|
}
|
|
player_inspect.print_player(player_data, sub_entry=True)
|
|
|
|
self.assertEqual(mock_stdout.getvalue(), expected)
|
|
|
|
def test_find_all_player_files(self):
|
|
dir_path = MagicMock(Path)
|
|
files = [Path('def0492d-0fe9-43ff-a3d5-8c3fc9160c94.dat'),
|
|
Path('074c808a-1f04-4bdd-8385-bd74601210a1.dat'),
|
|
Path('104e149d-a802-4a27-ac8f-ceab5279087c.dat')]
|
|
dir_path.iterdir.return_value = (f for f in files)
|
|
|
|
expected = [(Path('def0492d-0fe9-43ff-a3d5-8c3fc9160c94.dat'),
|
|
'def0492d-0fe9-43ff-a3d5-8c3fc9160c94'),
|
|
(Path('074c808a-1f04-4bdd-8385-bd74601210a1.dat'),
|
|
'074c808a-1f04-4bdd-8385-bd74601210a1'),
|
|
(Path('104e149d-a802-4a27-ac8f-ceab5279087c.dat'),
|
|
'104e149d-a802-4a27-ac8f-ceab5279087c')]
|
|
result = player_inspect.find_all_player_files(dir_path)
|
|
self.assertListEqual(list(result), expected)
|
|
|
|
def test_find_player_file(self):
|
|
dir_path = MagicMock(Path)
|
|
files = [Path('def0492d-0fe9-43ff-a3d5-8c3fc9160c94.dat'),
|
|
Path('074c808a-1f04-4bdd-8385-bd74601210a1.dat'),
|
|
Path('104e149d-a802-4a27-ac8f-ceab5279087c.dat')]
|
|
dir_path.iterdir.return_value = (f for f in files)
|
|
|
|
expected = (Path('104e149d-a802-4a27-ac8f-ceab5279087c.dat'),
|
|
'104e149d-a802-4a27-ac8f-ceab5279087c')
|
|
result = player_inspect.find_player_file(
|
|
dir_path, selected_player='104e149d-a802-4a27-ac8f-ceab5279087c')
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_find_player_file_raises_when_selected_player_not_found(self):
|
|
dir_path = MagicMock(Path)
|
|
files = [Path('def0492d-0fe9-43ff-a3d5-8c3fc9160c94.dat'),
|
|
Path('104e149d-a802-4a27-ac8f-ceab5279087c.dat')]
|
|
dir_path.iterdir.return_value = (f for f in files)
|
|
|
|
with self.assertRaises(FileNotFoundError):
|
|
player_inspect.find_player_file(dir_path, selected_player='NON_EXISTENT_UUID')
|