Make data layer pass post/save test

django/website/data_layer/handlers.py

+from data_layer import models
 
-def post_message(message):
-    pass
+
+class Message(object):
+
+    @classmethod
+    def create(cls, message):
+        models.Message(**message).save()

django/website/data_layer/migrations/0001_initial.py

+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import models, migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Message',
+            fields=[
+                ('id', models.AutoField(verbose_name='ID', serialize=False, auto_created=True, primary_key=True)),
+                ('created', models.DateTimeField(auto_now_add=True)),
+                ('body', models.TextField()),
+                ('timestamp', models.DateTimeField()),
+            ],
+            options={
+                'abstract': False,
+            },
+        ),
+    ]

django/website/data_layer/models.py

+from django.db import models
+
+
+class DataLayerModel(models.Model):
+    created = models.DateTimeField(auto_now_add=True)
+
+    class Meta:
+        abstract = True
+
+class Message(DataLayerModel):
+    body = models.TextField()
+    timestamp = models.DateTimeField()

django/website/data_layer/tests/handler_tests.py

 from __future__ import unicode_literals, absolute_import
-from datetime import datetime
 
-from django.test.testcases import SimpleTestCase
+from django.test import TestCase
+from django.utils import timezone
 
 from data_layer import handlers
+from data_layer import models
 
 
-class HandlerTests(SimpleTestCase):
+class HandlerTests(TestCase):
 
-    def test_message_post_handler(self):
-        message = dict(body="Text", timestamp=datetime.now())
-        handlers.post_message(message)
+    def test_message_create(self):
+        now = timezone.now().replace(
+            microsecond=0  # MySQL discards microseconds
+        )
+        inmessage = dict(body="Text", timestamp=now)
+
+        handlers.Message.create(inmessage)
+
+        outmessage = models.Message.objects.get()
+        self.assertEqual(outmessage.body, "Text")
+        self.assertEqual(outmessage.timestamp, now)

django/website/settings.py

@@ -134,6 +134,7 @@ LOCAL_APPS = (
     'users',
     'chn_spreadsheet',
     'transport',
+    'data_layer',
 )
 
 INSTALLED_APPS = DJANGO_APPS + THIRD_PARTY_APPS + LOCAL_APPS