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

Remove reliance on hard-coded PK values

Browse Source
This commit is contained in:
Oliver 2021-08-12 15:14:13 +10:00
parent e36b1e6c70
commit f59ed27cf9
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
InvenTree/report/tests.py
View File

@ -12,6 +12,8 @@ from InvenTree.api_tester import InvenTreeAPITestCase
import report.models as report_models import report.models as report_models
from common.models import InvenTreeUserSetting from common.models import InvenTreeUserSetting
from stock.models import StockItem
from build.models import Build
class ReportTest(InvenTreeAPITestCase): class ReportTest(InvenTreeAPITestCase):
@ -140,7 +142,9 @@ class TestReportTest(ReportTest):
Printing tests for the TestReport Printing tests for the TestReport
""" """
url = reverse(self.print_url, kwargs={'pk': 1}) report = self.model.objects.first()
url = reverse(self.print_url, kwargs={'pk': report.pk})
# Try to print without providing a valid StockItem # Try to print without providing a valid StockItem
response = self.get(url, expected_code=400) response = self.get(url, expected_code=400)
@ -149,7 +153,9 @@ class TestReportTest(ReportTest):
response = self.get(url, {'item': 9999}, expected_code=400) response = self.get(url, {'item': 9999}, expected_code=400)
# Now print with a valid StockItem # Now print with a valid StockItem
response = self.get(url, {'item': 1}) item = StockItem.objects.first()
response = self.get(url, {'item': item.pk})
# Response should be a StreamingHttpResponse (PDF file) # Response should be a StreamingHttpResponse (PDF file)
self.assertEqual(type(response), StreamingHttpResponse) self.assertEqual(type(response), StreamingHttpResponse)
@ -178,7 +184,9 @@ class BuildReportTest(ReportTest):
Printing tests for the BuildReport Printing tests for the BuildReport
""" """
url = reverse(self.print_url, kwargs={'pk': 1}) report = self.model.objects.first()
url = reverse(self.print_url, kwargs={'pk': report.pk})
# Try to print without providing a valid BuildOrder # Try to print without providing a valid BuildOrder
response = self.get(url, expected_code=400) response = self.get(url, expected_code=400)
@ -187,7 +195,10 @@ class BuildReportTest(ReportTest):
response = self.get(url, {'build': 9999}, expected_code=400) response = self.get(url, {'build': 9999}, expected_code=400)
# Now print with a valid BuildOrder # Now print with a valid BuildOrder
response = self.get(url, {'build': 1})
build = Build.objects.first()
response = self.get(url, {'build': build.pk})
self.assertEqual(type(response), StreamingHttpResponse) self.assertEqual(type(response), StreamingHttpResponse)