coverage changes

rowers/tests/test_team.py

@@ -63,6 +63,7 @@ class TeamTest(TestCase):
         self.c = Client()
 
         self.u = self.users[0]
+        self.r = self.users[0].rower
         self.password = self.user_passwords[0]
 
         self.t = TeamFactory(manager=self.u)
@@ -83,6 +84,29 @@ class TeamTest(TestCase):
             )
         self.rekwest.save()
 
+        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',
+        )
+
+
     def tearDown(self):
         for csvfilename in self.csvfilenames:
             try:
@@ -126,6 +150,71 @@ class TeamTest(TestCase):
         response = self.c.get(url)
         self.assertEqual(response.status_code, 200)
 
+    @patch('rowers.dataprep.create_engine')
+    @patch('rowers.dataprep.getsmallrowdata_db')
+    @patch('rowers.middleware.myqueue')
+    def test_compares(self, mocked_sqlalchemy, mocked_getsmallrowdata_db,
+                      mocked_myqueue):
+        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('team_comparison_select',kwargs={'teamid':self.t.id})
+        response = self.c.get(url)
+        self.assertEqual(response.status_code,200)
+
+        d1 = self.werg1.date-timedelta(days=2)
+        d2 = self.werg2.date+timedelta(days=2)
+
+        form_data = {
+            'startdate': d1.strftime('%Y-%m-%d'),
+            'enddate': d2.strftime('%Y-%m-%d'),
+            'modality':'water',
+            'waterboattype':['1x'],
+            }
+
+        form = DateRangeForm(form_data)
+        self.assertTrue(form.is_valid())
+
+        form = TrendFlexModalForm(form_data)
+        self.assertTrue(form.is_valid())
+
+        response = self.c.post(url,form_data)
+        self.assertEqual(response.status_code,200)
+
+        url = reverse('multi_compare_view',kwargs={
+            'userid':self.u.id,
+            'id':encoder.encode_hex(self.werg1.id),
+        })
+
+        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)
+
     def test_teamsview(self):
 
         url = '/rowers/me/teams/'