def _validate_cas2(self, response, is_correct=True, pgtUrl=None):
if is_correct:
self.assertEqual(response.status_code, 302)
self.assertTrue(response.has_header('location'))
location = urlparse(response['location'])
ticket = location.query.split('=')[1]
if pgtUrl:
response = self.client.get(reverse('cas_service_validate'), {
'ticket': ticket,
'service': self.service,
'pgtUrl': pgtUrl
},
follow=False)
else:
response = self.client.get(reverse('cas_service_validate'), {
'ticket': ticket,
'service': self.service
},
follow=False)
self.assertEqual(response.status_code, 200)
else:
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context['form'].errors), 1)
response = self.client.get(reverse('cas_service_validate'), {
'ticket': 'ST-12312312312312312312312',
'service': self.service
},
follow=False)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.content,
_cas2_error_response(INVALID_TICKET).content)
return response
def _validate_cas2(self, response, is_correct=True, pgtUrl=None):
if is_correct:
self.assertEqual(response.status_code, 302)
self.assertTrue(response.has_header("location"))
location = urlparse(response["location"])
ticket = location.query.split("=")[1]
if pgtUrl:
response = self.client.get(
reverse("cas_service_validate"),
{"ticket": ticket, "service": self.service, "pgtUrl": pgtUrl},
follow=False,
)
else:
response = self.client.get(
reverse("cas_service_validate"), {"ticket": ticket, "service": self.service}, follow=False
)
self.assertEqual(response.status_code, 200)
else:
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context["form"].errors), 1)
response = self.client.get(
reverse("cas_service_validate"),
{"ticket": "ST-12312312312312312312312", "service": self.service},
follow=False,
)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.content, _cas2_error_response(INVALID_TICKET).content)
return response
def test_cas2_validate_twice(self):
response = self._login_user('root', '123')
response2 = self._validate_cas2(response, True)
user = User.objects.get(username=self.username)
self.assertEqual(response2.content, _cas2_sucess_response(user).content)
response2 = self._validate_cas2(response, True)
self.assertEqual(response2.content, _cas2_error_response(INVALID_TICKET).content)
def test_cas2_validate_twice_with_pgt(self):
proxyTarget = "http://my.sweet.service_1"
response = self._login_user('root', '123')
response2 = self._validate_cas2(response, True, proxyTarget)
user = User.objects.get(username=self.username)
self.assertEqual(response2.content, _cas2_sucess_response(user).content)
response2 = self._validate_cas2(response, True, proxyTarget)
self.assertEqual(response2.content, _cas2_error_response(INVALID_TICKET).content)
def _validate_cas2(self, response, is_correct=True):
if is_correct:
self.assertEqual(response.status_code, 302)
self.assertTrue(response.has_header('location'))
location = urlparse(response['location'])
ticket = location.query.split('=')[1]
response = self.client.get(reverse('cas_service_validate'), {'ticket': ticket, 'service': self.service}, follow=False)
self.assertEqual(response.status_code, 200)
else:
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context['form'].errors), 1)
response = self.client.get(reverse('cas_service_validate'), {'ticket': 'ST-12312312312312312312312', 'service': self.service}, follow=False)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.content, _cas2_error_response(INVALID_TICKET).content)
return response
