1 files changed, 89 insertions, 0 deletions
diff --git a/tests/test_gameState.py b/tests/test_gameState.py
new file mode 100644
index 0000000..638e269
--- /dev/null
+++ b/tests/test_gameState.py
@@ -0,0 +1,89 @@
+"""Tests for the input/output component."""
+
+import unittest
+
+from imago.data.enums import Player
+from imago.gameLogic.gameState import GameState
+
+class TestGameState(unittest.TestCase):
+    """Test GameState component."""
+
+    def testCurrentPlayer(self):
+        """Test simple commands."""
+        size = 9
+        state = GameState(size)
+
+        self.assertEqual(state.getCurrentPlayer(), Player.BLACK)
+        self.assertEqual(state.getPlayerCode(), 'B')
+
+        state.playMove(0, 0)
+        self.assertEqual(state.getCurrentPlayer(), Player.WHITE)
+        self.assertEqual(state.getPlayerCode(), 'W')
+
+    def testPlays(self):
+        """Test simple commands."""
+        size = 3
+        state = GameState(size)
+
+        self.assertEqual(state.getBoard().getBoard(),
+                         [
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY],
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY],
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY]
+                         ])
+
+        state.playMove(1, 1)
+        self.assertEqual(state.getBoard().getBoard(),
+                         [
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY],
+                             [Player.EMPTY, Player.BLACK, Player.EMPTY],
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY]
+                         ])
+
+        state.playPass()
+        self.assertEqual(state.getBoard().getBoard(),
+                         [
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY],
+                             [Player.EMPTY, Player.BLACK, Player.EMPTY],
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY]
+                         ])
+
+        state.undo()
+        self.assertEqual(state.getBoard().getBoard(),
+                         [
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY],
+                             [Player.EMPTY, Player.BLACK, Player.EMPTY],
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY]
+                         ])
+
+        state.undo()
+        self.assertEqual(state.getBoard().getBoard(),
+                         [
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY],
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY],
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY]
+                         ])
+
+        state.passForPlayer(Player.WHITE)
+        self.assertEqual(state.getBoard().getBoard(),
+                         [
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY],
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY],
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY]
+                         ])
+
+
+        self.assertFalse(state.playMove(-1, -1))
+        self.assertEqual(state.getBoard().getBoard(),
+                         [
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY],
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY],
+                             [Player.EMPTY, Player.EMPTY, Player.EMPTY]
+                         ])
+
+        state.lastMove = None
+        self.assertEqual(state.getCurrentPlayer(), Player.BLACK)
+        self.assertRaises(RuntimeError, state.playMove, 0, 0)
+
+if __name__ == '__main__':
+    unittest.main()