def test_everyone_redirected_to_dashboard():
user = UserFactory()
user.is_superuser = True
response = get_response(user)
assert response.status_code == 302
assert response['Location'] == reverse(u"dashboard")
def setUp(self):
self.user = UserFactory()
self.view = Home.as_view()
# Logframe is now created automatically, don't create it if it exists
try:
lf = LogFrame.objects.get()
except LogFrame.DoesNotExist:
lf = G(LogFrame)
rating = G(Rating, log_frame=lf)
G(Result, log_frame=lf, parent=None, rating=rating)
def test_export_annual_plan(self):
# Can we log in?
user = UserFactory()
user.set_password("test")
user.save()
self.client.login(username=user.business_email, password="******")
response = self.client.get(
reverse('export-annual-plan',
kwargs={
'pk': self.log_frame.pk,
'year': '2014',
}))
self.assertEqual(200, response.status_code)
self.assertEqual(response['Content-Disposition'],
'attachment; filename="%s"' % '2014_annual_plan.xlsx')
