|
|
# Django imports.
|
|
|
from django.test import TestCase
|
|
|
|
|
|
# Application imports.
|
|
|
from conntrackt.forms import EntityForm
|
|
|
|
|
|
|
|
|
class EntityFormTest(TestCase):
|
|
|
"""
|
|
|
Tests for the custom Entity model form.
|
|
|
"""
|
|
|
|
|
|
def test_styling(self):
|
|
|
"""
|
|
|
Test that the form styling is set-up correctly.
|
|
|
"""
|
|
|
|
|
|
form = EntityForm()
|
|
|
|
|
|
self.assertIn("span6", form.fields["name"].widget.attrs["class"])
|
|
|
self.assertIn("span6", form.fields["description"].widget.attrs["class"])
|
|
|
self.assertIn("span6", form.fields["project"].widget.attrs["class"])
|
|
|
self.assertIn("span6", form.fields["location"].widget.attrs["class"])
|