added compare

rowers/tests/test_aworkouts.py

@@ -2,7 +2,7 @@ from statements import *
 
 nu = datetime.datetime.now()
 
-class WaterWorkoutViewTest(TestCase):
+class WorkoutViewTest(TestCase):
     def setUp(self):
         self.u = UserFactory()
         
@@ -26,7 +26,41 @@ class WaterWorkoutViewTest(TestCase):
                                      starttime=result['starttime'],
                                      startdatetime=result['startdatetime'],
                                      duration=result['duration'],
-                                     distance=result['totaldist']
+                                     distance=result['totaldist'],
+                                     workouttype = 'water',
+        )
+
+        result = get_random_file(filename='rowers/tests/testdata/erg1.csv')
+        
+        self.werg1 = WorkoutFactory(user=self.r,
+                                     csvfilename=result['filename'],
+                                     starttime=result['starttime'],
+                                     startdatetime=result['startdatetime'],
+                                     duration=result['duration'],
+                                     distance=result['totaldist'],
+                                     workouttype = 'rower',
+        )
+
+        result = get_random_file(filename='rowers/tests/testdata/erg2.csv')
+        
+        self.werg2 = WorkoutFactory(user=self.r,
+                                     csvfilename=result['filename'],
+                                     starttime=result['starttime'],
+                                     startdatetime=result['startdatetime'],
+                                     duration=result['duration'],
+                                     distance=result['totaldist'],
+                                     workouttype = 'rower',
+        )
+
+        result = get_random_file(filename='rowers/tests/testdata/erg3.csv')
+        
+        self.werg3 = WorkoutFactory(user=self.r,
+                                     csvfilename=result['filename'],
+                                     starttime=result['starttime'],
+                                     startdatetime=result['startdatetime'],
+                                     duration=result['duration'],
+                                     distance=result['totaldist'],
+                                     workouttype = 'rower',
         )
 
     def tearDown(self):
@@ -50,3 +84,99 @@ class WaterWorkoutViewTest(TestCase):
         response = self.c.post(url,form_data)
         self.assertEqual(response.status_code,200)
         
+    @patch('rowers.dataprep.create_engine')
+    @patch('rowers.dataprep.getsmallrowdata_db')
+    def test_joins(self, mocked_sqlalchemy, mocked_getsmallrowdata_db):
+        login = self.c.login(username=self.u.username, password=self.password)
+        self.assertTrue(login)
+
+        url = reverse('workouts_join_select')
+        response = self.c.get(url)
+        self.assertEqual(response.status_code,200)
+
+        d1 = self.werg1.date-datetime.timedelta(days=2)
+        d2 = self.werg2.date+datetime.timedelta(days=2)
+        
+        date_form_data = {
+            'startdate': d1.strftime('%Y-%m%d'),
+            'enddate': d2.strftime('%Y-%m%d')
+            }
+
+        response = self.c.post(url,date_form_data)
+        self.assertEqual(response.status_code,200)
+        
+        url  = reverse('workouts_join_view')
+
+        response = self.c.get(url,follow=True)
+        self.assertEqual(response.status_code,200)
+
+        expected_url = reverse('workouts_join_select')
+        self.assertRedirects(response,
+                             expected_url=expected_url,
+                             status_code=302,
+                             target_status_code=200)
+
+        form_data = {
+            'workout_name': 'joined',
+            'set_private': False,
+            'workouts': [str(self.werg1.id),str(self.werg2.id),str(self.werg3.id)],
+            }
+
+        form = WorkoutMultipleCompareForm(form_data)
+        self.assertTrue(form.is_valid())
+
+        form = WorkoutJoinParamForm(form_data)
+        self.assertTrue(form.is_valid())
+
+        response = self.c.post(url,form_data,follow=True)
+        self.assertEqual(response.status_code,200)
+
+        expected_url = reverse(self.r.defaultlandingpage,
+                               kwargs = {
+                                   'id':10
+                                   })
+
+        self.assertRedirects(response,
+                             expected_url=expected_url,
+                             status_code=302,target_status_code=200)
+        
+    @patch('rowers.dataprep.create_engine')
+    @patch('rowers.dataprep.getsmallrowdata_db')
+    def test_compares(self, mocked_sqlalchemy, mocked_getsmallrowdata_db):
+        login = self.c.login(username=self.u.username, password=self.password)
+        self.assertTrue(login)
+
+
+        session = self.c.session
+        session['ids'] = [self.werg1.id,self.werg2.id]
+        session.save()
+
+        response = self.c.get('/')
+        url = reverse('multi_compare_view',kwargs={
+            'userid':self.u.id,
+            'id':self.werg1.id,
+        })
+        print url
+
+        form_data = {
+            'xparam':'time',
+            'yparam':'power',
+            'plottype':'line',
+            'teamid': '',
+            }
+        
+        
+        response = self.c.post(url,form_data,follow=True)
+        self.assertEqual(response.status_code,200)
+
+        session['plottype'] = 'scatter'
+        session['xparam'] = 'time'
+        session['yparam'] = 'hr'
+
+        session.save()
+
+        response = self.c.get('/')
+
+        response = self.c.get(url,follow=True)
+        self.assertEqual(response.status_code,200)
+