def test_unauthorized_user(self) -> None:
user = self.UserFactory.create()
student = self.StudentFactory()
request = self.request_factory.authenticated_get(user)
with self.assertRaises(Http404):
views.add_school(request, student.id)
def test_form_in_context(self, render: mock.MagicMock) -> None:
user = self.UserFactory.create()
student = self.StudentFactory(user=user)
request = self.request_factory.authenticated_get(user)
views.add_school(request, student.id)
context = render.call_args[0][2]
self.assertIn("form", context)
def test_query_in_context(self, render: mock.MagicMock) -> None:
user = self.UserFactory.create()
student = self.StudentFactory(user=user)
data = {"q": "University of Virginia"}
request = self.request_factory.authenticated_get(user, data=data)
views.add_school(request, student.id)
context = render.call_args[0][2]
self.assertEqual("University of Virginia", context["q"])
def test_requires_login(self) -> None:
request = self.request_factory.get()
response = views.add_school(request, 1)
self.assertEqual(302, response.status_code)
self.assertIn(reverse("login"), response.get("Location"))
def test_valid_get(self) -> None:
user = self.UserFactory.create()
student = self.StudentFactory(user=user)
request = self.request_factory.authenticated_get(user)
response = views.add_school(request, student.id)
self.assertEqual(200, response.status_code)
def test_valid_post(self) -> None:
user = self.UserFactory.create()
student = self.StudentFactory(user=user)
school = self.SchoolFactory.create()
data = {"school": str(school.id)}
request = self.request_factory.authenticated_post(user, data=data)
response = views.add_school(request, student.id)
self.assertEqual(302, response.status_code)
self.assertIn(reverse("student-profile", args=[student.id]),
response.get("Location"))