Enforce head in Design a Pod

Change-Id: Ibd9858750e4529450483c66907061440dc8389f3
Signed-off-by: Sean Smith <ssmith@iol.unh.edu>

1 files changed, 5 insertions, 0 deletions

diff --git a/src/workflow/resource_bundle_workflow.py b/src/workflow/resource_bundle_workflow.py
index 391d33e..639e7a8 100644
--- a/src/workflow/resource_bundle_workflow.py
+++ b/src/workflow/resource_bundle_workflow.py
@@ -10,6 +10,7 @@
 
 from django.conf import settings
 from django.forms import formset_factory
+from django.core.exceptions import ValidationError
 
 from typing import List
 
@@ -265,6 +266,10 @@ class Define_Software(WorkflowStep):
                 host.image = image
                 host.save()
 
+            if not has_headnode:
+                self.set_invalid("No headnode. Please set a headnode.")
+                return
+
             self.set_valid("Completed")
         else:
             self.set_invalid("Please complete all fields")