Some improvements to allow easier mocking. Added a few tests.

fdChasm · Feb 26, 2014 · 2c74296 · 2c74296
1 parent 0f18cc2
commit 2c74296
Show file tree

Hide file tree

Showing 7 changed files with 564 additions and 465 deletions.
diff --git a/src/spyd/game/room/room.py b/src/spyd/game/room/room.py
@@ -191,6 +191,9 @@ def is_teammode(self):
     def mode_name(self):
         return self._map_mode_state.mode_name
 
+    def get_map_names(self):
+        return self._map_mode_state.get_map_names()
+
     def is_name_duplicate(self, name):
         return self._players.is_name_duplicate(name)
 

diff --git a/src/spyd/protocol/__init__.py b/src/spyd/protocol/__init__.py
@@ -1 +1 @@
-from . import server_write_helper as swh
+from server_write_helper import swh