Add data model list handler and tests

django/website/data_layer/handlers.py

@@ -6,3 +6,8 @@ class Message(object):
     @classmethod
     def create(cls, message):
         models.Message(**message).save()
+
+    @classmethod
+    def list(cls):
+        # TODO: I think we should probably use a DRF serializer here
+        return models.Message.objects.values().iterator()

django/website/data_layer/tests/handler_tests.py

@@ -20,3 +20,15 @@ class HandlerTests(TestCase):
         outmessage = models.Message.objects.get()
         self.assertEqual(outmessage.body, "Text")
         self.assertEqual(outmessage.timestamp, now)
+
+    def test_message_list_empty(self):
+        messages = handlers.Message.list()
+        self.assertEqual(len(list(messages)), 0)
+
+    def test_message_list_one(self):
+        models.Message(body="Test").save()
+
+        [message] = handlers.Message.list()
+
+        self.assertEqual(message['body'], "Test")
+