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 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)
class FrontpageTests(TestCase):
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)
@pytest.mark.integration
def test_homepage_logged_out(self):
request = RequestFactory().get('/')
request.user = AnonymousUser()
response = self.view(request)
next_url = "{0}?next={1}".format(reverse("login"), reverse("home"))
self.assertEqual(response.status_code, 302)
self.assertEqual(response['Location'], next_url)
@pytest.mark.integration
def test_homepage_logged_in(self):
request = RequestFactory().get('/')
request.user = self.user # So we're logged in
response = self.view(request)
self.assertEqual(response.status_code, 302)
self.assertEqual(response['Location'], reverse("dashboard"))
@pytest.mark.integration
@pytest.mark.client
def test_login(self):
email = self.user.business_email
password = '******'
self.user.set_password(password)
self.user.save()
# Can we log in?
login_data = {
'username': email,
'password': password,
}
c = Client()
response = c.post(reverse('login'), login_data, follow=True)
self.assertContains(response, 'Log out')
class FrontpageTests(TestCase):
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)
@pytest.mark.integration
def test_homepage_logged_out(self):
request = RequestFactory().get('/')
request.user = AnonymousUser()
response = self.view(request)
next_url = "{0}?next={1}".format(reverse("login"), reverse("home"))
self.assertEqual(response.status_code, 302)
self.assertEqual(response['Location'], next_url)
@pytest.mark.integration
def test_homepage_logged_in(self):
request = RequestFactory().get('/')
request.user = self.user # So we're logged in
response = self.view(request)
self.assertEqual(response.status_code, 302)
self.assertEqual(response['Location'], reverse("dashboard"))
@pytest.mark.integration
@pytest.mark.client
def test_login(self):
email = self.user.business_email
password = '******'
self.user.set_password(password)
self.user.save()
# Can we log in?
login_data = {
'username': email,
'password': password,
}
c = Client()
response = c.post(reverse('login'), login_data, follow=True)
self.assertContains(response, 'Log out')
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')
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')
