55 lines
2 KiB
Python
55 lines
2 KiB
Python
from django.core import mail
|
|
from rest_framework import status
|
|
|
|
from userausfall.models import TrustBridge
|
|
from userausfall.rest_api.tests import UserausfallAPITestCase, UserMixin
|
|
|
|
|
|
class TrustBridgeTestCase(UserMixin, UserausfallAPITestCase):
|
|
def create_trust_bridge(self):
|
|
self.trust_giver = self.create_user()
|
|
self.create_user()
|
|
self.trust_bridge = TrustBridge.objects.create(trust_taker=self.user, trust_giver=self.trust_giver)
|
|
return self.trust_bridge
|
|
|
|
def test_create_trust_bridge(self):
|
|
"""Create a trust bridge for the current user."""
|
|
url = "/trust-bridges/"
|
|
trust_giver = self.create_user()
|
|
self.create_user()
|
|
self.authenticate_user()
|
|
response = self.client.post(
|
|
self.get_api_url(url),
|
|
{
|
|
"trust_giver": {
|
|
"username": trust_giver.username,
|
|
},
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
|
self.assertEqual(TrustBridge.objects.count(), 1)
|
|
self.assertEqual(TrustBridge.objects.get().trust_giver, trust_giver)
|
|
|
|
def test_retrieve_trust_bridge(self):
|
|
"""Retrieve the trust bridge information of a user without an ldap account."""
|
|
url = "/trust-bridges/{pk}/"
|
|
self.create_trust_bridge()
|
|
self.authenticate_user()
|
|
response = self.client.get(self.get_api_url(url, pk=self.trust_bridge.pk))
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertEqual(
|
|
response.data,
|
|
{
|
|
"is_trusted": False,
|
|
"trust_giver": {
|
|
"username": self.trust_giver.username,
|
|
},
|
|
},
|
|
)
|
|
|
|
def test_send_confirmation_email_on_creation(self):
|
|
"""When setting a trust giver a confirmation email is sent."""
|
|
self.create_trust_bridge()
|
|
self.assertEqual(len(mail.outbox), 1)
|
|
self.assertIn(self.user.trust_bridge.get_confirmation_token(), mail.outbox[0].body)
|