diff gamelib/tests/game_logic_utils.py @ 60:90b32447b239

Some test refactorage.
author Jeremy Thurgood <firxen@gmail.com>
date Mon, 23 Aug 2010 17:16:05 +0200
parents
children 6b0f7364f3bf
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/gamelib/tests/game_logic_utils.py	Mon Aug 23 17:16:05 2010 +0200
@@ -0,0 +1,43 @@
+import unittest
+
+import pygame
+
+from gamelib import state
+
+from pygame.locals import SWSURFACE
+from gamelib.constants import SCREEN
+
+
+# We need this stuff set up so we can load images and whatnot.
+pygame.display.init()
+pygame.display.set_mode(SCREEN, SWSURFACE)
+
+
+class GameLogicTestCase(unittest.TestCase):
+    CURRENT_SCENE = None
+
+    def setUp(self):
+        self.state = state.initial_state()
+        self.state.set_current_scene(self.CURRENT_SCENE)
+
+    def set_game_data(self, key, value, thing=None, scene=None):
+        if scene is None:
+            scene = self.CURRENT_SCENE
+        gizmo = self.state.scenes[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):
+        if scene is None:
+            scene = self.CURRENT_SCENE
+        gizmo = self.state.scenes[scene]
+        if thing is not None:
+            gizmo = gizmo.things[thing]
+        self.assertEquals(value, gizmo.get_data(key))
+
+    def interact_thing(self, thing_name, item_name=None):
+        item = None
+        if item_name is not None:
+            item = self.state.items[item_name]
+        self.state.scenes[self.CURRENT_SCENE].things[thing_name].interact(item)