slapos_cloud: Catch not found computer when getting compute partition

61b8f6f3 · Cédric Le Ninivin · Titouan Soulard · 7ca3a9b7 · 61b8f6f3
Commit 61b8f6f3 authored Sep 28, 2022 by Cédric Le Ninivin Committed by Titouan Soulard Nov 15, 2024
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 4 deletions

master/bt5/slapos_cloud/MixinTemplateItem/portal_components/mixin.erp5.SlapOSCatalogToolCacheMixin.py ...rtal_components/mixin.erp5.SlapOSCatalogToolCacheMixin.py +8 -4

No files found.
--- a/master/bt5/slapos_cloud/MixinTemplateItem/portal_components/mixin.erp5.SlapOSCatalogToolCacheMixin.py
+++ b/master/bt5/slapos_cloud/MixinTemplateItem/portal_components/mixin.erp5.SlapOSCatalogToolCacheMixin.py
@@ -71,9 +71,13 @@ class SlapOSCatalogToolCacheMixin(object):
  @UnrestrictedMethod
  def _getNonCachedComputeNodeUid(self, reference):
-    return self.unrestrictedSearchResults(
+    compute_node_list = self.unrestrictedSearchResults(
-      portal_type=['Compute Node', 'Remote Node'], reference=reference,
+      portal_type=['Compute Node', 'Remote Node'],
-      validation_state="validated")[0].UID
+      reference=reference,
+      validation_state="validated")
+    if len(compute_node_list) != 1:
+      raise NotFound, "No document found with parameters: %s" % reference
+    return compute_node_list[0].UID
  def getComputePartitionObject(self, compute_node_reference,
                                    compute_partition_reference):