221 lines
9.2 KiB
Python
221 lines
9.2 KiB
Python
from django.test import TestCase
|
|
from rest_framework.test import APIRequestFactory, force_authenticate
|
|
from backend.models import Report
|
|
from users.models import CustomUser
|
|
from unittest.mock import Mock, patch
|
|
from datetime import date
|
|
from backend.views import *
|
|
import json
|
|
|
|
class ReportTests(TestCase):
|
|
|
|
def create_test_user(self, email, first, last, password):
|
|
"""
|
|
Create a test user and return it.
|
|
"""
|
|
user = CustomUser.objects.create_user(username=email, email=email, first_name=first, last_name=last, password=password)
|
|
return user
|
|
|
|
def mock_report():
|
|
"""
|
|
Generates a mock object with the attributes of a report.
|
|
"""
|
|
r = Mock()
|
|
r.report_pk = 1
|
|
r.title = 'Report Title'
|
|
r.date_created = '2019-03-01'
|
|
r.date_submitted = '2019-03-01'
|
|
r.submitted = False
|
|
r.reference_number = '12345'
|
|
return r
|
|
|
|
def setUp(self):
|
|
"""
|
|
Create a couple test users and save them in the database.
|
|
"""
|
|
self.test_user_1 = self.create_test_user('one@one.com', 'One', 'Mr. One', '1password')
|
|
self.test_user_1.save()
|
|
self.test_user_2 = self.create_test_user('two@two.com', 'Two', 'Mr. Two', '1password')
|
|
self.test_user_2.save()
|
|
|
|
def test_create_report_logged_in(self):
|
|
"""
|
|
Test when an authenticated user tries to submit a report.
|
|
"""
|
|
factory = APIRequestFactory()
|
|
request = factory.post('/api/v1/report', {'title':'Test Report', 'reference':'12345'})
|
|
user = CustomUser.objects.get(email='one@one.com')
|
|
force_authenticate(request, user=user)
|
|
response = create_report(request)
|
|
self.assertEqual(response.status_code, 200)
|
|
report = Report.objects.get(user_id=user)
|
|
self.assertEqual(report.title, 'Test Report')
|
|
|
|
def test_create_report_logged_out(self):
|
|
"""
|
|
Test when an unauthenticated user tries to create a report.
|
|
"""
|
|
factory = APIRequestFactory()
|
|
request = factory.post('/api/v1/report', {'title':'Test Report', 'reference':'12345'})
|
|
response = create_report(request)
|
|
self.assertEqual(response.status_code, 401)
|
|
|
|
@patch('backend.models.Report.objects.filter', Mock(return_value=[mock_report()]))
|
|
@patch('backend.views.get_sections', Mock(return_value={}))
|
|
def test_get_report(self):
|
|
result = get_report(1)
|
|
self.assertEqual(
|
|
result,
|
|
{
|
|
'date_created':'2019-03-01',
|
|
'reference_number':'12345',
|
|
'report_pk':1,
|
|
'title':'Report Title',
|
|
'date_submitted':'2019-03-01',
|
|
'submitted':False
|
|
}
|
|
)
|
|
|
|
def test_report_submit_for_review_logged_out(self):
|
|
"""
|
|
Test for when an unauthenticated user tries to submit a report for review.
|
|
"""
|
|
factory = APIRequestFactory()
|
|
request = factory.put('/api/v1/report/1')
|
|
response = report_detail(request)
|
|
self.assertEqual(response.status_code, 401)
|
|
|
|
def test_report_submit_for_review_logged_in_not_finalized(self):
|
|
"""
|
|
Test for when an authenticated user tries to submit for review a report
|
|
that has not been finalized yet.
|
|
"""
|
|
factory = APIRequestFactory()
|
|
add_report_request = factory.post('/api/v1/report', {'title':'Test Report', 'reference':'12345'})
|
|
user = CustomUser.objects.get(email='one@one.com')
|
|
force_authenticate(add_report_request, user=user)
|
|
create_report(add_report_request)
|
|
review_request = factory.put('/api/v1/report/1')
|
|
force_authenticate(review_request, user=user)
|
|
response = report_detail(review_request, 1)
|
|
self.assertEqual(response.status_code, 200)
|
|
report = Report.objects.get(user_id=user)
|
|
self.assertFalse(report.submitted)
|
|
|
|
def test_report_submit_for_review_logged_in_already_finalized(self):
|
|
"""
|
|
Test for when an authenticated user tries to submit for review a report
|
|
that has already been finalized.
|
|
"""
|
|
factory = APIRequestFactory()
|
|
add_report_request = factory.post('/api/v1/report', {'title':'Test Report', 'reference':'12345'})
|
|
user = CustomUser.objects.get(email='one@one.com')
|
|
force_authenticate(add_report_request, user=user)
|
|
create_report(add_report_request)
|
|
report = Report.objects.get(user_id=user)
|
|
report.submitted = True
|
|
report.save()
|
|
review_request = factory.put('/api/v1/report/1')
|
|
force_authenticate(review_request, user=user)
|
|
response = report_detail(review_request, 1)
|
|
self.assertEqual(response.status_code, 409)
|
|
|
|
def test_report_finalize_logged_out(self):
|
|
"""
|
|
Test for when an unauthenticated user tries to finalize a report.
|
|
"""
|
|
factory = APIRequestFactory()
|
|
request = factory.put('/api/v1/report/1/final')
|
|
response = finalize_report(request, 1)
|
|
self.assertEqual(response.status_code, 401)
|
|
|
|
def test_report_finalize_wrong_owner(self):
|
|
"""
|
|
Test for when an authenticated user tries to finalize someone else's report.
|
|
"""
|
|
factory = APIRequestFactory()
|
|
|
|
# Create a report for user One
|
|
add_report_1_request = factory.post('/api/v1/report', {'title':'One\'s Report', 'reference':'12345'})
|
|
force_authenticate(add_report_1_request, user=self.test_user_1)
|
|
create_report(add_report_1_request)
|
|
|
|
# Create a report for user Two
|
|
add_report_2_request = factory.post('/api/v1/report', {'title':'Two\'s Report', 'reference':'12345'})
|
|
force_authenticate(add_report_2_request, user=self.test_user_2)
|
|
create_report(add_report_2_request)
|
|
|
|
# Try finalizing user Two's report with user One
|
|
request = factory.put('/api/v1/report/2/final')
|
|
force_authenticate(request, user=self.test_user_1)
|
|
response = finalize_report(request, 2)
|
|
self.assertEqual(response.status_code, 401)
|
|
|
|
def test_report_finalize_logged_in_not_finalized(self):
|
|
"""
|
|
Test for when an authenticated user tries to finalize a report
|
|
that has not been finalized yet.
|
|
"""
|
|
factory = APIRequestFactory()
|
|
add_report_request = factory.post('/api/v1/report', {'title':'Test Report', 'reference':'12345'})
|
|
user = CustomUser.objects.get(email='one@one.com')
|
|
force_authenticate(add_report_request, user=user)
|
|
create_report(add_report_request)
|
|
review_request = factory.put('/api/v1/report/1/final')
|
|
force_authenticate(review_request, user=user)
|
|
response = finalize_report(review_request, 1)
|
|
self.assertEqual(response.status_code, 200)
|
|
report = Report.objects.get(user_id=user)
|
|
self.assertTrue(report.submitted)
|
|
|
|
def test_report_finalize_logged_in_already_finalized(self):
|
|
"""
|
|
Test for when an authenticated user tries to finalize a report
|
|
that has already been finalized.
|
|
"""
|
|
factory = APIRequestFactory()
|
|
add_report_request = factory.post('/api/v1/report', {'title':'Test Report', 'reference':'12345'})
|
|
user = CustomUser.objects.get(email='one@one.com')
|
|
force_authenticate(add_report_request, user=user)
|
|
create_report(add_report_request)
|
|
report = Report.objects.get(user_id=user)
|
|
report.submitted = True
|
|
report.save()
|
|
review_request = factory.put('/api/v1/report/1/final')
|
|
force_authenticate(review_request, user=user)
|
|
response = finalize_report(review_request, 1)
|
|
self.assertEqual(response.status_code, 409)
|
|
|
|
def test_report_get_report_logged_in(self):
|
|
"""
|
|
Test for when an authenticated user tries to view a report.
|
|
"""
|
|
factory = APIRequestFactory()
|
|
add_report_request = factory.post('/api/v1/report', {'title':'Test Report', 'reference':'12345'})
|
|
user = CustomUser.objects.get(email='one@one.com')
|
|
force_authenticate(add_report_request, user=user)
|
|
create_report(add_report_request)
|
|
get_request = factory.get('/api/v1/report/1')
|
|
force_authenticate(get_request, user=user)
|
|
response = report_detail(get_request, 1)
|
|
self.assertEqual(response.status_code, 200)
|
|
report = Report.objects.get(user_id=user)
|
|
# Check that the json response contains the title of the report we want
|
|
j = json.loads(response.content.decode("utf-8", "strict"))
|
|
self.assertEqual(report.title, j['title'])
|
|
|
|
def test_report_delete_report_logged_in(self):
|
|
"""
|
|
Test for when an authenticated user tries to delete a report.
|
|
"""
|
|
factory = APIRequestFactory()
|
|
add_report_request = factory.post('/api/v1/report', {'title':'Test Report', 'reference':'12345'})
|
|
user = CustomUser.objects.get(email='one@one.com')
|
|
force_authenticate(add_report_request, user=user)
|
|
create_report(add_report_request)
|
|
delete_request = factory.delete('/api/v1/report/1')
|
|
force_authenticate(delete_request, user=user)
|
|
response = report_detail(delete_request, 1)
|
|
self.assertEqual(response.status_code, 200)
|
|
reports = Report.objects.filter(user_id=user)
|
|
self.assertEqual(len(reports), 0)
|