More state management.

[naja.git] / naja / tests / test_player.py

diff --git a/naja/tests/test_player.py b/naja/tests/test_player.py
index 8373eee2d53a5cc35d9049854646bf6a0ea947d8..c748c7295827b9e1355c6cb3f0e6182f049585b2 100644 (file)
--- a/naja/tests/test_player.py
+++ b/naja/tests/test_player.py
@@ -1,7 +1,7 @@
 from unittest import TestCase
 
 from naja.constants import BITS
-from naja.player import PlayerBits
+from naja.player import PlayerBits, Player
 
 
 class TestPlayerBits(TestCase):
@@ -52,35 +52,57 @@ class TestPlayerBits(TestCase):
 
     def test_check_bits(self):
         bits = PlayerBits(0x03)
-        self.assertEqual(bits.check_bits(BITS.NORTH), True)
-        self.assertEqual(bits.check_bits(BITS.NORTH, BITS.SOUTH), True)
-        self.assertEqual(bits.check_bits(BITS.CYAN), False)
-        self.assertEqual(bits.check_bits(BITS.CYAN, BITS.MSB), False)
+        self.assertEqual(bits.check_bits([BITS.NORTH]), True)
+        self.assertEqual(bits.check_bits([BITS.NORTH, BITS.SOUTH]), True)
+        self.assertEqual(bits.check_bits([BITS.CYAN]), False)
+        self.assertEqual(bits.check_bits([BITS.CYAN, BITS.MSB]), False)
         self.assertEqual(
-            bits.check_bits(BITS.NORTH, BITS.SOUTH, BITS.CYAN), False)
+            bits.check_bits([BITS.NORTH, BITS.SOUTH, BITS.CYAN]), False)
 
     def test_set_bits(self):
         bits = PlayerBits(0x03)
         self.assertEqual(bits._bits, 0x03)
-        bits.set_bits(BITS.NORTH, BITS.CYAN)
+        bits.set_bits([BITS.NORTH, BITS.CYAN])
         self.assertEqual(bits._bits, 0x13)
-        bits.set_bits(BITS.NORTH, BITS.CYAN, BITS.MSB)
+        bits.set_bits([BITS.NORTH, BITS.CYAN, BITS.MSB])
         self.assertEqual(bits._bits, 0x93)
 
     def test_clear_bits(self):
         bits = PlayerBits(0x03)
         self.assertEqual(bits._bits, 0x03)
-        bits.clear_bits(BITS.NORTH, BITS.CYAN)
+        bits.clear_bits([BITS.NORTH, BITS.CYAN])
         self.assertEqual(bits._bits, 0x02)
-        bits.clear_bits(BITS.NORTH, BITS.CYAN, BITS.MSB)
+        bits.clear_bits([BITS.NORTH, BITS.CYAN, BITS.MSB])
         self.assertEqual(bits._bits, 0x02)
 
     def test_toggle_bits(self):
         bits = PlayerBits(0x03)
         self.assertEqual(bits._bits, 0x03)
-        bits.toggle_bits(BITS.NORTH, BITS.CYAN)
+        bits.toggle_bits([BITS.NORTH, BITS.CYAN])
         self.assertEqual(bits._bits, 0x12)
-        bits.toggle_bits(BITS.NORTH, BITS.CYAN)
+        bits.toggle_bits([BITS.NORTH, BITS.CYAN])
         self.assertEqual(bits._bits, 0x03)
-        bits.toggle_bits(BITS.NORTH, BITS.CYAN, BITS.MSB)
+        bits.toggle_bits([BITS.NORTH, BITS.CYAN, BITS.MSB])
         self.assertEqual(bits._bits, 0x92)
+
+
+class TestPlayer(TestCase):
+    def test_new_player(self):
+        player = Player(0x0f, (0, 1))
+        self.assertEqual(player.bits.bits, 0x0f)
+        self.assertEqual(player.position, (0, 1))
+
+    def test_import_player(self):
+        player = Player.import_player({
+            'bits': 0xaa,
+            'position': [1, 2],
+        })
+        self.assertEqual(player.bits.bits, 0xaa)
+        self.assertEqual(player.position, (1, 2))
+
+    def test_export_player(self):
+        player = Player(0xaa, (1, 2))
+        self.assertEqual(player.export(), {
+            'bits': 0xaa,
+            'position': [1, 2],
+        })