garfield/wpscan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rspec tests

Browse Source
This commit is contained in:
Christian Mehlmauer
2012-09-22 16:00:30 +02:00
parent ae96d93cee
commit b9d8e5a3e2
3 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/browser.rb
View File

@@ -18,7 +18,7 @@
class Browser
@@instance = nil
@@user_agent_modes = %w{ static semi-static random }
USER_AGENT_MODES = %w{ static semi-static random }
ACCESSOR_OPTIONS = [
:user_agent,
@@ -69,7 +69,7 @@ class Browser
def user_agent_mode=(ua_mode)
ua_mode ||= "static"
if @@user_agent_modes.include?(ua_mode)
if USER_AGENT_MODES.include?(ua_mode)
@user_agent_mode = ua_mode
# For semi-static user agent mode, the user agent has to be nil the first time (it will be set with the getter)
@user_agent = nil if ua_mode === "semi-static"