64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
from django.urls import reverse
|
|
from rest_framework import status
|
|
from rest_framework.test import APITestCase
|
|
from apiserver.api.models import Member, User
|
|
import json
|
|
from parameterized import parameterized
|
|
|
|
data = {
|
|
"username": "registrationtc",
|
|
"email": "unittest@email.com",
|
|
"password1": "unittest",
|
|
"password2": "unittest",
|
|
"first_name": "John",
|
|
"last_name": "Doe",
|
|
|
|
# need to fake this for updating progress
|
|
"request_id": "lol"
|
|
}
|
|
|
|
|
|
class RegistrationTests(APITestCase):
|
|
def setUp(self):
|
|
self.url = reverse('rest_name_register')
|
|
# TODO: expose data to be used for E2E testing from a webclient
|
|
self.data = data
|
|
# TODO: match with config
|
|
self.allowed_ip = '24.66.110.96'
|
|
|
|
def test_success(self):
|
|
"""Ensure we can create a new account object."""
|
|
response = self.client.post(
|
|
self.url,
|
|
self.data,
|
|
format='json',
|
|
HTTP_X_REAL_IP=self.allowed_ip
|
|
)
|
|
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
|
user = User.objects.get(username=self.data['username'])
|
|
assert user is not None
|
|
assert Member.objects.get(user=user) is not None
|
|
|
|
def test_allowed_ip_wrong(self):
|
|
"""Ensure creation only allowed when HTTP_X_REAL_IP header matched IP in whitelist"""
|
|
response = self.client.post(
|
|
self.url,
|
|
self.data,
|
|
format='json',
|
|
HTTP_X_REAL_IP="0.0.0.0"
|
|
)
|
|
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
|
|
|
|
@parameterized.expand([(f"{key} is missing", key, status.HTTP_400_BAD_REQUEST) for key in data.keys() if key is not 'request_id'])
|
|
def test_malformed_data(self, name, inp, expected):
|
|
"""Delete specific properties from data and confirm it is not accepted by API"""
|
|
copy = self.data.copy()
|
|
del copy[inp]
|
|
response = self.client.post(
|
|
self.url,
|
|
copy,
|
|
format='json',
|
|
HTTP_X_REAL_IP=self.allowed_ip
|
|
)
|
|
self.assertEqual(response.status_code, expected)
|