adding wind/metar

rowers/tests/mocks.py

@@ -127,6 +127,11 @@ def mocked_getsmallrowdata_db(*args, **kwargs):
 
     return df
 
+def mocked_getsmallrowdata_db_water(*args, **kwargs):
+    df = pd.read_csv('rowers/tests/testdata/colsfromdb.csv')
+
+    return df
+
 def mocked_getpowerdata_db(*args, **kwargs):
     df = pd.read_csv('rowers/tests/testdata/fake_powerdata.csv')
 

rowers/tests/test_aworkouts.py

@@ -180,3 +180,30 @@ class WorkoutViewTest(TestCase):
         response = self.c.get(url,follow=True)
         self.assertEqual(response.status_code,200)
         
+    @patch('rowers.dataprep.create_engine')
+    @patch('rowers.dataprep.getsmallrowdata_db')
+    def test_waterworkout_view(self, mocked_sqlalchemy, mocked_getsmallrowdata_db):
+        login = self.c.login(username=self.u.username, password=self.password)
+        self.assertTrue(login)
+
+        url  = reverse('workout_view',kwargs={'id':self.wwater.id})
+
+        response = self.c.get(url)
+        self.assertEqual(response.status_code,200)
+
+        url = reverse('workout_downloadmetar_view',kwargs={
+            'id': self.wwater.id,
+            'airportcode': 'LKHO'
+            }
+            )
+
+        response = self.c.get(url,follow=True)
+        self.assertEqual(response.status_code,200)
+
+        url = reverse('workout_downloadwind_view',kwargs={
+            'id': self.wwater.id,
+            }
+            )
+
+        response = self.c.get(url,follow=True)
+        self.assertEqual(response.status_code,200)        

rowers/tests/test_plans.py

@@ -993,7 +993,7 @@ class MandatoryTestCompleteTest(TestCase):
         self.assertEqual(response.status_code,200)
 
         url = reverse('plannedsession_compare_view',kwargs={'id':self.ps_dist.id})
-        response = self.c.get(url)
+        response = self.c.get(url,follow=True)
         self.assertEqual(response.status_code,200)
         
 class PlannedSessionsView(TestCase):

rowers/views/workoutviews.py

@@ -987,8 +987,11 @@ def plannedsession_compare_view(request,id=0,userid=0):
     request.session['yparam'] = yparam
     request.session['plottype'] = plottype
     request.session['ps'] = ps.id
-    
-    url = reverse('multi_compare_view',kwargs={'userid':userid,'id':ids[0]})
+
+    if ids:
+        url = reverse('multi_compare_view',kwargs={'userid':userid,'id':ids[0]})
+    else:
+        url = reverse('plannedsession_view',kwargs={'id':ps.id})
 
     return HttpResponseRedirect(url)