Finished code and tests.

Author: basilfx

tests/__init__.py

@@ -0,0 +1,186 @@
+import unittest
+
+import tinylink
+
+class DummyHandle(object):
+    """
+    Dummy handler, so the TinyLink class can exchange data with itself.
+    """
+
+    def __init__(self):
+        self.buffer = bytearray()
+        self.index = 0
+        self.length = 0
+
+    def send(self, data):
+        self.buffer.extend(data)
+        self.length += len(data)
+
+        # Return number of bytes written
+        return len(data)
+
+    def recv(self, count):
+        data = self.buffer[self.index:min(self.length, self.index+count)]
+        self.index += len(data)
+
+        # Return data
+        return data
+
+class TinyLinkTest(unittest.TestCase):
+    """
+    Test TinyLink
+    """
+
+    def test_basic(self):
+        """
+        Test send/receive by using a dummy handle.
+        """
+
+        handle = DummyHandle()
+        link = tinylink.TinyLink(handle)
+
+        message = "Hello, this is a test"
+        size = link.write(message)
+
+        self.assertEqual(size, tinylink.LEN_PREAMBLE + tinylink.LEN_HEADER +
+            tinylink.LEN_BODY + len(message))
+
+        # Read `size' bytes to receive the full frame, test it partially
+        link.read(1)
+        link.read(1)
+        link.read(1)
+        frames = link.read(size - 3)
+
+        self.assertEqual(len(frames), 1)
+        self.assertEqual(frames[0].data, message)
+
+    def test_reset(self):
+        """
+        Test reset feature.
+        """
+
+        handle = DummyHandle()
+        link = tinylink.TinyLink(handle)
+
+        size = link.reset()
+        frames = link.read(size)
+
+        self.assertEqual(len(frames), 1)
+        self.assertIsInstance(frames[0], tinylink.ResetFrame)
+
+    def test_multiple(self):
+        """
+        Test multiple messages
+        """
+
+        handle = DummyHandle()
+        link = tinylink.TinyLink(handle)
+
+        size = 0
+
+        for i in xrange(5):
+            size += link.write(chr(97 + i))
+
+        frames = link.read(size)
+
+        self.assertEqual(len(frames), 5)
+
+        for i in xrange(5):
+            self.assertEqual(frames[i].data, chr(97 + i))
+
+    def test_sync(self):
+        """
+        Test preamble synchronization.
+        """
+
+        handle = DummyHandle()
+        link = tinylink.TinyLink(handle)
+
+        garbage = "Garbage here that doesn't synchronize."
+        message = "Hi!"
+
+        size = handle.send(garbage) + link.write(message)
+        frames = link.read(size)
+
+        self.assertEqual(len(frames), 1)
+        self.assertEqual(frames[0].data, message)
+
+    def test_sync_small(self):
+        """
+        Test preamble synchronization with smaller buffer.
+        """
+
+        handle = DummyHandle()
+        link = tinylink.TinyLink(handle, max_length=4)
+
+        garbage = "Garbage here that doesn't synchronize."
+        message = "Hi!"
+
+        size = handle.send(garbage) + link.write(message)
+        frames = link.read(size)
+
+        self.assertEqual(len(frames), 1)
+        self.assertEqual(frames[0].data, message)
+
+    def test_size_fit(self):
+        """
+        Test smaller sized TinyLink that does fit.
+        """
+
+        handle = DummyHandle()
+        link = tinylink.TinyLink(handle, max_length=4)
+
+        message = "blub"
+
+        size = link.write(message)
+        frames = link.read(size)
+
+        self.assertEqual(len(frames), 1)
+        self.assertEqual(frames[0].data, message)
+
+    def test_size_no_fit(self):
+        """
+        Test smaller sized TinyLink that does not fit.
+        """
+
+        handle = DummyHandle()
+        link = tinylink.TinyLink(handle, max_length=2)
+
+        message = "blub"
+
+        with self.assertRaises(ValueError):
+            size = link.write(message)
+
+    def test_damaged_a(self):
+        """
+        Test damaged frame (in total) that will return a DamagedFrame.
+        """
+
+        handle = DummyHandle()
+        link = tinylink.TinyLink(handle)
+
+        message = "Hello, this is a test"
+
+        size = link.write(message)
+        handle.buffer[-tinylink.LEN_CRC:] = "\x00" * tinylink.LEN_CRC
+        frames = link.read(size)
+
+        self.assertEqual(len(frames), 1)
+        self.assertEqual(frames[0].data, message)
+        self.assertIsInstance(frames[0], tinylink.DamagedFrame)
+
+    def test_damaged_a(self):
+        """
+        Test damaged frame (header) that won't return anything.
+        """
+
+        handle = DummyHandle()
+        link = tinylink.TinyLink(handle)
+
+        message = "Hello, this is a test"
+
+        size = link.write(message)
+        handle.buffer[tinylink.LEN_PREAMBLE+tinylink.LEN_HEADER-1] = "\x00"
+        frames = link.read(size)
+
+        self.assertEqual(len(frames), 0)