Add missing arg in wait function which overwrites parent function (#2574)

Co-authored-by: Meni Yakove <441263+myakove@users.noreply.github.com>

Author: rnetser

ocp_resources/datavolume.py

@@ -248,15 +248,16 @@ def wait_deleted(self, timeout=TIMEOUT_4MINUTES):
         super().wait_deleted(timeout=timeout)
         return self.pvc.wait_deleted(timeout=timeout)
 
-    def wait(self, timeout=TIMEOUT_10MINUTES, failure_timeout=TIMEOUT_2MINUTES, wait_for_exists_only=False):
+    def wait(self, timeout=TIMEOUT_10MINUTES, failure_timeout=TIMEOUT_2MINUTES, wait_for_exists_only=False, sleep=1):
         if wait_for_exists_only:
-            return super().wait(timeout=timeout)
+            return super().wait(timeout=timeout, sleep=sleep)
         else:
             self._check_none_pending_status(failure_timeout=failure_timeout)
 
             # If DV's status is not Pending, continue with the flow
             self.wait_for_status(status=self.Status.SUCCEEDED, timeout=timeout)
             self.pvc.wait_for_status(status=PersistentVolumeClaim.Status.BOUND, timeout=timeout)
+            return None
 
     @property
     def pvc(self):

ocp_resources/virtual_machine_import.py

@@ -195,11 +195,12 @@ def wait(
         cond_reason=SucceededConditionReason.VIRTUAL_MACHINE_READY,
         cond_status=Condition.Status.TRUE,
         cond_type=Condition.SUCCEEDED,
+        sleep=1,
     ):
         self.logger.info(f"Wait for {self.kind} {self.name} {cond_reason} condition to be {cond_status}")
         samples = TimeoutSampler(
             wait_timeout=timeout,
-            sleep=1,
+            sleep=sleep,
             exceptions_dict=PROTOCOL_ERROR_EXCEPTION_DICT,
             func=self.api.get,
             field_selector=f"metadata.name=={self.name}",