Added initial model tests

This commit is contained in:
David Taylor
2015-08-05 17:39:19 +03:00
committed by Tom Price
parent 56a2ea68fb
commit e56cff17fa
2 changed files with 18 additions and 3 deletions

18
forms/test_models.py Normal file
View File

@@ -0,0 +1,18 @@
from django.test import TestCase
from forms import models
class FormModelsTestCase(TestCase):
def setUp(self):
self.schemaType1 = models.Type.objects.create(name="Test Type",description="Description of a test type")
models.Schema.objects.create(start_at='2014-03-01',comment='test1', schema_type=self.schemaType1)
models.Schema.objects.create(start_at='2016-03-01',comment='test2', schema_type=self.schemaType1)
self.schemaType2 = models.Type.objects.create(name="Test Type 2",description="Description of the second test type")
models.Schema.objects.create(start_at='2014-03-01',comment='test3', schema_type=self.schemaType2)
models.Schema.objects.create(start_at='2016-03-01',comment='test4', schema_type=self.schemaType2)
def test_find_correct(self):
r = models.Schema.objects.find_schema(self.schemaType1, '2015-03-01')
self.assertEqual(r.comment, 'test1')
r = models.Schema.objects.find_schema(self.schemaType1, '2016-03-01')
self.assertEqual(r.comment, 'test2')