I04_1-156: added unit tests

Author: IsabellaRey

dls_util/image/__init__.py

@@ -1,3 +1,3 @@
 from .color import Color
 from .image import Image
-from .overlay import Overlay
+from .overlay import Overlay, TextOverlay

tests/unit_tests/test_dls_barcode/test_camera/test_plate_overlay.py

@@ -0,0 +1,61 @@
+import unittest
+from mock import MagicMock
+import time
+
+from dls_barcode.camera.plate_overlay import PlateOverlay
+
+
+class TestPlateOverlay(unittest.TestCase):
+    def setUp(self):
+        self._plate = MagicMock()
+        self._options = MagicMock()
+        self._image = MagicMock()
+
+    def test_overlay_has_default_lifetime(self):
+        # Arrange
+        expected_lifetime = 2
+
+        # Act
+        overlay = PlateOverlay(self._plate, self._options)
+
+        # Assert
+        time.sleep(expected_lifetime/2)
+        self.assertFalse(overlay.has_expired())
+        time.sleep(expected_lifetime / 2)
+        self.assertTrue(overlay.has_expired())
+
+    def test_lifetime_can_be_initialised(self):
+        # Arrange
+        lifetime = 0.2
+
+        # Act
+        overlay = PlateOverlay(self._plate, self._options, lifetime)
+
+        # Assert
+        time.sleep(lifetime)
+        self.assertTrue(overlay.has_expired())
+
+    def test_expired_overlay_is_not_drawn_on_image(self):
+        # Arrange
+        lifetime = 0.001
+        overlay = PlateOverlay(self._plate, self._options, lifetime)
+
+        # Act
+        time.sleep(lifetime)
+        overlay.draw_on_image(self._image)
+
+        # Assert
+        self._plate.draw_plate.assert_not_called()
+        self._plate.draw_pins.assert_not_called()
+
+    def test_not_expired_plate_is_drawn_on_image(self):
+        # Arrange
+        lifetime = 1000
+        overlay = PlateOverlay(self._plate, self._options, lifetime)
+
+        # Act
+        overlay.draw_on_image(self._image)
+
+        # Assert
+        self._plate.draw_plate.assert_called_once()
+        self._plate.draw_pins.assert_called_once()

tests/unit_tests/test_dls_util/test_image/test_overlay.py

@@ -0,0 +1,37 @@
+import unittest
+from mock import MagicMock
+import time
+
+from dls_util.image import TextOverlay
+
+
+class TestTextOverlay(unittest.TestCase):
+    def setUp(self):
+        self._color = MagicMock()
+
+    def test_overlay_has_default_lifetime(self):
+        # Arrange
+        expected_lifetime = 2
+
+        # Act
+        overlay = TextOverlay("blah", self._color)
+
+        # Assert
+        time.sleep(expected_lifetime/2)
+        self.assertFalse(overlay.has_expired())
+        time.sleep(expected_lifetime / 2)
+        self.assertTrue(overlay.has_expired())
+
+    def test_lifetime_can_be_initialised(self):
+        # Arrange
+        lifetime = 0.2
+
+        # Act
+        overlay = TextOverlay("blah", self._color, lifetime)
+
+        # Assert
+        time.sleep(lifetime)
+        self.assertTrue(overlay.has_expired())
+
+# Can't test the drawing method because overlay is passed an opencv image.
+# It should be refactored to get an Image object instead of creating it itself

tests/unit_tests/test_dls_util/test_message/test_message.py

@@ -0,0 +1,42 @@
+import unittest
+import time
+
+from dls_util.message import Message, MessageType
+
+
+class TestMessage(unittest.TestCase):
+    def test_message_has_default_lifetime(self):
+        # Arrange
+        expected_lifetime = 2
+
+        # Act
+        msg = Message(MessageType.INFO, "hello")
+
+        # Assert
+        time.sleep(expected_lifetime/2)
+        self.assertFalse(msg.has_expired())
+        time.sleep(expected_lifetime / 2)
+        self.assertTrue(msg.has_expired())
+
+    def test_lifetime_can_be_initialised(self):
+        # Arrange
+        lifetime = 0.1
+
+        # Act
+        msg = Message(MessageType.INFO, "hello", lifetime)
+
+        # Assert
+        time.sleep(lifetime)
+        self.assertTrue(msg.has_expired())
+
+    def test_parameters_initalised_correctly(self):
+        # Arrange
+        type = MessageType.WARNING
+        content = "hello there"
+
+        # Act
+        msg = Message(type, content)
+
+        # Assert
+        self.assertEqual(msg.type(), type)
+        self.assertEqual(msg.content(), content)

tests/unit_tests/test_dls_util/test_object_with_lifetime.py

@@ -0,0 +1,26 @@
+import unittest
+import time
+from dls_util.object_with_lifetime import ObjectWithLifetime
+
+
+class TestObjectWithLifetime(unittest.TestCase):
+    def test_object_with_life_less_than_lifetime_has_not_expired(self):
+        # Arrange
+        lifetime = 0.5
+
+        # Act
+        obj = ObjectWithLifetime(lifetime)
+
+        # Assert
+        self.assertFalse(obj.has_expired())
+
+    def test_object_with_life_greater_than_lifetime_has_expired(self):
+        # Arrange
+        lifetime = 0.5
+
+        # Act
+        obj = ObjectWithLifetime(lifetime)
+        time.sleep(lifetime)
+
+        # Assert
+        self.assertTrue(obj.has_expired())