def op(self, stage, mocker): op = pipeline_ops_provisioning.PollStatusOperation( fake_operation_id, " ", callback=mocker.MagicMock()) yield op # Clean up any timers set on it if op.polling_timer: op.polling_timer.cancel() if op.retry_after_timer: op.retry_after_timer.cancel() if op.provisioning_timeout_timer: op.provisioning_timeout_timer.cancel()
def send_query_op(self, mocker): op = pipeline_ops_provisioning.PollStatusOperation( fake_operation_id, " ", callback=mocker.MagicMock()) return op