[PROGRESS]

Author: ultimatecoder

tests/__pycache__/__init__.cpython-35.pyc

@@ -0,0 +1,33 @@
+from unittest import TestCase
+
+from yamq import utils
+
+class DemoAutoID(utils.AutoID):
+    pass
+
+
+class TestAutoID(TestCase):
+
+    def test_auto_id_works_okay(self):
+        """Combined tests in one method.
+
+        AutoID increments the id according to number of objects created. There
+        is no guarentee that the tests will perform in certain order. It might
+        be diffecult assert with pre-fixed ID number. A quick work around is
+        to test everything in one method so that you have grip over the expected
+        ID number.
+        """
+
+
+        demo_obj = DemoAutoID()
+        self.assertTrue(hasattr(demo_obj, '_id'))
+        self.assertTrue(isinstance(demo_obj._id, int))
+        self.assertEqual(demo_obj._id, 0)
+
+        another_obj = DemoAutoID()
+        self.assertEqual(another_obj._id, 1)
+
+        del another_obj
+
+        yet_another_obj = DemoAutoID()
+        self.assertEqual(yet_another_obj._id, 2)

tests/unit/__pycache__/test_auto_id.cpython-35.pyc

@@ -0,0 +1,33 @@
+from unittest import TestCase
+
+from yamq import message
+
+
+class TestMessage(TestCase):
+
+    def test_message_creation(self):
+        demo_message = message.Message("test message", "text/text")
+
+        self.assertEqual(demo_message.message, "test message")
+        self.assertEqual(demo_message.content_type, "text/text")
+        self.assertTrue(hasattr(demo_message, "_id"))
+        self.assertTrue(isinstance(demo_message._id, int))
+        self.assertIn(demo_message._id, message.Message._messages)
+
+    def test_message_is_deleted(self):
+        demo_message = message.Message("test message", "text/text")
+
+        self.assertIn(demo_message._id, message.Message._messages)
+
+        demo_id = demo_message._id
+        demo_message.delete()
+
+        with self.assertRaises(KeyError):
+            message.Message._messages[demo_id]
+
+    def test_message_get(self):
+        demo_message = message.Message("test message", "text/text")
+
+        founded_demo_message = message.Message.get(demo_message._id)
+
+        self.assertIs(demo_message, founded_demo_message)

tests/unit/__pycache__/test_message.cpython-35.pyc

@@ -32,15 +32,26 @@ def test_dump_error(self):
             stomp.dumps("invalid")
 
     def test_loads_okay(self):
-        response = (
-            'SUBSCRIBE\r\n'
-            'ack:client\r\n'
-            'destination:/queue/foo\r\n'
-            '\r\n'
-            '\r\n'
-            '\x00'
-        )
-        self.assertIs(type(stomp.loads(response)), stomp.Frame)
+        responses = [
+            (
+                'SUBSCRIBE\r\n'
+                'ack:client\r\n'
+                'destination:/queue/foo\r\n'
+                '\r\n'
+                '\r\n'
+                '\x00'
+            ),
+            (
+                'SEND\n'
+                'content-length:11\n'
+                'destination:/queue/test\n'
+                '\n'
+                'hello world'
+                '\x00'
+            )
+        ]
+        for response in responses:
+            self.assertIs(type(stomp.loads(response)), stomp.Frame)
 
     def test_loads_error(self):
         response = (