diff --git a/InvenTree/build/tests.py b/InvenTree/build/tests.py index 1030a4e198..36e8223a9b 100644 --- a/InvenTree/build/tests.py +++ b/InvenTree/build/tests.py @@ -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)