Sound

requirements.txt

@@ -3,4 +3,6 @@ nose==1.3.7
 pylint==1.7.4
 coverage==4.4.2
 colorama==0.3.9
-opencensus-ext-azure==1.1.1
+opencensus-ext-azure==1.1.1
+playsound
+PyObjC

tests/test_battleship.py

@@ -1,10 +1,40 @@
 import unittest
 
-from torpydo.battleship import parse_position
+from torpydo.battleship import parse_position, is_fleet_down, get_random_position
+from torpydo.ship import Color, Letter, Position, Ship
+from unittest.mock import patch
 
 class TestBattleship(unittest.TestCase):
+    def setUp(self):
+        self.ships = []
+        self.ships.append(init_ship(Ship("Test", 2, Color.RED), [Position(Letter.A, 1), Position(Letter.A, 2)]))
+
     def test_parse_position_true(self):
         self.assertTrue(parse_position("A1"))
 
+    def test_is_fleet_down(self):
+        self.assertFalse(is_fleet_down(self.ships))
+        self.ships[0].is_sunk = True
+        self.assertTrue(is_fleet_down(self.ships))
+
+    @patch("torpydo.battleship.NUMBER_ROWS", 2)
+    @patch("torpydo.battleship.NUMBER_COL", 2)
+    def test_random_duplicate_position(self):
+        for i in range(10):
+            board = []
+            x = [
+                get_random_position(board),
+                get_random_position(board),
+                get_random_position(board),
+                get_random_position(board),
+            ]
+            y = list(set(x))
+            self.assertTrue(len(y) == 4)
+
+def init_ship(ship: Ship, positions: list):
+    ship.positions = positions
+
+    return ship
+
 if '__main__' == __name__:
     unittest.main()

tests/test_colours.py

@@ -0,0 +1,8 @@
+from torpydo.battleship import right_colour
+from colorama import Fore
+
+def test_right_colour_hit():
+    assert right_colour(True) == Fore.RED
+
+def test_right_colour_water():
+    assert right_colour(False) == Fore.BLUE

tests/test_game_controller.py

@@ -10,6 +10,7 @@ def setUp(self):
 
     def test_check_is_hit_true(self):
         self.assertTrue(GameController.check_is_hit(self.ships, Position(Letter.A, 1)))
+        self.assertTrue(self.ships[0].positions[0].is_shot)
 
     def test_check_is_hit_false(self):
         self.assertFalse(GameController.check_is_hit(self.ships, Position(Letter.B, 1)))

tests/test_random.py

@@ -0,0 +1,44 @@
+from torpydo.battleship import overlaps, set_direction, set_forward_rear
+from torpydo.ship import Letter, Position, Ship, Color
+
+def test_not_overlaps():
+    ship = Ship("test",5,Color.RED)
+    ship.add_position("B1")
+    assert overlaps(
+        [
+            Position(Letter.A,1),
+            Position(Letter.A,2)
+        ],
+        fleet=[
+            ship
+        ]
+    ) == False
+
+def test_overlaps():
+    ship = Ship("test",5, Color.RED)
+    ship.add_position("A2")
+    assert overlaps(
+        [
+            Position(Letter.A,1),
+            Position(Letter.A,2)
+        ],
+        fleet=[
+            ship
+        ]
+    ) == True
+
+def test_set_forward_read():
+    assert set_forward_rear(5,3,6) == 0
+    assert set_forward_rear(5,1,7) == 1
+    assert set_forward_rear(2,3,8) == 1
+    assert set_forward_rear(3,5,7) == 1
+    assert set_forward_rear(3,5,6) == -1
+
+def test_set_direction():
+    axis, st_value = set_direction(Position(Letter.A,1))
+    assert axis == "horizontal"
+    assert st_value == 1
+    axis, st_value = set_direction(Position(Letter.A,2))
+    assert axis == "vertical"
+    assert st_value == 2
+

tests/test_ship.py

@@ -0,0 +1,23 @@
+import unittest
+
+from torpydo.ship import Color, Letter, Position, Ship
+from torpydo.game_controller import GameController
+
+class TestBattleship(unittest.TestCase):
+    def setUp(self):
+        self.ships = []
+        self.ships.append(init_ship(Ship("Test", 2, Color.RED), [Position(Letter.A, 1), Position(Letter.A, 2)]))
+
+    def test_check_sunk(self):
+        self.assertFalse(self.ships[0].is_sunk)
+        GameController.check_is_hit(self.ships, Position(Letter.A, 1))
+        self.assertFalse(self.ships[0].is_sunk)
+        GameController.check_is_hit(self.ships, Position(Letter.A, 2))
+        self.assertTrue(self.ships[0].is_sunk)
+
+def init_ship(ship: Ship, positions: list):
+    ship.positions = positions
+
+    return ship
+if '__main__' == __name__:
+    unittest.main()

tests/test_sound.py

@@ -0,0 +1,12 @@
+from unittest import mock
+from torpydo.battleship import is_hit_sound
+
+def test_call_right_sound_fire():
+    with mock.patch("torpydo.battleship.playsound") as mf:
+        is_hit_sound(True)
+        mf.assert_called_once_with('sound/explosion2.mp3')
+
+def test_call_right_sound_water():
+    with mock.patch("torpydo.battleship.playsound") as mf:
+        is_hit_sound(False)
+        mf.assert_called_once_with('sound/splash2.mp3', block=False)