Mercurial > pyntnclick

annotate gamelib/tests/game_logic_utils.py @ 247:0bad554d0926

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
More tests, some cryo room fixes and docs in the build.
author Jeremy Thurgood <firxen@gmail.com>
date Fri, 27 Aug 2010 13:32:27 +0200
parents 30f2308c1efc
children dfc89bc64fdb
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
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
4
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
5 from gamelib import state
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
6
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
7 from pygame.locals import SWSURFACE
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
8 from gamelib.constants import SCREEN
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
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
11 # 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
12 pygame.display.init()
190
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
13 pygame.font.init()
60
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
14 pygame.display.set_mode(SCREEN, SWSURFACE)
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
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
17 class GameLogicTestCase(unittest.TestCase):
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
18 CURRENT_SCENE = None
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
19
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
20 def setUp(self):
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
21 self.state = state.initial_state()
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
22 self.state.set_current_scene(self.CURRENT_SCENE)
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
23
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
24 def set_game_data(self, key, value, thing=None, scene=None):
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
25 if scene is None:
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
26 scene = self.CURRENT_SCENE
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
27 gizmo = self.state.scenes[scene]
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
28 if thing is not None:
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
29 gizmo = gizmo.things[thing]
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
30 gizmo.set_data(key, value)
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
31
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
32 def assert_game_data(self, key, value, thing=None, scene=None):
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
33 if scene is None:
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
34 scene = self.CURRENT_SCENE
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
35 gizmo = self.state.scenes[scene]
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
36 if thing is not None:
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
37 gizmo = gizmo.things[thing]
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
38 self.assertEquals(value, gizmo.get_data(key))
90b32447b239 Some test refactorage.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
39
67
6b0f7364f3bf Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents: 60
diff changeset
40 def assert_inventory_item(self, item, in_inventory=True):
6b0f7364f3bf Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents: 60
diff changeset
41 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
42
190
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
43 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
44 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
45
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
46 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
47 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
48
247
0bad554d0926 More tests, some cryo room fixes and docs in the build.
Jeremy Thurgood <firxen@gmail.com>
parents: 190
diff changeset
49 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
50 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
51
190
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
52 def interact_thing(self, thing, item=None, detail=False):
67
6b0f7364f3bf Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents: 60
diff changeset
53 item_obj = None
6b0f7364f3bf Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents: 60
diff changeset
54 if item is not None:
6b0f7364f3bf Inventory-related game state.
Jeremy Thurgood <firxen@gmail.com>
parents: 60
diff changeset
55 item_obj = self.state.items[item]
190
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
56 thing_container = self.state.current_scene
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
57 if detail:
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
58 thing_container = self.state.current_detail
30f2308c1efc Fix tests and add a (currently unhooked) laser welder.
Jeremy Thurgood <firxen@gmail.com>
parents: 67
diff changeset
59 return thing_container.things[thing].interact(item_obj)