starting plan view tests

rowers/tests/test_plans.py

@@ -1,6 +1,7 @@
 #from __future__ import print_function
 from statements import *
 nu = datetime.datetime.now()
+from rowers.utils import allmonths,allsundays
 
 import rowers.plannedsessions as plannedsessions
 
@@ -988,3 +989,213 @@ class MandatoryTestCompleteTest(TestCase):
         response = self.c.get(url)
         self.assertEqual(response.status_code,200)
         
+class PlannedSessionsView(TestCase):
+    def setUp(self):
+        # user
+        self.u = UserFactory()
+
+        self.r = Rower.objects.create(user=self.u,
+                                      birthdate=faker.profile()['birthdate'],
+                                      gdproptin=True,
+                                      gdproptindate=timezone.now(),
+                                      rowerplan='coach')
+        self.r.save()
+        self.c = Client()
+        
+        # workouts
+        # workout 1 - 2019-01-13, rScore 69
+        result = get_random_file(filename='rowers/tests/testdata/2019-01-13_session.csv',name='sprintervals')
+
+
+        self.factory = RequestFactory()
+        self.password = faker.word()
+        self.u.set_password(self.password)
+        self.u.save()
+
+        self.w1 = Workout.objects.create(
+            name='sprintervals',
+            notes=faker.text(),
+            startdatetime = result['startdatetime'],
+            starttime = result['starttime'],
+            workouttype='rower',
+            date=result['date'],
+            duration=result['duration'],
+            distance=result['totaldist'],
+            csvfilename=result['filename'],
+            trimp = 77,
+            rscore = 69,
+            hrtss = 43,
+            normp = 236,
+            user=self.u.rower,
+        )
+    
+
+        # plan
+        self.target = TrainingTarget.objects.create(
+            name = faker.word(),
+            manager = self.u.rower,
+            notes = faker.text()
+        )
+        self.target.rowers.add(self.u.rower)
+        self.target.save()
+    
+        self.plan = TrainingPlan.objects.create(
+            manager = self.u.rower,
+            name = faker.word(),
+            status=True,
+            target = self.target,
+            startdate=timezone.now().date(),
+            enddate = self.target.date,
+        )
+
+        self.plan.rowers.add(self.u.rower)
+        self.plan.save()
+
+        # cycles
+        self.macro = TrainingMacroCycle.objects.create(
+            plan=self.plan,
+            name=faker.word(),
+            type='userdefined',
+            notes = faker.text(),
+            startdate = self.plan.startdate,
+            enddate = self.plan.enddate,
+        )
+
+        mesos = TrainingMesoCycle.objects.filter(plan=self.macro)
+        for m in mesos:
+            m.delete()
+
+        monthstarts = [d for d in allmonths(self.macro.startdate,self.macro.enddate)]
+        monthstarts.append(self.macro.enddate)
+
+        for i in range(len(monthstarts)-1):
+            firstday = monthstarts[i]
+            lastday = monthstarts[i+1]-datetime.timedelta(days=1)
+            if lastday < self.macro.enddate and i == len(monthstarts)-2:
+                lastday = self.macro.enddate
+
+            meso = TrainingMesoCycle(startdate=firstday,
+                                     enddate=lastday,
+                                     plan=self.macro,
+                                     name = '%s' % firstday.strftime("%B"),
+                                     type = 'userdefined')
+            meso.save()
+
+        mesos = TrainingMesoCycle.objects.filter(plan=self.macro)
+
+        for cycle in mesos:
+            micros = TrainingMicroCycle.objects.filter(plan=cycle)
+            for m in micros:
+                m.delete()
+
+            sundays = [s for s in allsundays(cycle.startdate,cycle.enddate)]
+
+            if sundays and sundays[-1] < cycle.enddate:
+                sundays = sundays+[cycle.enddate]
+            elif not sundays:
+                sundays = [cycle.enddate]
+
+            for i in range(len(sundays)):
+                if i==0:
+                    monday = cycle.startdate
+                else:
+                    monday = sundays[i]-datetime.timedelta(days=6)
+                    if monday < cycle.startdate:
+                        monday = cycle.startdate
+
+                nextsunday = sundays[i]
+
+                micro = TrainingMicroCycle(startdate=monday,
+                                           enddate=nextsunday,
+                                           plan=cycle,
+                                           name = 'Week %s' % monday.isocalendar()[1],
+                                           type='userdefined')
+                micro.save()
+                
+
+        # sessions
+        startdatetime = self.w1.startdatetime
+
+        startdate = (startdatetime-datetime.timedelta(days=1)).date()
+        enddate = (startdatetime+datetime.timedelta(days=1)).date()
+        preferreddate = startdatetime.date()
+
+        self.ps_rscore = SessionFactory(
+            startdate=startdate,enddate=enddate,
+            sessiontype='test',
+            sessionmode = 'rScore',
+            criterium = 'none',
+            sessionvalue = 69,
+            sessionunit='None',
+            preferreddate=preferreddate,
+            manager=self.u,
+        )
+            
+
+        self.ps_rscore.save()
+        added = plannedsessions.add_rower_session(self.u.rower,self.ps_rscore)
+
+        self.ps_dist = SessionFactory(
+            startdate=startdate,enddate=enddate,
+            sessiontype='test',
+            sessionmode = 'distance',
+            criterium = 'none',
+            sessionvalue = result['totaldist'],
+            sessionunit='m',
+            preferreddate=preferreddate,
+            manager=self.u,
+        )
+            
+
+        self.ps_dist.save()
+        added = plannedsessions.add_rower_session(self.u.rower,self.ps_dist)
+        
+        self.ps_time = SessionFactory(
+            startdate=startdate,enddate=enddate,
+            sessiontype='test',
+            sessionmode = 'time',
+            criterium = 'none',
+            sessionvalue = 38,
+            sessionunit='min',
+            preferreddate=preferreddate,
+            manager=self.u,
+        )
+            
+
+        self.ps_time.save()
+        added = plannedsessions.add_rower_session(self.u.rower,self.ps_time)
+        
+        self.ps_trimp = SessionFactory(
+            startdate=startdate,enddate=enddate,
+            sessiontype='test',
+            sessionmode = 'TRIMP',
+            criterium = 'none',
+            sessionvalue = 77,
+            sessionunit='none',
+            preferreddate=preferreddate,
+            manager=self.u,
+        )
+            
+
+        self.ps_trimp.save()
+        added = plannedsessions.add_rower_session(self.u.rower,self.ps_trimp)
+        
+
+    def tearDown(self):
+        try:
+            os.remove(self.w1.csvfilename)
+        except (IOError, WindowsError):
+            pass
+
+    def test_clone_view(self):
+        login = self.c.login(username=self.u.username, password=self.password)
+        self.assertTrue(login)
+
+        url = '/rowers/sessions/{id}/clone/'.format(id=self.ps_trimp.id)
+
+        response = self.c.get(url,follow=True)
+        self.assertEqual(response.status_code,200)
+        self.assertRedirects(response,
+                             expected_url='/rowers/session/5/edit/user/1/?when=2019-02-04/2019-02-10',
+                             status_code=302,target_status_code=200)
+