Class: SeleniumStandaloneDSL::Base

Inherits:
Object
  • Object
show all
Defined in:
lib/selenium_standalone_dsl/base.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(log_path: '/tmp/selenium_standalone_dsl', user_agent: 'Selenium Standalone DSL', headless: false) ⇒ Base

Returns a new instance of Base.



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
 
# File 'lib/selenium_standalone_dsl/base.rb', line 5

def initialize(log_path: '/tmp/selenium_standalone_dsl',
               user_agent: 'Selenium Standalone DSL', headless: false)

  if headless
    Headless.new(reuse: false, destroy_at_exit: true).start
  end

  # Extends timeout
  client = Selenium::WebDriver::Remote::Http::Default.new
  client.timeout = 120

  # Create profile so that each driver instance has its own cookie
  profile = Selenium::WebDriver::Firefox::Profile.new
  profile['general.useragent.override'] = user_agent
  profile.native_events = true

  @driver = Selenium::WebDriver::Driver.for(:firefox, :http_client => client, :profile => profile)

  @logger = Logger.new(log_path)
  @logger.info '=' * 100
  @logger.info 'SeleniumStandaloneDsl started'
end

Instance Attribute Details

#driver ⇒ Object (readonly)

Returns the value of attribute driver.



3
4
5
 
# File 'lib/selenium_standalone_dsl/base.rb', line 3

def driver
  @driver
end

Instance Method Details

#click(selector, find_by: :link_text) ⇒ Object

The following methods are utility methods for SeleniumStandaloneDsl-DSL. You can easily handle driver with this DSL.



30
31
32
33
34
35
36
 
# File 'lib/selenium_standalone_dsl/base.rb', line 30

def click(selector, find_by: :link_text)
  sleep Random.new.rand(1..2)
  with_frame do
    @driver.find_element(find_by, selector).click
  end
  sleep Random.new.rand(1..2)
end

#fill_in(selector, find_by: :name, with: nil) ⇒ Object 



44
45
46
47
48
49
50
 
# File 'lib/selenium_standalone_dsl/base.rb', line 44

def fill_in(selector, find_by: :name, with: nil)
  return if !with

  element = @driver.find_element(find_by, selector)
  0.upto(100) { element.send_keys "\b" }
  element.send_keys with
end

#has_element?(selector, find_by: :link_text) ⇒ Boolean

Returns:

  • (Boolean) 


67
68
69
70
71
72
73
74
75
76
77
 
# File 'lib/selenium_standalone_dsl/base.rb', line 67

def has_element?(selector, find_by: :link_text)
  if [:text, :link_text].include?(find_by)
    @driver.page_source.match(selector) != nil
  else
    begin
      @driver.find_element(find_by, selector).is_a? Selenium::WebDriver::Element
    rescue Selenium::WebDriver::Error::NoSuchElementError
      false
    end
  end
end

#iframe(method, selector) ⇒ Object 



85
86
87
88
89
90
91
 
# File 'lib/selenium_standalone_dsl/base.rb', line 85

def iframe(method, selector)
  current_frame = @driver.window_handles.first
  @driver.switch_to.frame @driver.find_element(method, selector)
  yield
  # Frame might has gone away
  @driver.switch_to.frame current_frame rescue Selenium::WebDriver::Error::NoSuchFrameError
end

#popup ⇒ Object 



38
39
40
41
42
 
# File 'lib/selenium_standalone_dsl/base.rb', line 38

def popup
  @driver.switch_to.window @driver.window_handles.last
  yield
  @driver.switch_to.window @driver.window_handles.first
end

#prompt_ok ⇒ Object 



52
53
54
 
# File 'lib/selenium_standalone_dsl/base.rb', line 52

def prompt_ok
  @driver.switch_to.alert.accept
end

#quit ⇒ Object 



110
111
112
 
# File 'lib/selenium_standalone_dsl/base.rb', line 110

def quit
  @driver.quit
end

#search(selector) ⇒ Object

Provide jQuery-like search using Nokogiri::HTML

Returns:

  • Array of [Selenium::WebDriver::Element]



81
82
83
 
# File 'lib/selenium_standalone_dsl/base.rb', line 81

def search(selector)
  Nokogiri::HTML.parse(@driver.page_source).search selector
end

#select(text, from: nil, find_by: :name) ⇒ Object 



56
57
58
59
60
61
62
63
64
65
 
# File 'lib/selenium_standalone_dsl/base.rb', line 56

def select(text, from: nil, find_by: :name)
  return if !from

  # For legacy sites using Frame
  element = with_frame do
    @driver.find_element(find_by, from)
  end
  select = Selenium::WebDriver::Support::Select.new(element)
  select.select_by :text, text
end

#visit(url) ⇒ Object 



106
107
108
 
# File 'lib/selenium_standalone_dsl/base.rb', line 106

def visit(url)
  @driver.get url
end

#with_frame(&block) ⇒ Object 



93
94
95
96
97
98
99
100
101
102
103
104
 
# File 'lib/selenium_standalone_dsl/base.rb', line 93

def with_frame(&block)
  begin
    block.call
  rescue Selenium::WebDriver::Error::NoSuchElementError
    begin
      @driver.switch_to.frame 1
    rescue Selenium::WebDriver::Error::NoSuchFrameError
      raise Selenium::WebDriver::Error::NoSuchElementError
    end
    block.call
  end
end