From 42a794e8e47213294e2c3e190c56b93e94ee08f6 Mon Sep 17 00:00:00 2001
From: Oliver <oliver.henry.walters@gmail.com>
Date: Thu, 4 Nov 2021 15:05:54 +1100
Subject: [PATCH] Fix CI errors

---
 InvenTree/build/tasks.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/InvenTree/build/tasks.py b/InvenTree/build/tasks.py
index 7455d6eac2..6fe4be5119 100644
--- a/InvenTree/build/tasks.py
+++ b/InvenTree/build/tasks.py
@@ -12,6 +12,7 @@ from allauth.account.models import EmailAddress
 import build.models
 import InvenTree.helpers
 import InvenTree.tasks
+import part.models as part_models
 
 
 logger = logging.getLogger('inventree')
@@ -27,7 +28,18 @@ def check_build_stock(build: build.models.Build):
 
     lines = []
 
-    for bom_item in build.part.get_bom_items():
+    if not build:
+        logger.error("Invalid build passed to 'build.tasks.check_build_stock'")
+        return
+
+    try:
+        part = build.part
+    except part_models.Part.DoesNotExist:
+        # Note: This error may be thrown during unit testing...
+        logger.error("Invalid build.part passed to 'build.tasks.check_build_stock'")
+        return
+
+    for bom_item in part.get_bom_items():
 
         sub_part = bom_item.sub_part