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
|
##
# This file is part of WhatWeb and may be subject to
# redistribution and commercial restrictions. Please see the WhatWeb
# web site for more information on licensing and terms of use.
# https://morningstarsecurity.com/research/whatweb
##
require 'minitest/autorun'
require './lib/whatweb'
class WhatWebTest < Minitest::Test
def setup
@test_host = 'whatweb.net'
end
#
# @note test public methods
#
def test_public_methods
assert_equal(true, WhatWeb::Scan.public_method_defined?(:scan))
assert_equal(true, WhatWeb::Scan.public_method_defined?(:add_target))
assert_equal(true, WhatWeb::Scan.public_method_defined?(:scan_from_plugin))
end
#
# @note test private methods
#
def test_private_methods
assert_equal(true, WhatWeb::Scan.private_method_defined?(:prepare_target))
assert_equal(true, WhatWeb::Scan.private_method_defined?(:make_target_list))
end
def test_invalid_url
assert_raises 'No targets selected' do
WhatWeb::Scan.new(nil)
end
assert_raises 'No targets selected' do
WhatWeb::Scan.new('')
end
assert_raises 'No targets selected' do
WhatWeb::Scan.new([])
end
assert_raises 'No targets selected' do
WhatWeb::Scan.new({})
end
assert_raises 'No targets selected' do
WhatWeb::Scan.new([[]])
end
assert_raises 'No targets selected' do
WhatWeb::Scan.new([{}])
end
end
def test_scanner
scanner = WhatWeb::Scan.new(@test_host)
assert(scanner)
end
def test_scan
max_redirects = 5
plugins = PluginSupport.load_plugins
assert(plugins)
scanner = WhatWeb::Scan.new(@test_host, max_threads: 25)
scanner.scan do |target|
assert(target)
result = WhatWeb::Parser.run_plugins(target, plugins, scanner: scanner)
assert(result)
WhatWeb::Redirect.new(target, scanner, max_redirects)
whatweb_result = WhatWeb::Parser.parse(target, result)
assert(whatweb_result['target'])
assert(whatweb_result['status'])
assert(whatweb_result['result'])
countries = whatweb_result['result'].select { |a| a[0] == 'Country' }
assert_equal('Country', countries.first[0])
end
end
end
|
