diff --git a/rowers/tests/test_staticcharts.py b/rowers/tests/test_staticcharts.py index 82bccb50..552c048c 100644 --- a/rowers/tests/test_staticcharts.py +++ b/rowers/tests/test_staticcharts.py @@ -16,7 +16,10 @@ class PlotTests(TestCase): u = User.objects.create_user('john', 'sander@ds.ds', 'koeinsloot') - r = Rower.objects.create(user=u) + r = Rower.objects.create(user=u, + gdproptin=True, + gdproptindate=timezone.now(), + rowerplan='basic') self.nu = datetime.datetime.now() filename = 'rowers/tests/testdata/testdata.csv' self.wotw = Workout.objects.create(name='testworkout', @@ -199,17 +202,29 @@ class PlotTests(TestCase): @patch('rowers.tasks.FigureCanvas') def test_chartrequests(self, mocked_rowingdata, mocked_canvas): opid = encoder.encode_hex(self.wotw.id) + url = reverse('workout_add_chart_view', kwargs={'id':opid, 'plotnr':13 }) + workout_edit_url = reverse('workout_edit_view', + kwargs={'id':opid}) + + + login = self.c.login(username='john', password='koeinsloot') self.assertTrue(login) - response = self.c.get(url) + response = self.c.get(workout_edit_url,follow=True) self.assertEqual(response.status_code,200) + response = self.c.get(url,follow=True) + self.assertEqual(response.status_code,200) + self.assertRedirects(response, + expected_url = workout_edit_url, + status_code=302,target_status_code=200) + i = GraphImage.objects.filter(workout=self.wotw) self.assertEqual(len(i),1) @@ -222,6 +237,9 @@ class PlotTests(TestCase): response = self.c.post(url,{},follow=True) self.assertEqual(response.status_code,200) + self.assertRedirects(response, + expected_url = workout_edit_url, + status_code=302,target_status_code=200) i = GraphImage.objects.filter(workout=self.wotw) self.assertEqual(len(i),0)