bug fix (pro user entering plan trial)

rowers/tests/test_rowerplans.py

@@ -0,0 +1,64 @@
+from statements import *
+nu = datetime.datetime.now()
+
+from rowers.views import hasplannedsessions,iscoachmember,ispromember
+
+class TrialsTest(TestCase):
+    def setUp(self):
+        self.u = UserFactory()
+        
+        self.r = Rower.objects.create(user=self.u,
+                                      birthdate=faker.profile()['birthdate'],
+                                      gdproptin=True,
+                                      gdproptindate=timezone.now(),
+                                      rowerplan='basic')
+
+        self.c = Client()
+        self.user_workouts = WorkoutFactory.create_batch(5, user=self.r)
+        self.factory = RequestFactory()
+        self.password = faker.word()
+        self.u.set_password(self.password)
+        self.u.save()
+
+    def tearDown(self):
+        for workout in self.user_workouts:
+            try:
+                os.remove(workout.csvfilename)
+            except (IOError, WindowsError):
+                pass
+
+    def test_basictrial(self):
+        self.assertEqual(hasplannedsessions(self.u),False)
+        self.assertEqual(iscoachmember(self.u),False)
+        self.assertEqual(ispromember(self.u),False)
+
+        self.r.protrialexpires = (nu+datetime.timedelta(days=10)).date()
+        self.r.save()
+
+        self.assertEqual(hasplannedsessions(self.u),False)
+        self.assertEqual(iscoachmember(self.u),False)
+        self.assertEqual(ispromember(self.u),True)
+
+        self.r.plantrialexpires = (nu+datetime.timedelta(days=10)).date()
+        self.r.save()
+        
+        self.assertEqual(hasplannedsessions(self.u),True)
+        self.assertEqual(iscoachmember(self.u),False)
+        self.assertEqual(ispromember(self.u),True)
+
+    def test_protrial(self):
+        self.r.rowerplan = 'pro'
+        self.r.save()
+        self.assertEqual(hasplannedsessions(self.u),False)
+        self.assertEqual(iscoachmember(self.u),False)
+        self.assertEqual(ispromember(self.u),True)
+
+        self.r.plantrialexpires = (nu+datetime.timedelta(days=10)).date()
+        self.r.save()
+        
+        self.assertEqual(hasplannedsessions(self.u),True)
+        self.assertEqual(iscoachmember(self.u),False)
+        self.assertEqual(ispromember(self.u),True)
+
+
+        

rowers/tests/testdata/testdata.tcx

@@ -2502,7 +2502,7 @@
           </Trackpoint>
         </Track>
       </Lap>
-      <Notes>&lt;Element 'Notes' at 0x1f6b6c18&gt;</Notes>
+      <Notes>&lt;Element 'Notes' at 0x15216240&gt;</Notes>
     </Activity>
   </Activities>
   <Creator>