-
-
Notifications
You must be signed in to change notification settings - Fork 8.1k
/
wait.rb
83 lines (70 loc) · 2.54 KB
/
wait.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
# frozen_string_literal: true
# Licensed to the Software Freedom Conservancy (SFC) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The SFC licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
module Selenium
module WebDriver
class Wait
DEFAULT_TIMEOUT = 5
DEFAULT_INTERVAL = 0.2
#
# Create a new Wait instance
#
# @param [Hash] opts Options for this instance
# @option opts [Numeric] :timeout (5) Seconds to wait before timing out.
# @option opts [Numeric] :interval (0.2) Seconds to sleep between polls.
# @option opts [String] :message Exception mesage if timed out.
# @option opts [Array, Exception] :ignore Exceptions to ignore while polling (default: Error::NoSuchElementError)
#
def initialize(opts = {})
@timeout = opts.fetch(:timeout, DEFAULT_TIMEOUT)
@interval = opts.fetch(:interval, DEFAULT_INTERVAL)
@message = opts[:message]
@ignored = Array(opts[:ignore] || Error::NoSuchElementError)
end
#
# Wait until the given block returns a true value.
#
# @raise [Error::TimeoutError]
# @return [Object] the result of the block
#
def until
end_time = current_time + @timeout
last_error = nil
until current_time > end_time
begin
result = yield
return result if result
rescue *@ignored => last_error # rubocop:disable Naming/RescuedExceptionsVariableName
# swallowed
end
sleep @interval
end
msg = if @message
@message.dup
else
"timed out after #{@timeout} seconds"
end
msg << " (#{last_error.message})" if last_error
raise Error::TimeoutError, msg
end
private
def current_time
Process.clock_gettime(Process::CLOCK_MONOTONIC)
end
end # Wait
end # WebDriver
end # Selenium