diff pyntnclick/tests/game_logic_utils.py @ 730:eca4d2fc11a4 pyntnclick

Move game_logic_utils to pyntnclick
author Stefano Rivera <stefano@rivera.za.net>
date Tue, 15 Jan 2013 22:56:30 +0200
parents gamelib/tests/game_logic_utils.py@7d777c9a1af1
children 2f74064bc779
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/pyntnclick/tests/game_logic_utils.py	Tue Jan 15 22:56:30 2013 +0200
@@ -0,0 +1,91 @@
+import unittest
+
+import pygame.display
+
+import pyntnclick.state
+
+# Disable alpha conversion which requires a screen
+import pyntnclick.resources
+pyntnclick.resources.Resources.CONVERT_ALPHA = False
+
+
+class GameLogicTestCase(unittest.TestCase):
+    CURRENT_SCENE = None
+    GAME_DESCRIPTION_CLASS = None
+
+    def setUp(self):
+        # Events require us to initialize the display
+        pygame.display.init()
+
+        self.game_description = self.GAME_DESCRIPTION_CLASS()
+        self.state = self.game_description.initial_state()
+        self.state.current_scene = self.state.scenes[self.CURRENT_SCENE]
+
+        # We aren't handling events, monkey patch change_scene
+        def change_scene(name):
+            self.state.current_scene = self.state.scenes[name]
+        self.state.change_scene = change_scene
+
+    def tearDown(self):
+        for item in self.state.items.values():
+            if isinstance(item, pyntnclick.state.CloneableItem):
+                type(item)._counter = 0
+
+    def set_game_data(self, key, value, thing=None):
+        gizmo = self.state.current_scene
+        if thing is not None:
+            gizmo = gizmo.things[thing]
+        gizmo.set_data(key, value)
+
+    def assert_game_data(self, key, value, thing=None, scene=None, detail=None):
+        gizmo = self.state.current_scene
+        if scene is not None:
+            gizmo = self.state.scenes[scene]
+        if detail is not None:
+            gizmo = self.state.detail_views[detail]
+        if thing is not None:
+            gizmo = gizmo.things[thing]
+        self.assertEquals(value, gizmo.get_data(key))
+
+    def assert_inventory_item(self, item, in_inventory=True):
+        self.assertEquals(in_inventory, self.state.is_in_inventory(item))
+
+    def assert_scene_thing(self, thing, in_scene=True):
+        self.assertEquals(in_scene, thing in self.state.current_scene.things)
+
+    def assert_detail_thing(self, thing, in_detail=True):
+        return
+        self.assertEquals(in_detail, thing in self.state.current_detail.things)
+
+    def assert_item_exists(self, item, exists=True):
+        self.assertEquals(exists, item in self.state.items)
+
+    def assert_current_scene(self, scene):
+        self.assertEquals(scene, self.state.current_scene.name)
+
+    def handle_result(self, result):
+        if result is None:
+            return None
+        if hasattr(result, 'process'):
+            if result.detail_view:
+                self.state.show_detail(result.detail_view)
+            return result
+        return [self.handle_result(r) for r in result]
+
+    def interact_thing(self, thing, item=None, detail=None):
+        item_obj = None
+        if item is not None:
+            self.assert_inventory_item(item)
+            item_obj = self.state.items[item]
+        thing_container = self.state.current_scene
+        if detail is not None:
+            thing_container = self.state.detail_views[detail]
+        result = thing_container.things[thing].interact(item_obj)
+        return self.handle_result(result)
+
+    def interact_item(self, target_item, item):
+        self.assert_inventory_item(target_item)
+        item_obj = self.state.items[item]
+        target_obj = self.state.items[target_item]
+        result = target_obj.interact(item_obj)
+        return self.handle_result(result)