additional tests defined

rowers/tests/test_models.py

@@ -2,13 +2,47 @@ from django.test import TestCase
 from rowers.models import *
 
 class FormTests(TestCase):
-    def SetUp(self):
+    def setUp(self):
         self.u = User.objects.create_user('john',
 				     'sander@ds.ds',
 				     'koeinsloot')
 
-        r = Rower.objects.create(user=u)
+        r = Rower.objects.create(user=self.u)
 
 
     def test_form(self):
         form = TeamInviteForm()
+
+
+class ModelTests(TestCase):
+    def setUp(self):
+        self.u = User.objects.create_user('john',
+				     'Sander@ds.ds',
+				     'koeinsloot')
+
+        self.u.first_name = 'Sander'
+        self.u.last_name = 'Roosendaal'
+        self.u.save()
+
+        self.r = Rower.objects.create(user=self.u)
+
+    def create_team(self):
+        return Team.objects.create(
+            name = 'test team',
+            manager = self.u,
+            )
+
+    def create_paidplan(self):
+        return PaidPlan.objects.create(
+            name = 'test plan',
+            shortname = 'pro',
+            price = 15.0,
+            )
+
+    def test_models(self):
+        pp = self.create_paidplan()
+        team = self.create_team()
+        self.assertEqual(u'%s' % pp,u'test plan - pro at 15.00 EURO (single payment)')
+        self.assertEqual(u'%s' % self.r, u'Sander Roosendaal')
+        self.assertEqual('%s' % self.r, 'Sander Roosendaal')
+        self.assertEqual(self.r.clean_email(),'sander@ds.ds')