Skip to content
Snippets Groups Projects
Commit 52c39ce5 authored by Mark Skipper's avatar Mark Skipper
Browse files

Make data layer pass post/save test

parent 26debcd2
No related branches found
No related tags found
No related merge requests found
from data_layer import models
def post_message(message):
pass
class Message(object):
@classmethod
def create(cls, message):
models.Message(**message).save()
# -*- 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,
},
),
]
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()
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)
......@@ -134,6 +134,7 @@ LOCAL_APPS = (
'users',
'chn_spreadsheet',
'transport',
'data_layer',
)
INSTALLED_APPS = DJANGO_APPS + THIRD_PARTY_APPS + LOCAL_APPS
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment