Ensure VmRestorePoints are an Array. (#3)

* Ensure VmRestorePoints are an Array.
virtuozzo · Mar 3, 2020 · c809957 · c809957
1 parent 21e9786
commit c809957
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/recovery_points_for_virtual_server.rb b/recovery_points_for_virtual_server.rb
@@ -2,10 +2,7 @@
   def call(virtual_server)
 
     # Get VmRestorePoint for Virtual Server
-    vm_recovery_points =
-      api_get(
-        build_query(:vmrestorepoint, HierarchyObjRef: "\"#{virtual_server.metadata[:veeam_uid]}\"")
-      ).dig(:QueryResult, :Entities, :VmRestorePoints, :VmRestorePoint)
+    vm_recovery_points = get_vm_recovery_points(virtual_server.metadata[:veeam_uid])
 
     # Iterate over Jobs => Backup => restorePoints => vmRestorePoints
     virtual_server.metadata[:veeam_related_job_ids].flat_map do |job_id|
@@ -72,4 +69,12 @@ def get_recovery_points(resource_type, resource_id, points_type)
 
       points.is_a?(Hash) ? [points] : points
   end
+
+  def get_vm_recovery_points(resource_id)
+    vm_points = api_get(build_query(:vmrestorepoint, HierarchyObjRef: "\"#{resource_id}\""))
+                  .dig(:QueryResult, :Entities, :VmRestorePoints, :VmRestorePoint) || []
+
+    vm_points.is_a?(Hash) ? [vm_points] : vm_points
+  end
+
 end