def wait_until_completed(self, interval=5, timeout=240): """Notifications need a longer timeout, since the backend often has to wait for the request (sending the notification) to timeout itself """ adjusted_timeout = timeout - seconds_since_date_string(self.created) return super(Notification, self).wait_until_completed(interval, adjusted_timeout)
def wait_for_job(self, interval=5, timeout=60, **kw): """Waits until node's job exists""" adjusted_timeout = timeout - seconds_since_date_string(self.created) poll_until(self.job_exists, interval=interval, timeout=adjusted_timeout, **kw) return self
def wait_until_completed(self, interval=5, timeout=60 * 8, since_job_created=True, **kwargs): if since_job_created: timeout = timeout - seconds_since_date_string(self.created) return super(UnifiedJob, self).wait_until_completed(interval, timeout, **kwargs)
def wait_until_status(self, status, interval=1, timeout=60, since_job_created=True, **kwargs): if since_job_created: timeout = timeout - seconds_since_date_string(self.created) return super(UnifiedJob, self).wait_until_status(status, interval, timeout, **kwargs)
def wait_until_status(self, status, interval=5, timeout=30, **kwargs): adjusted_timeout = timeout - seconds_since_date_string(self.created) return super(Notification, self).wait_until_status(status, interval, adjusted_timeout, **kwargs)
def test_seconds_since_date_string(date_string, now, expected): with mock.patch('awxkit.utils.utcnow', return_value=now): assert utils.seconds_since_date_string(date_string) == expected