Mercurial > boomslang

annotate gamelib/tests/game_logic_utils.py @ 254:ca0c2875ad8f

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
More test fixes.
author Jeremy Thurgood <firxen@gmail.com>
date Fri, 27 Aug 2010 17:35:15 +0200
parents dfc89bc64fdb
children a1ffc919d0c1
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
60
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
1 import unittest
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
2
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
3 import pygame
252
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
4 from pygame.locals import SWSURFACE
60
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
5
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
6 from gamelib import state
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
7 from gamelib.constants import SCREEN
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
8
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
9
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
10 # We need this stuff set up so we can load images and whatnot.
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
11 pygame.display.init()
190
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
12 pygame.font.init()
60
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
13 pygame.display.set_mode(SCREEN, SWSURFACE)
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
14
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
15
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
16 class GameLogicTestCase(unittest.TestCase):
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
17 CURRENT_SCENE = None
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
18
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
19 def setUp(self):
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
20 self.state = state.initial_state()
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
21 self.state.set_current_scene(self.CURRENT_SCENE)
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
22
254
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
23 def tearDown(self):
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
24 for item in self.state.items.values():
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
25 if isinstance(item, state.CloneableItem):
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
26 type(item)._counter = 0
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
27
252
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
28 def set_game_data(self, key, value, thing=None):
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
29 gizmo = self.state.current_scene
60
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
30 if thing is not None:
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
31 gizmo = gizmo.things[thing]
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
32 gizmo.set_data(key, value)
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
33
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
34 def assert_game_data(self, key, value, thing=None, scene=None):
252
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
35 gizmo = self.state.current_scene
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
36 if scene is not None:
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
37 gizmo = self.state.scenes[scene]
60
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
38 if thing is not None:
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
39 gizmo = gizmo.things[thing]
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
40 self.assertEquals(value, gizmo.get_data(key))
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
41
67
6b0f7364f3bf Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents: 60
diff changeset
42 def assert_inventory_item(self, item, in_inventory=True):
6b0f7364f3bf Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents: 60
diff changeset
43 self.assertEquals(in_inventory, self.state.items[item] in self.state.inventory)
6b0f7364f3bf Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents: 60
diff changeset
44
190
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
45 def assert_scene_thing(self, thing, in_scene=True):
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
46 self.assertEquals(in_scene, thing in self.state.current_scene.things)
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
47
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
48 def assert_detail_thing(self, thing, in_detail=True):
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
49 self.assertEquals(in_detail, thing in self.state.current_detail.things)
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
50
247
0bad554d0926 More tests, some cryo room fixes and docs in the build.
Jeremy Thurgood <firxen@gmail.com>
parents: 190
diff changeset
51 def assert_item_exists(self, item, exists=True):
0bad554d0926 More tests, some cryo room fixes and docs in the build.
Jeremy Thurgood <firxen@gmail.com>
parents: 190
diff changeset
52 self.assertEquals(exists, item in self.state.items)
0bad554d0926 More tests, some cryo room fixes and docs in the build.
Jeremy Thurgood <firxen@gmail.com>
parents: 190
diff changeset
53
252
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
54 def assert_current_scene(self, scene):
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
55 self.assertEquals(scene, self.state.current_scene.name)
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
56
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
57 def interact_thing(self, thing, item=None):
67
6b0f7364f3bf Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents: 60
diff changeset
58 item_obj = None
6b0f7364f3bf Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents: 60
diff changeset
59 if item is not None:
254
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
60 self.assert_inventory_item(item)
67
6b0f7364f3bf Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents: 60
diff changeset
61 item_obj = self.state.items[item]
252
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
62 thing_container = self.state.current_detail or self.state.current_scene
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
63 result = thing_container.things[thing].interact(item_obj)
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
64 if result and result.detail_view:
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
65 self.state.set_current_detail(result.detail_view)
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
66 return result
dfc89bc64fdb Start of walkthrough "unit test" and associated fixes and tweaks.
Jeremy Thurgood <firxen@gmail.com>
parents: 247
diff changeset
67
254
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
68 def interact_item(self, target_item, item):
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
69 self.assert_inventory_item(target_item)
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
70 item_obj = self.state.items[item]
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
71 target_obj = self.state.items[target_item]
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
72 result = target_obj.interact(item_obj, self.state)
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
73 if result and result.detail_view:
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
74 self.state.set_current_detail(result.detail_view)
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
75 return result
ca0c2875ad8f More test fixes.
Jeremy Thurgood <firxen@gmail.com>
parents: 252
diff changeset
76