Minor test addition for builds

This commit is contained in:
Oliver Walters 2019-06-17 22:41:44 +10:00
parent 38ef9b2b13
commit 760d08608f

View File

@ -14,6 +14,7 @@ class BuildTestSimple(TestCase):
def setUp(self):
part = Part.objects.create(name='Test part',
description='Simple description')
Build.objects.create(part=part,
batch='B1',
status=BuildStatus.PENDING,
@ -35,6 +36,8 @@ class BuildTestSimple(TestCase):
self.assertEqual(b.batch, 'B2')
self.assertEqual(b.quantity, 21)
self.assertEqual(str(b), 'Build 21 x Test part - Simple description')
def test_url(self):
b1 = Build.objects.get(pk=1)
self.assertEqual(b1.get_absolute_url(), '/build/1/')
@ -46,6 +49,8 @@ class BuildTestSimple(TestCase):
self.assertEqual(b1.is_complete, False)
self.assertEqual(b2.is_complete, True)
self.assertEqual(b2.status, BuildStatus.COMPLETE)
def test_is_active(self):
b1 = Build.objects.get(pk=1)
b2 = Build.objects.get(pk=2)
@ -56,3 +61,14 @@ class BuildTestSimple(TestCase):
def test_required_parts(self):
# TODO - Generate BOM for test part
pass
def cancel_build(self):
""" Test build cancellation function """
build = Build.objects.get(id=1)
self.assertEqual(build.status, BuildStatus.PENDING)
build.cancelBuild()
self.assertEqual(build.status, BuildStatus.CANCELLED)