Changeset 164:06c681ff53aa for nagslang

Timestamp:

Sep 3, 2013, 8:47:27 AM (7 years ago)

Author:

Neil Muller <drnlmuller@…>

Branch:

default

Message:

Round-tripping through load/save shouldn't discard objects

Location:

nagslang

Files:

• level.py (modified) (4 diffs)
• tests/test_level.py (modified) (5 diffs)

nagslang/level.py

@@ -30 +30 @@
         self._glue = go.PuzzleGlue()
         self._drawables = []
+        self._game_objects = []
 
     def _get_data(self):
@@ -35 +36 @@
         with resources.get_file('levels', self.name) as f:
             return load(f)
+
+    def _dump_data(self, f):
+        # For manipulation in tests.
+        dump({
+            'size': [self.x, self.y],
+            'base_tile': self.basetile,
+            'polygons': self.polygons,
+            'game_objects': self._game_objects,
+        }, f)
 
     def load(self, space):
@@ -44 +54 @@
             for point in points:
                 self.polygons[i].append(tuple(point))
-        for game_object_dict in data.get('game_objects', []):
+        self._game_objects = data.get('game_objects', [])
+        for game_object_dict in self._game_objects:
             self._create_game_object(space, **game_object_dict)
 
@@ -84 +95 @@
             return False
         with resources.get_file('levels', self.name, mode='w') as f:
-            dump({
-                'size': [self.x, self.y],
-                'base_tile': self.basetile,
-                'polygons': self.polygons,
-            }, f)
+            self._dump_data(f)
         return True
 

nagslang/tests/test_level.py

@@ -1 +1 @@
 from unittest import TestCase
+
+from StringIO import StringIO
 
 from nagslang import game_object as go
 from nagslang.level import Level
+from nagslang.yamlish import load
 
 
@@ -16 +19 @@
         return level
 
+    def roundtrip_level(self, level):
+        newlevel = Level(level.name)
+        f = StringIO()
+        level._dump_data(f)
+        f.seek(0)
+        newlevel._get_data = lambda: load(f)
+        return newlevel
+
     def test_empty_level(self):
+        def do_test(level):
+            level.load(FakeSpace())
+            self.assertEqual((5, 10), level.get_size())
+            self.assertEqual([], level.get_walls())
+            self.assertEqual([], level.get_drawables())
+
         level = self.make_level('foo', {
             'size': [5, 10],
@@ -22 +39 @@
             'polygons': {},
         })
-        level.load(FakeSpace())
-        self.assertEqual((5, 10), level.get_size())
-        self.assertEqual([], level.get_walls())
-        self.assertEqual([], level.get_drawables())
+
+        do_test(level)
+        level2 = self.roundtrip_level(level)
+        do_test(level2)
 
     def test_level_walls(self):
+        def do_test(level):
+            level.load(FakeSpace())
+            self.assertEqual((5, 10), level.get_size())
+            self.assertEqual([[(1, 1), (2, 1), (1, 2)]], level.get_walls())
+            self.assertEqual([], level.get_drawables())
+
         level = self.make_level('foo', {
             'size': [5, 10],
@@ -35 +58 @@
             },
         })
-        level.load(FakeSpace())
-        self.assertEqual((5, 10), level.get_size())
-        self.assertEqual([[(1, 1), (2, 1), (1, 2)]], level.get_walls())
-        self.assertEqual([], level.get_drawables())
+        do_test(level)
+        level2 = self.roundtrip_level(level)
+        do_test(level2)
 
     def test_level_game_objects(self):
+        def do_test(level):
+            level.load(FakeSpace())
+            self.assertEqual((5, 10), level.get_size())
+            self.assertEqual([], level.get_walls())
+            [box, switch] = level.get_drawables()
+            self.assertTrue(isinstance(box, go.Box))
+            self.assertEqual(box.shape.body.position, (3, 3))
+            self.assertTrue(isinstance(switch, go.FloorSwitch))
+            self.assertEqual(switch.shape.body.position, (4, 4))
+
+            puzzle_bits = level._glue._components
+            self.assertEqual(['foo', 'foo_proxy'],
+                             sorted(puzzle_bits.keys()))
+            self.assertTrue(
+                isinstance(puzzle_bits['foo_proxy'], go.StateProxyPuzzler))
+            self.assertEqual('foo', puzzle_bits['foo_proxy']._state_source)
+            self.assertTrue(isinstance(puzzle_bits['foo'],
+                            go.FloorSwitchPuzzler))
+
         level = self.make_level('foo', {
             'size': [5, 10],
@@ -62 +103 @@
             ],
         })
-        level.load(FakeSpace())
-        self.assertEqual((5, 10), level.get_size())
-        self.assertEqual([], level.get_walls())
-        [box, switch] = level.get_drawables()
-        self.assertTrue(isinstance(box, go.Box))
-        self.assertEqual(box.shape.body.position, (3, 3))
-        self.assertTrue(isinstance(switch, go.FloorSwitch))
-        self.assertEqual(switch.shape.body.position, (4, 4))
 
-        puzzle_bits = level._glue._components
-        self.assertEqual(['foo', 'foo_proxy'], sorted(puzzle_bits.keys()))
-        self.assertTrue(
-            isinstance(puzzle_bits['foo_proxy'], go.StateProxyPuzzler))
-        self.assertEqual('foo', puzzle_bits['foo_proxy']._state_source)
-        self.assertTrue(isinstance(puzzle_bits['foo'], go.FloorSwitchPuzzler))
+        do_test(level)
+        level2 = self.roundtrip_level(level)
+        do_test(level2)