ewrite transport tests using pytest

django/website/transport/tests/message_tests.py

 from __future__ import unicode_literals, absolute_import
-from mock import patch
-
-from django.test import TestCase
+import pytest
 
+from data_layer.serializers import ItemSerializer
+from data_layer.tests.factories import ItemFactory
 from transport import data_layer_transport as dl
 
 
-class TransportLayerMessageTests(TestCase):
+@pytest.mark.django_db
+def test_get_messages_exists():
+    messages = dl.get_messages()
+    assert messages == []
+
+@pytest.mark.django_db
+def test_get_messages_returns_items():
+    item = ItemFactory(body="test")
 
-    @patch('data_layer.handlers.Message.list')
-    def test_get_messages_uses_list(self, list):
-        list.return_value = []
-        messages = dl.get_messages()
-        self.assertEqual(messages, [])
-        list.assert_called_with()
+    messages = dl.get_messages()
+    [message] = messages
+    assert message['body'] == 'test'
 
-    @patch('data_layer.handlers.Message.create')
-    def test_store_message_uses_create(self, create):
-        message = {}
-        dl.create_message(message)
-        create.assert_called_with(message)
+@pytest.mark.xfail
+def test_store_message_uses_create():
+    #message = {}
+    #:l.create_message(message)
+    #create.assert_called_with(message)
+    assert False