forked from teampoltergeist/poltergeist
/
driver.rb
154 lines (122 loc) · 3.08 KB
/
driver.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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
module Capybara::Poltergeist
class Driver < Capybara::Driver::Base
DEFAULT_TIMEOUT = 30
attr_reader :app, :app_server, :server, :client, :browser, :options
attr_accessor :headers
def initialize(app, options = {})
@app = app
@options = options
@browser = nil
@inspector = nil
@server = nil
@client = nil
@headers = {}
@app_server = Capybara::Server.new(app)
@app_server.boot if Capybara.run_server
end
def browser
@browser ||= Browser.new(server, client, logger, js_errors)
end
def inspector
@inspector ||= options[:inspector] && Inspector.new(options[:inspector])
end
def server
@server ||= Server.new(options.fetch(:timeout, DEFAULT_TIMEOUT))
end
def client
@client ||= Client.start(server.port,
:path => options[:phantomjs],
:window_size => options[:window_size],
:phantomjs_options => phantomjs_options
)
end
def phantomjs_options
list = options[:phantomjs_options] || []
list += ["--remote-debugger-port=#{inspector.port}", "--remote-debugger-autorun=yes"] if inspector
list
end
def client_pid
client.pid
end
def timeout
server.timeout
end
def timeout=(sec)
server.timeout = sec
end
def restart
browser.restart
end
def quit
server.stop
client.stop
end
# logger should be an object that responds to puts, or nil
def logger
options[:logger] || (options[:debug] && STDERR)
end
def js_errors
options.fetch(:js_errors, true)
end
def visit(path)
browser.visit app_server.url(path), @headers
end
def current_url
browser.current_url
end
def status_code
browser.status_code
end
def body
browser.body
end
def source
browser.source.to_s
end
def find(selector)
browser.find(selector).map { |page_id, id| Capybara::Poltergeist::Node.new(self, page_id, id) }
end
def evaluate_script(script)
browser.evaluate(script)
end
def execute_script(script)
browser.execute(script)
nil
end
def within_frame(id, &block)
browser.within_frame(id, &block)
end
def reset!
browser.reset
@headers = {}
end
def render(path, options = {})
browser.render(path, options)
end
def resize(width, height)
browser.resize(width, height)
end
def network_traffic
browser.network_traffic
end
def debug
if @options[:inspector]
inspector.open
pause
else
raise Error, "To use the remote debugging, you have to launch the driver " \
"with `:inspector => true` configuration option"
end
end
def pause
STDERR.puts "Poltergeist execution paused. Press enter to continue."
STDIN.gets
end
def wait?
true
end
def invalid_element_errors
[Capybara::Poltergeist::ObsoleteNode, Capybara::Poltergeist::ClickFailed]
end
end
end