diff --git a/rowers/tests/test_models.py b/rowers/tests/test_models.py index 08c0b88e..67e555d9 100644 --- a/rowers/tests/test_models.py +++ b/rowers/tests/test_models.py @@ -189,8 +189,11 @@ class PlannedSessionTests(TestCase): self.u.save() self.user_sessions = SessionFactory.create_batch( 5, - manager=self.u, - rower=self.r) + manager=self.u) + + for session in self.user_sessions: + session.rower.add(self.r) + session.save() def test_ics_download(self): login = self.c.login(username=self.u.username,password=self.password) @@ -202,9 +205,9 @@ class PlannedSessionTests(TestCase): filename = 'training_plan_{u}_{d1}_{d2}.ics'.format( u = self.u.username, d1 = timezone.now().date().strftime("%Y%m%d"), - d2 = a_week_from_now.date().strftime("%Y%m%d"), + d2 = (a_week_from_now()-datetime.timedelta(days=1)).date().strftime("%Y%m%d"), ) self.assertEquals( response.get('Content-Disposition'), - "attachment; filename={name}".format(name=filename) + 'attachment; filename="{name}"'.format(name=filename) ) diff --git a/rowers/tests/testdata/testdata.csv.gz b/rowers/tests/testdata/testdata.csv.gz index daba0e6d..20563141 100644 Binary files a/rowers/tests/testdata/testdata.csv.gz and b/rowers/tests/testdata/testdata.csv.gz differ