Mirrored_Repos/InvenTree
1
0
Fork 0
mirror of https://github.com/inventree/InvenTree synced 2024-08-30 18:33:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Unit tests for report printing

Browse Source
This commit is contained in:
Oliver 2021-08-12 14:43:04 +10:00
parent 96ec8c4eb6
commit 6748f37405
1 changed files with 64 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
InvenTree/report/tests.py
View File

@ -4,12 +4,14 @@ from __future__ import unicode_literals
import os
import shutil
from django.http.response import StreamingHttpResponse
from django.urls import reverse
from django.conf import settings
from InvenTree.api_tester import InvenTreeAPITestCase
import report.models as report_models
from common.models import InvenTreeUserSetting
class ReportTest(InvenTreeAPITestCase):
@ -22,6 +24,8 @@ class ReportTest(InvenTreeAPITestCase):
'supplier_part',
'stock',
'stock_tests',
'bom',
'build',
]
model = None
@ -131,6 +135,31 @@ class TestReportTest(ReportTest):
return super().setUp()
def test_print(self):
"""
Printing tests for the TestReport
"""
report = self.model.objects.get(pk=1)
url = reverse(self.print_url, kwargs={'pk': 1})
# Try to print without providing a valid StockItem
response = self.get(url, expected_code=400)
# Try to print with an invalid StockItem
response = self.get(url, {'item': 9999}, expected_code=400)
# Now print with a valid StockItem
response = self.get(url, {'item': 1})
# Response should be a StreamingHttpResponse (PDF file)
self.assertEqual(type(response), StreamingHttpResponse)
headers = response.headers
self.assertEqual(headers['Content-Type'], 'application/pdf')
class BuildReportTest(ReportTest):
@ -146,6 +175,41 @@ class BuildReportTest(ReportTest):
return super().setUp()
def test_print(self):
"""
Printing tests for the BuildReport
"""
report = self.model.objects.get(pk=1)
url = reverse(self.print_url, kwargs={'pk': 1})
# Try to print without providing a valid BuildOrder
response = self.get(url, expected_code=400)
# Try to print with an invalid BuildOrder
response = self.get(url, {'build': 9999}, expected_code=400)
# Now print with a valid BuildOrder
response = self.get(url, {'build': 1})
self.assertEqual(type(response), StreamingHttpResponse)
headers = response.headers
self.assertEqual(headers['Content-Type'], 'application/pdf')
self.assertEqual(headers['Content-Disposition'], 'attachment; filename="report.pdf"')
# Now, set the download type to be "inline"
inline = InvenTreeUserSetting.get_setting_object('REPORT_INLINE', self.user)
inline.value = True
inline.save()
response = self.get(url, {'build': 1})
headers = response.headers
self.assertEqual(headers['Content-Type'], 'application/pdf')
self.assertEqual(headers['Content-Disposition'], 'inline; filename="report.pdf"')
class BOMReportTest(ReportTest):