squirrel-battle/dungeonbattle/tests/entities_test.py

import unittest

from dungeonbattle.entities.items import Bomb, Item
from dungeonbattle.entities.monsters import Hedgehog
from dungeonbattle.entities.player import Player
from dungeonbattle.interfaces import Entity, Map


class TestEntities(unittest.TestCase):
    def setUp(self) -> None:
        """
        Load example map that can be used in tests.
        """
        self.map = Map.load("resources/example_map.txt")

    def test_basic_entities(self) -> None:
        """
        Test some random stuff with basic entities.
        """
        entity = Entity()
        entity.move(42, 64)
        self.assertEqual(entity.y, 42)
        self.assertEqual(entity.x, 64)
        self.assertIsNone(entity.act(self.map))

        other_entity = Entity()
        other_entity.move(45, 68)
        self.assertEqual(entity.distance_squared(other_entity), 25)
        self.assertEqual(entity.distance(other_entity), 5)

    def test_fighting_entities(self) -> None:
        """
        Test some random stuff with fighting entities.
        """
        entity = Hedgehog()
        self.map.add_entity(entity)
        self.assertEqual(entity.maxhealth, 10)
        self.assertEqual(entity.maxhealth, entity.health)
        self.assertEqual(entity.strength, 3)
        self.assertIsNone(entity.hit(entity))
        self.assertFalse(entity.dead)
        self.assertIsNone(entity.hit(entity))
        self.assertFalse(entity.dead)
        self.assertIsNone(entity.hit(entity))
        self.assertFalse(entity.dead)
        self.assertIsNone(entity.hit(entity))
        self.assertTrue(entity.dead)

        entity = Hedgehog()
        self.map.add_entity(entity)
        entity.move(2, 6)
        self.map.tick()
        self.assertFalse(entity.y == 2 and entity.x == 6)

    def test_items(self) -> None:
        """
        Test some random stuff with items.
        """
        item = Item()
        self.assertFalse(item.held)
        item.hold()
        self.assertTrue(item.held)
        item.drop(42, 42)
        self.assertEqual(item.y, 42)
        self.assertEqual(item.x, 42)

    def test_bombs(self) -> None:
        """
        Test some random stuff with bombs.
        """
        item = Bomb()
        hedgehog = Hedgehog()
        self.map.add_entity(item)
        self.map.add_entity(hedgehog)
        hedgehog.health = 2
        hedgehog.move(41, 42)
        item.act(self.map)
        self.assertFalse(hedgehog.dead)
        item.drop(42, 42)
        self.assertEqual(item.y, 42)
        self.assertEqual(item.x, 42)
        item.act(self.map)
        self.assertTrue(hedgehog.dead)

    def test_players(self) -> None:
        """
        Test some random stuff with players.
        """
        player = Player()
        self.map.add_entity(player)
        player.move(1, 6)
        self.assertEqual(player.strength, 5)
        self.assertEqual(player.health, player.maxhealth)
        self.assertEqual(player.maxhealth, 20)

        # Test movements and ensure that collisions are working
        self.assertFalse(player.move_up())
        self.assertTrue(player.move_left())
        self.assertFalse(player.move_left())
        for i in range(8):
            self.assertTrue(player.move_down())
        self.assertFalse(player.move_down())
        self.assertTrue(player.move_right())
        self.assertTrue(player.move_right())
        self.assertTrue(player.move_right())
        self.assertFalse(player.move_right())
        self.assertTrue(player.move_down())
        self.assertTrue(player.move_down())

        player.add_xp(70)
        self.assertEqual(player.current_xp, 10)
        self.assertEqual(player.max_xp, 40)
        self.assertEqual(player.level, 4)