fix rhevm.mark_as_template, need Vm not VmService (#262)

RedHatQE · May 24, 2018 · 7fee2f2 · 7fee2f2
1 parent 7f9e72b
commit 7fee2f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wrapanapi/rhevm.py b/wrapanapi/rhevm.py
@@ -426,8 +426,8 @@ def mark_as_template(self, vm_name, delete=True, temporary_name=None,
 
                 if create_new_template:
                     self.stop_vm(vm_name)
-                    vm = self._get_vm_service(vm_name)
-                    actual_cluster = self._get_cluster(cluster) if cluster else vm.get_cluster()
+                    vm = self._get_vm(vm_name)
+                    actual_cluster = self._get_cluster(cluster) if cluster else vm.cluster
                     new_template = types.Template(
                         name=temp_template_name, vm=vm, cluster=actual_cluster)
                     self.api.system_service().templates_service().add(new_template)