source: nagslang/tests/test_game_object.py@ 124:ae61528436f1

from unittest import TestCase

from nagslang.constants import COLLISION_TYPE_OTHER, SWITCH_PUSHERS
from nagslang import game_object


class FakeShape(object):
    def __init__(self, collision_type=COLLISION_TYPE_OTHER):
        self.collision_type = collision_type


class FakeSpace(object):
    def __init__(self, *shapes):
        self._shapes = shapes

    def shape_query(self, shape):
        return self._shapes


class FakeGameObject(object):
    def __init__(self, shape, space):
        self._shape = shape
        self._space = space

    def get_shape(self):
        return self._shape

    def get_space(self):
        return self._space


class FakePuzzler(game_object.Puzzler):
    def __init__(self, fake_state):
        self.fake_state = fake_state

    def get_state(self):
        return self.fake_state


class TestPuzzles(TestCase):
    def mkpuzzler(self, gobj, cls, *args, **kw):
        puzzler = cls(*args, **kw)
        puzzler.set_game_object(gobj)
        return puzzler

    def test_floor_switch_puzzler(self):
        gobj = FakeGameObject(None, FakeSpace())
        puzzler = self.mkpuzzler(gobj, game_object.FloorSwitchPuzzler)
        self.assertFalse(puzzler.get_state())

        gobj = FakeGameObject(None, FakeSpace(FakeShape()))
        puzzler = self.mkpuzzler(gobj, game_object.FloorSwitchPuzzler)
        self.assertFalse(puzzler.get_state())

        for collision_type in SWITCH_PUSHERS:
            gobj = FakeGameObject(
                None, FakeSpace(FakeShape(collision_type)))
            puzzler = self.mkpuzzler(gobj, game_object.FloorSwitchPuzzler)
            self.assertTrue(puzzler.get_state())

    def test_state_proxy_puzzler(self):
        glue = game_object.PuzzleGlue()
        puzzler = game_object.StateProxyPuzzler('faker')
        glue.add_component('puzzler', puzzler)
        faker = FakePuzzler('foo')
        glue.add_component('faker', faker)

        self.assertEqual('foo', puzzler.get_state())
        faker.fake_state = 'bar'
        self.assertEqual('bar', puzzler.get_state())